temp: NodeDict.update hack

Just a temporary solution. Will be removed shortly

Author: aMahanna

nx_arangodb/classes/dict.py

@@ -715,9 +715,21 @@ def clear(self) -> None:
 
     @keys_are_strings
     @logger_debug
-    def update(self, nodes: Any) -> None:
+    def update(self, nodes: dict[str, dict[str, Any]]) -> None:
         """g._node.update({'node/1': {'foo': 'bar'}, 'node/2': {'baz': 'qux'}})"""
-        raise NotImplementedError("NodeDict.update()")
+        # TODO: Remove in preparation for GA-153
+        # This is just a temporary solution to test something out
+        import networkx as nx
+
+        config = nx.config.backends.arangodb
+        sample_node_id = next(iter(nodes.keys()))
+        collection = sample_node_id.split("/")[0]
+        nodes_list = [
+            {"_id": node_id, **node_data} for node_id, node_data in nodes.items()
+        ]
+        self.db.collection(collection).import_bulk(
+            nodes_list, batch_size=config.write_batch_size, on_duplicate="update"
+        )
 
     @logger_debug
     def values(self) -> Any:

nx_arangodb/classes/reportviews.py

@@ -21,6 +21,9 @@ def data(self, data=True, default=None):
             return self
         return CustomNodeDataView(self._nodes, data, default)
 
+    def update(self, data):
+        self._nodes.update(data)
+
 
 class CustomNodeDataView(nx.classes.reportviews.NodeDataView):
     def __iter__(self):