def __init__(self, start, end, is_global):
self.start = start
self.end = end
self.is_global = is_global
self.prewritten = OrderedSet()
self.maybe_written = OrderedSet()
self.written = OrderedSet()
self.read = OrderedSet()
self.postread = OrderedSet()
self.postwritten = OrderedSet()
self.host_function = True
self.conditional = False
def __init__(self, start, end, is_global):
self.start = start
self.end = end
self.is_global = is_global
self.prewritten = OrderedSet()
self.maybe_written = OrderedSet()
self.written = OrderedSet()
self.read = OrderedSet()
self.postread = OrderedSet()
self.postwritten = OrderedSet()
self.host_function = True
self.conditional = False
class _FunctionInformationCollector(object):
def __init__(self, start, end, is_global):
self.start = start
self.end = end
self.is_global = is_global
self.prewritten = OrderedSet()
self.maybe_written = OrderedSet()
self.written = OrderedSet()
self.read = OrderedSet()
self.postread = OrderedSet()
self.postwritten = OrderedSet()
self.host_function = True
self.conditional = False
def _read_variable(self, name, lineno):
if self.start <= lineno <= self.end:
if name not in self.written:
if not self.conditional or name not in self.maybe_written:
self.read.add(name)
if self.end < lineno:
if name not in self.postwritten:
self.postread.add(name)
def _written_variable(self, name, lineno):
if self.start <= lineno <= self.end:
if self.conditional:
self.maybe_written.add(name)
else:
self.written.add(name)
if self.start > lineno:
self.prewritten.add(name)
if self.end < lineno:
self.postwritten.add(name)
def _FunctionDef(self, node):
if not self.is_global and self.host_function:
self.host_function = False
for name in _get_argnames(node.args):
self._written_variable(name, node.lineno)
for child in node.body:
ast.walk(child, self)
else:
self._written_variable(node.name, node.lineno)
visitor = _VariableReadsAndWritesFinder()
for child in node.body:
ast.walk(child, visitor)
for name in visitor.read - visitor.written:
self._read_variable(name, node.lineno)
def _Name(self, node):
if isinstance(node.ctx, (ast.Store, ast.AugStore)):
self._written_variable(node.id, node.lineno)
if not isinstance(node.ctx, ast.Store):
self._read_variable(node.id, node.lineno)
def _Assign(self, node):
ast.walk(node.value, self)
for child in node.targets:
ast.walk(child, self)
def _ClassDef(self, node):
self._written_variable(node.name, node.lineno)
def _handle_conditional_node(self, node):
self.conditional = True
try:
for child in ast.get_child_nodes(node):
ast.walk(child, self)
finally:
self.conditional = False
def _If(self, node):
self._handle_conditional_node(node)
def _While(self, node):
self._handle_conditional_node(node)
def _For(self, node):
self.conditional = True
try:
# iter has to be checked before the target variables
ast.walk(node.iter, self)
ast.walk(node.target, self)
for child in node.body:
ast.walk(child, self)
for child in node.orelse:
ast.walk(child, self)
finally:
self.conditional = False
class _FunctionInformationCollector(object):
def __init__(self, start, end, is_global):
self.start = start
self.end = end
self.is_global = is_global
self.prewritten = OrderedSet()
self.maybe_written = OrderedSet()
self.written = OrderedSet()
self.read = OrderedSet()
self.postread = OrderedSet()
self.postwritten = OrderedSet()
self.host_function = True
self.conditional = False
def _read_variable(self, name, lineno):
if self.start <= lineno <= self.end:
if name not in self.written:
if not self.conditional or name not in self.maybe_written:
self.read.add(name)
if self.end < lineno:
if name not in self.postwritten:
self.postread.add(name)
def _written_variable(self, name, lineno):
if self.start <= lineno <= self.end:
if self.conditional:
self.maybe_written.add(name)
else:
self.written.add(name)
if self.start > lineno:
self.prewritten.add(name)
if self.end < lineno:
self.postwritten.add(name)
def _FunctionDef(self, node):
if not self.is_global and self.host_function:
self.host_function = False
for name in _get_argnames(node.args):
self._written_variable(name, node.lineno)
for child in node.body:
ast.walk(child, self)
else:
self._written_variable(node.name, node.lineno)
visitor = _VariableReadsAndWritesFinder()
for child in node.body:
ast.walk(child, visitor)
for name in visitor.read - visitor.written:
self._read_variable(name, node.lineno)
def _Name(self, node):
if isinstance(node.ctx, (ast.Store, ast.AugStore)):
self._written_variable(node.id, node.lineno)
if not isinstance(node.ctx, ast.Store):
self._read_variable(node.id, node.lineno)
def _Assign(self, node):
ast.walk(node.value, self)
for child in node.targets:
ast.walk(child, self)
def _ClassDef(self, node):
self._written_variable(node.name, node.lineno)
def _handle_conditional_node(self, node):
self.conditional = True
try:
for child in ast.get_child_nodes(node):
ast.walk(child, self)
finally:
self.conditional = False
def _If(self, node):
self._handle_conditional_node(node)
def _While(self, node):
self._handle_conditional_node(node)
def _For(self, node):
self.conditional = True
try:
# iter has to be checked before the target variables
ast.walk(node.iter, self)
ast.walk(node.target, self)
for child in node.body:
ast.walk(child, self)
for child in node.orelse:
ast.walk(child, self)
finally:
self.conditional = False
