1
0
forked from VimPlug/jedi

Fix ExprStmt docstring bugs.

This commit is contained in:
Dave Halter
2015-03-03 17:42:49 +01:00
parent ff0c7e27d3
commit 40e61fc96d
2 changed files with 15 additions and 3 deletions

View File

@@ -65,10 +65,17 @@ class DocstringMixin(object):
""" Returns a cleaned version of the docstring token. """
if isinstance(self, Module):
stmt = self.children[0]
else:
elif isinstance(self, ClassOrFunc):
stmt = self.children[self.children.index(':') + 1]
if is_node(stmt, 'suite'): # Normally a suite
stmt = stmt.children[2] # -> NEWLINE INDENT stmt
else:
c = self.parent.children
index = c.index(self) - 1
if not index:
return ''
stmt = c[index - 1]
if is_node(stmt, 'simple_stmt'):
stmt = stmt.children[0]