code cleanups

invana · Mar 10, 2022 · dce3af3 · dce3af3
1 parent aee4ce6
commit dce3af3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/invana_engine/core/schema_generator.py b/invana_engine/core/schema_generator.py
@@ -296,7 +296,7 @@ def create_label_fields_of_search_type(self, search_type):
 
     # def create_label_fields_of_label(self, record, )
 
-    def create_schema_dynamically(self):
+    def create_schema_dynamically(self, *extra_schema_types):
         types = []
         query_schemas = []
         for search_type in ["node", "edge"]:
@@ -318,7 +318,7 @@ def create_schema_dynamically(self):
         # class Query(ModellerQuery, GraphSchema, *query_schemas):
         #     pass
 
-        class Query(*query_schemas):
+        class Query(*extra_schema_types, *query_schemas):
             pass
 
         return graphene.Schema(query=Query,

diff --git a/invana_engine/server/schema.py b/invana_engine/server/schema.py
@@ -33,7 +33,7 @@ def get_schema():
 
     schema_generator = DynamicSchemaGenerator(schema_store)
 
-    return schema_generator.create_schema_dynamically()
+    return schema_generator.create_schema_dynamically(ModellerQuery, GraphSchema)
 
     # edge_schema_generator = DynamicSchemaGenerator(edges_schema_data_json, "edge", schema_store)
     #