make write_network_to_db return net metadata

e2nIEE · Feb 12, 2024 · 28be19c · 28be19c
1 parent d6ceeac
commit 28be19c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pandahub/lib/PandaHub.py b/pandahub/lib/PandaHub.py
@@ -3,6 +3,7 @@
 import importlib
 import json
 import logging
+import pymongo
 import traceback
 import warnings
 from inspect import signature, _empty
@@ -385,7 +386,7 @@ def _get_project_document(self, filter_dict: dict) -> Optional[dict]:
         else:
             return project_doc
 
-    def _get_project_database(self):
+    def _get_project_database(self) -> pymongo.mongo_client:
         return self.mongo_client[str(self.active_project["_id"])]
 
     def _get_global_database(self):
@@ -1068,6 +1069,7 @@ def write_network_to_db(
         if metadata is not None:
             net_dict.update(metadata)
         db["_networks"].insert_one(net_dict)
+        return net_dict
 
     def _write_net_collections_to_db(self, db, collections):
         for element, element_data in collections.items():
@@ -1725,6 +1727,7 @@ def get_variant_filter(self, variants):
                     "$or": [
                         {"var_type": "base", "not_in_var": {"$nin": variants}},
                         {
+                            # redundant?
                             "var_type": {"$in": ["change", "addition"]},
                             "variant": {"$in": variants},
                         },
@@ -1737,6 +1740,7 @@ def get_variant_filter(self, variants):
             return {
                 "$or": [
                     {"var_type": "base", "not_in_var": {"$ne": variants}},
+                    # var type redundant
                     {"var_type": {"$in": ["change", "addition"]}, "variant": variants},
                 ]
             }