Bounded and Undirected give access to attributes of underlying graph

MaineKuehn · Jul 31, 2018 · cb39fdd · cb39fdd
1 parent b80b9c4
commit cb39fdd
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/graphi/types/bounded.py b/graphi/types/bounded.py
@@ -58,6 +58,14 @@ def _ensure_bounds(self):
         for tail, head in blacklist:
             del self._graph[tail:head]
 
+    def __getattr__(self, item):
+        return getattr(self._graph, item)
+
+    def __setattr__(self, key, value):
+        if key not in ('_graph', 'value_bound'):
+            setattr(self._graph, key, value)
+        object.__setattr__(self, key, value)
+
     def __getitem__(self, item):
         return self._graph[item]
 

diff --git a/graphi/types/undirected.py b/graphi/types/undirected.py
@@ -83,6 +83,14 @@ def _ensure_symmetry(self):
         if graph_diff:
             graph.update(graph_diff)
 
+    def __getattr__(self, item):
+        return getattr(self._graph, item)
+
+    def __setattr__(self, key, value):
+        if key != '_graph':
+            setattr(self._graph, key, value)
+        object.__setattr__(self, key, value)
+
     def __getitem__(self, item):
         return self._graph[item]