1
0
forked from VimPlug/jedi

Include function return type annotation in docstring if it is present

This commit is contained in:
Dima Gerasimov
2018-03-05 18:24:28 +00:00
committed by Dave Halter
parent 88243d2408
commit ceb5509170
3 changed files with 8 additions and 2 deletions

View File

@@ -156,7 +156,11 @@ def get_call_signature(funcdef, width=72, call_string=None):
p = '(' + ''.join(param.get_code() for param in funcdef.get_params()).strip() + ')'
else:
p = funcdef.children[2].get_code()
code = call_string + p
if funcdef.annotation:
rtype = " ->" + funcdef.annotation.get_code()
else:
rtype = ""
code = call_string + p + rtype
return '\n'.join(textwrap.wrap(code, width))