VimPlug/parso
1
0
Fork 0
mirror of https://github.com/davidhalter/parso.git synced 2025-12-06 21:04:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #152 from isidentical/issue-151

Browse Source 
Retrieve all kinds of assignment targets from with test
This commit is contained in:
Dave Halter
2020-09-29 00:14:19 +02:00
committed by GitHub
parent 6ae0efa415 52cfa5a8ac
commit 634df56d90
2 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
parso/python/tree.py
View File

@@ -775,8 +775,8 @@ class WithStmt(Flow):
return names return names
def get_test_node_from_name(self, name): def get_test_node_from_name(self, name):
node = name.parent node = search_ancestor(name, "with_item")
if node.type != 'with_item': if node is None:
raise ValueError('The name is not actually part of a with statement.') raise ValueError('The name is not actually part of a with statement.')
return node.children[0] return node.children[0]

10
test/test_parser_tree.py
View File

@@ -229,3 +229,13 @@ def test_iter_funcdefs():
module = parse(code, version='3.8') module = parse(code, version='3.8')
func_names = [f.name.value for f in module.iter_funcdefs()] func_names = [f.name.value for f in module.iter_funcdefs()]
assert func_names == ['normal', 'asyn', 'dec_normal', 'dec_async'] assert func_names == ['normal', 'asyn', 'dec_normal', 'dec_async']
def test_with_stmt_get_test_node_from_name():
code = "with A as X.Y, B as (Z), C as Q[0], D as Q['foo']: pass"
with_stmt = parse(code, version='3').children[0]
tests = [
with_stmt.get_test_node_from_name(name).value
for name in with_stmt.get_defined_names(include_setitem=True)
]
assert tests == ["A", "B", "C", "D"]