25 lines
822 B
Python
25 lines
822 B
Python
|
import ast
|
||
|
|
|
||
|
|
|
||
|
|
def find_globals(g, tree):
|
||
|
|
"""Uses AST to find globals in an ast tree"""
|
||
|
|
for child in tree:
|
||
|
|
if hasattr(child, 'body') and isinstance(child.body, list):
|
||
|
|
find_globals(g, child.body)
|
||
|
|
elif isinstance(child, (ast.FunctionDef, ast.ClassDef)):
|
||
|
|
g.add(child.name)
|
||
|
|
continue
|
||
|
|
elif isinstance(child, ast.Assign):
|
||
|
|
try:
|
||
|
|
g.add(child.targets[0].id)
|
||
|
|
except (IndexError, AttributeError):
|
||
|
|
pass
|
||
|
|
elif isinstance(child, ast.Import):
|
||
|
|
g.add(child.names[0].name)
|
||
|
|
elif isinstance(child, ast.ImportFrom):
|
||
|
|
for name in child.names:
|
||
|
|
g_name = name.asname or name.name
|
||
|
|
if g_name == '*':
|
||
|
|
continue
|
||
|
|
g.add(g_name)
|