Generate type hints, fixes #987

2020-02-02 16:55:10 +01:00
parent f4b1fc479d
commit 4c7179bc87
8 changed files with 91 additions and 0 deletions
@@ -130,6 +130,9 @@ class AbstractInstanceValue(Value):
            for name in names
        )

+    def get_type_hint(self, add_class_info=True):
+        return self.py__name__()
+
    def __repr__(self):
        return "<%s of %s>" % (self.__class__.__name__, self.class_value)

@@ -221,6 +221,11 @@ class ClassMixin(object):
    def _as_context(self):
        return ClassContext(self)

+    def get_type_hint(self, add_class_info=True):
+        if add_class_info:
+            return 'Type[%s]' % self.py__name__()
+        return self.py__name__()
+

 class ClassValue(use_metaclass(CachedMetaClass, ClassMixin, FunctionAndClassBase)):
    api_type = u'class'