Reworked ExprStmt.get_definition

This commit is contained in:
Dave Halter
2014-10-21 14:21:59 +02:00
parent 1c27759c4f
commit c1807e5f33

View File

@@ -1039,7 +1039,11 @@ class Statement(Simple, DocstringMixin):
names = []
for i in range(0, len(self.children) - 2, 2):
if self.children[i + 1].value == '=':
names.append(self.children[i])
current = self.children[i]
if is_node(current, 'testlist_star_expr'):
names += current.children[::2]
else:
names.append(current)
return names