1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
""" ImportedIds gathers identifiers imported by a node. """
from pythran.analyses.globals_analysis import Globals
from pythran.analyses.locals_analysis import Locals
from pythran.passmanager import NodeAnalysis
import pythran.metadata as md
import gast as ast
class ImportedIds(NodeAnalysis):
"""Gather ids referenced by a node and not declared locally."""
def __init__(self):
self.result = set()
self.current_locals = set()
self.is_list = False
self.in_augassign = False
super(ImportedIds, self).__init__(Globals, Locals)
def visit_Name(self, node):
if isinstance(node.ctx, ast.Store) and not self.in_augassign:
self.current_locals.add(node.id)
elif (node.id not in self.visible_globals and
node.id not in self.current_locals):
self.result.add(node.id)
def visit_FunctionDef(self, node):
self.current_locals.add(node.name)
current_locals = self.current_locals.copy()
self.current_locals.update(arg.id for arg in node.args.args)
for stmt in node.body:
self.visit(stmt)
self.current_locals = current_locals
def visit_AnyComp(self, node):
current_locals = self.current_locals.copy()
for generator in node.generators:
self.visit(generator)
self.visit(node.elt)
self.current_locals = current_locals
visit_ListComp = visit_AnyComp
visit_SetComp = visit_AnyComp
visit_DictComp = visit_AnyComp
visit_GeneratorExp = visit_AnyComp
def visit_Assign(self, node):
# order matter as an assignation
# is evaluated before being assigned
md.visit(self, node)
self.visit(node.value)
for target in node.targets:
self.visit(target)
def visit_AugAssign(self, node):
self.in_augassign = True
self.generic_visit(node)
self.in_augassign = False
def visit_Lambda(self, node):
current_locals = self.current_locals.copy()
self.current_locals.update(arg.id for arg in node.args.args)
self.visit(node.body)
self.current_locals = current_locals
def visit_Import(self, node):
self.current_locals.update(alias.name for alias in node.names)
def visit_StoredTuple(self, node):
for elt in node.elts:
if isinstance(elt, ast.Name):
self.current_locals.add(elt.id)
continue
if isinstance(elt, ast.Subscript):
self.visit(elt)
if isinstance(elt, ast.Tuple):
self.visit_StoredTuple(node)
def visit_Tuple(self, node):
if isinstance(node.ctx, ast.Load):
self.generic_visit(node)
else:
self.visit_StoredTuple(node)
visit_List = visit_Tuple
def visit_ImportFrom(self, node):
self.current_locals.update(alias.name for alias in node.names)
def visit_Attribute(self, node):
pass
def prepare(self, node):
super(ImportedIds, self).prepare(node)
if self.is_list: # so that this pass can be called on list
node = node.body[0]
self.visible_globals = set(self.globals) - self.locals[node]
def run(self, node):
if isinstance(node, list): # so that this pass can be called on list
self.is_list = True
node = ast.If(ast.Constant(1, None), node, [])
return super(ImportedIds, self).run(node)
|
