added detecting missing keys in schema context

Kraken-CI · Aug 26, 2023 · 76edd73 · 76edd73
1 parent 2f4f118
commit 76edd73
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/server/kraken/server/initdb.py b/server/kraken/server/initdb.py
@@ -306,7 +306,7 @@ def _prepare_builtin_tools():
             "required": ["source"],
             "properties": {
                 "action": {
-                    "description": "An action that artifacts tool should execute. Default is `upload`.",
+                    "description": "An action that artifacts tool should execute: 'download' or 'upload'. Default is `upload`.",
                     "type": "string",
                     "enum": ["download", "upload"]
                 },
@@ -426,7 +426,7 @@ def _prepare_builtin_tools():
             "required": ["action"],
             "properties": {
                 "action": {
-                    "description": "An action that the tool should perform.",
+                    "description": "An action that the tool should perform: 'save' or 'restore'.",
                     "type": "string",
                     "enum": ["save", "restore"]
                 },

diff --git a/server/kraken/server/schema.py b/server/kraken/server/schema.py
@@ -38,7 +38,8 @@ class SchemaError(Exception):
 
 
 class SchemaCodeContext(addict.Dict):
-    pass
+    def __missing__(self, key):
+        raise Exception('missing field %s' %key)
 
 
 def execute_schema_code(schema_code, context=None):