add BaseNodeType.get_self for de-referencing weakref.proxy(node)

devdoomari · devdoomari3 · commit 083a9f878446 · 2020-09-22T11:41:08.000+09:00
diff --git a/py_type_extractor/type_extractor/nodes/BaseNodeType.py b/py_type_extractor/type_extractor/nodes/BaseNodeType.py
@@ -9,6 +9,10 @@ class BaseOption:
 class BaseNodeType:
     options: Set[BaseOption]
 
+    # for de-referencing weakref.proxy
+    def get_self(self):
+        return self
+
 
 NodeType = Union[BaseNodeType, type]
 
