Treat lists like tuples in AST function get_lvar_namesl

This commit is contained in:
Peter Cock 2017-10-10 16:26:28 +01:00
parent 350f196c4f
commit cc89093777

View File

@ -51,7 +51,7 @@ def get_lvar_names(node, self=None):
return [node.id] # type: ignore return [node.id] # type: ignore
else: else:
raise TypeError('The assignment %r is not instance variable' % node) raise TypeError('The assignment %r is not instance variable' % node)
elif node_name == 'Tuple': elif node_name in ('Tuple', 'List'):
members = [get_lvar_names(elt) for elt in node.elts] # type: ignore members = [get_lvar_names(elt) for elt in node.elts] # type: ignore
return sum(members, []) return sum(members, [])
elif node_name == 'Attribute': elif node_name == 'Attribute':