pythonVSCode/pythonFiles/rope/base/astutils.py at master · meflyup/pythonVSCode · GitHub

History

64 lines (47 loc) · 1.61 KB

Raw

from rope.base import ast

def get_name_levels(node):

"""Return a list of ``(name, level)`` tuples for assigned names

The `level` is `None` for simple assignments and is a list of

numbers for tuple assignments for example in::

a, (b, c) = x

The levels for for `a` is ``[0]``, for `b` is ``[1, 0]`` and for

`c` is ``[1, 1]``.

"""

visitor = _NodeNameCollector()

ast.walk(node, visitor)

return visitor.names

class _NodeNameCollector(object):

def __init__(self, levels=None):

self.names = []

self.levels = levels

self.index = 0

def _add_node(self, node):

new_levels = []

if self.levels is not None:

new_levels = list(self.levels)

new_levels.append(self.index)

self.index += 1

self._added(node, new_levels)

def _added(self, node, levels):

if hasattr(node, 'id'):

self.names.append((node.id, levels))

def _Name(self, node):

self._add_node(node)

def _ExceptHandler(self, node):

self.names.append((node.name, []))

def _Tuple(self, node):

new_levels = []

if self.levels is not None:

new_levels = list(self.levels)

new_levels.append(self.index)

self.index += 1

visitor = _NodeNameCollector(new_levels)

for child in ast.get_child_nodes(node):

ast.walk(child, visitor)

self.names.extend(visitor.names)

def _Subscript(self, node):

self._add_node(node)

def _Attribute(self, node):

self._add_node(node)

def _Slice(self, node):

self._add_node(node)

pythonVSCode/pythonFiles/rope/base/astutils.py at master · meflyup/pythonVSCode · GitHub

Expand file tree

Breadcrumbs

astutils.py

Latest commit

History

Breadcrumbs

astutils.py

File metadata and controls