Cleanup with_stmt.

This commit is contained in:
Dave Halter committed 2017-04-09 21:20:33 +02:00
1 parent f855c2bb70
commit 769cc80d6b
2 files changed
+6 -7

No files matched your search

+5 -6
View File
@@ -789,12 +789,11 @@ class WithStmt(Flow):
names += _defined_names(with_item.children[2])
return names
def node_from_name(self, name):
node = name
while True:
node = node.parent
if node.type == 'with_item':
return node.children[0]
def get_context_manager_from_name(self, name):
node = name.parent
if node.type != 'with_item':
raise ValueError('The name is not actually part of a with statement.')
return node.children[0]
class Import(PythonBaseNode):