fix in custom schema (rest api)

aiida/restapi/translator/computer.py

@@ -30,10 +30,18 @@ class ComputerTranslator(BaseTranslator):
 
     _result_type = __label__
 
+    _default_projections = None
+
     # Extract the default projections from custom_schema if they are defined
-    if 'columns' in custom_schema:
-        _default_projections = custom_schema['columns'][__label__]
-    else:
+    if custom_schema is not None:
+        try:
+            # check if schema is defined for given node type
+            if __label__ in custom_schema['columns'].keys():
+                _default_projections = custom_schema['columns'][__label__]
+        except KeyError:
+            pass
+
+    if _default_projections is None:
         _default_projections = ['**']
 
     def __init__(self, **kwargs):

aiida/restapi/translator/group.py

@@ -31,10 +31,18 @@ class GroupTranslator(BaseTranslator):
 
     _result_type = __label__
 
+    _default_projections = None
+
     # Extract the default projections from custom_schema if they are defined
-    if 'columns' in custom_schema:
-        _default_projections = custom_schema['columns'][__label__]
-    else:
+    if custom_schema is not None:
+        try:
+            # check if schema is defined for given node type
+            if __label__ in custom_schema['columns'].keys():
+                _default_projections = custom_schema['columns'][__label__]
+        except KeyError:
+            pass
+
+    if _default_projections is None:
         _default_projections = ['**']
 
 

aiida/restapi/translator/node.py

@@ -59,11 +59,18 @@ def __init__(self, Class=None, **kwargs):
         # basic initialization
         super(NodeTranslator, self).__init__(Class=Class, **kwargs)
 
+        self._default_projections = None
+
         # Extract the default projections from custom_schema if they are defined
-        if self.custom_schema is not None and 'columns' in self.custom_schema:
-            self._default_projections = self.custom_schema['columns'][
-                self.__label__]
-        else:
+        if self.custom_schema is not None:
+            try:
+                # check if schema is defined for given node type
+                if self.__label__ in self.custom_schema['columns'].keys():
+                    self._default_projections = self.custom_schema['columns'][self.__label__]
+            except KeyError:
+                pass
+
+        if self._default_projections is None:
             self._default_projections = ['**']
 
         # Inspect the subclasses of NodeTranslator, to avoid hard-coding

aiida/restapi/translator/user.py

@@ -30,10 +30,18 @@ class UserTranslator(BaseTranslator):
 
     _result_type = __label__
 
+    _default_projections = None
+
     # Extract the default projections from custom_schema if they are defined
-    if 'columns' in custom_schema:
-        _default_projections = custom_schema['columns'][__label__]
-    else:
+    if custom_schema is not None:
+        try:
+            # check if schema is defined for given node type
+            if __label__ in custom_schema['columns'].keys():
+                _default_projections = custom_schema['columns'][__label__]
+        except KeyError:
+            pass
+
+    if _default_projections is None:
         # send only white listed keys in default response for user for security reasons
         _default_projections = ['id', 'first_name', "last_name", 'institution', 'date_joined']