Better annotation signature string for classes

This commit is contained in:
Dave Halter
2019-05-22 20:33:24 +02:00
parent 3bec1a6938
commit b2b08ab432

View File

@@ -16,7 +16,7 @@ class AbstractSignature(object):
s = self.name.string_name + '(' + param_code + ')'
annotation = self.annotation
if annotation is not None:
s += ' -> ' + annotation.get_code(include_prefix=False)
s += ' -> ' + annotation.get_code(include_prefix=False)
return s
def bind(self, context):
@@ -39,6 +39,10 @@ class TreeSignature(AbstractSignature):
@property
def annotation(self):
# Classes don't need annotations, even if __init__ has one. They always
# return themselves.
if self.context.is_class():
return None
return self._function_context.tree_node.annotation