| Index: third_party/logilab/astroid/brain/py2stdlib.py
|
| diff --git a/third_party/logilab/astroid/brain/py2stdlib.py b/third_party/logilab/astroid/brain/py2stdlib.py
|
| deleted file mode 100644
|
| index 2ead081b1e78b18441c25fb94258878bd65a679c..0000000000000000000000000000000000000000
|
| --- a/third_party/logilab/astroid/brain/py2stdlib.py
|
| +++ /dev/null
|
| @@ -1,331 +0,0 @@
|
| -"""Astroid hooks for the Python 2 standard library.
|
| -
|
| -Currently help understanding of :
|
| -
|
| -* hashlib.md5 and hashlib.sha1
|
| -"""
|
| -
|
| -import sys
|
| -from textwrap import dedent
|
| -
|
| -from astroid import (
|
| - MANAGER, AsStringRegexpPredicate,
|
| - UseInferenceDefault, inference_tip,
|
| - YES, InferenceError)
|
| -from astroid import exceptions
|
| -from astroid import nodes
|
| -from astroid.builder import AstroidBuilder
|
| -
|
| -MODULE_TRANSFORMS = {}
|
| -PY3K = sys.version_info > (3, 0)
|
| -PY33 = sys.version_info >= (3, 3)
|
| -
|
| -# general function
|
| -
|
| -def infer_func_form(node, base_type, context=None, enum=False):
|
| - """Specific inference function for namedtuple or Python 3 enum. """
|
| - def infer_first(node):
|
| - try:
|
| - value = node.infer(context=context).next()
|
| - if value is YES:
|
| - raise UseInferenceDefault()
|
| - else:
|
| - return value
|
| - except StopIteration:
|
| - raise InferenceError()
|
| -
|
| - # node is a CallFunc node, class name as first argument and generated class
|
| - # attributes as second argument
|
| - if len(node.args) != 2:
|
| - # something weird here, go back to class implementation
|
| - raise UseInferenceDefault()
|
| - # namedtuple or enums list of attributes can be a list of strings or a
|
| - # whitespace-separate string
|
| - try:
|
| - name = infer_first(node.args[0]).value
|
| - names = infer_first(node.args[1])
|
| - try:
|
| - attributes = names.value.replace(',', ' ').split()
|
| - except AttributeError:
|
| - if not enum:
|
| - attributes = [infer_first(const).value for const in names.elts]
|
| - else:
|
| - # Enums supports either iterator of (name, value) pairs
|
| - # or mappings.
|
| - # TODO: support only list, tuples and mappings.
|
| - if hasattr(names, 'items') and isinstance(names.items, list):
|
| - attributes = [infer_first(const[0]).value
|
| - for const in names.items
|
| - if isinstance(const[0], nodes.Const)]
|
| - elif hasattr(names, 'elts'):
|
| - # Enums can support either ["a", "b", "c"]
|
| - # or [("a", 1), ("b", 2), ...], but they can't
|
| - # be mixed.
|
| - if all(isinstance(const, nodes.Tuple)
|
| - for const in names.elts):
|
| - attributes = [infer_first(const.elts[0]).value
|
| - for const in names.elts
|
| - if isinstance(const, nodes.Tuple)]
|
| - else:
|
| - attributes = [infer_first(const).value
|
| - for const in names.elts]
|
| - else:
|
| - raise AttributeError
|
| - if not attributes:
|
| - raise AttributeError
|
| - except (AttributeError, exceptions.InferenceError) as exc:
|
| - raise UseInferenceDefault()
|
| - # we want to return a Class node instance with proper attributes set
|
| - class_node = nodes.Class(name, 'docstring')
|
| - class_node.parent = node.parent
|
| - # set base class=tuple
|
| - class_node.bases.append(base_type)
|
| - # XXX add __init__(*attributes) method
|
| - for attr in attributes:
|
| - fake_node = nodes.EmptyNode()
|
| - fake_node.parent = class_node
|
| - class_node.instance_attrs[attr] = [fake_node]
|
| - return class_node, name, attributes
|
| -
|
| -
|
| -# module specific transformation functions #####################################
|
| -
|
| -def transform(module):
|
| - try:
|
| - tr = MODULE_TRANSFORMS[module.name]
|
| - except KeyError:
|
| - pass
|
| - else:
|
| - tr(module)
|
| -MANAGER.register_transform(nodes.Module, transform)
|
| -
|
| -# module specific transformation functions #####################################
|
| -
|
| -def hashlib_transform(module):
|
| - template = '''
|
| -
|
| -class %s(object):
|
| - def __init__(self, value=''): pass
|
| - def digest(self):
|
| - return u''
|
| - def update(self, value): pass
|
| - def hexdigest(self):
|
| - return u''
|
| -'''
|
| -
|
| - algorithms = ('md5', 'sha1', 'sha224', 'sha256', 'sha384', 'sha512')
|
| - classes = "".join(template % hashfunc for hashfunc in algorithms)
|
| -
|
| - fake = AstroidBuilder(MANAGER).string_build(classes)
|
| -
|
| - for hashfunc in algorithms:
|
| - module.locals[hashfunc] = fake.locals[hashfunc]
|
| -
|
| -def collections_transform(module):
|
| - fake = AstroidBuilder(MANAGER).string_build('''
|
| -
|
| -class defaultdict(dict):
|
| - default_factory = None
|
| - def __missing__(self, key): pass
|
| -
|
| -class deque(object):
|
| - maxlen = 0
|
| - def __init__(self, iterable=None, maxlen=None): pass
|
| - def append(self, x): pass
|
| - def appendleft(self, x): pass
|
| - def clear(self): pass
|
| - def count(self, x): return 0
|
| - def extend(self, iterable): pass
|
| - def extendleft(self, iterable): pass
|
| - def pop(self): pass
|
| - def popleft(self): pass
|
| - def remove(self, value): pass
|
| - def reverse(self): pass
|
| - def rotate(self, n): pass
|
| - def __iter__(self): return self
|
| -
|
| -''')
|
| -
|
| - for klass in ('deque', 'defaultdict'):
|
| - module.locals[klass] = fake.locals[klass]
|
| -
|
| -def pkg_resources_transform(module):
|
| - fake = AstroidBuilder(MANAGER).string_build('''
|
| -
|
| -def resource_exists(package_or_requirement, resource_name):
|
| - pass
|
| -
|
| -def resource_isdir(package_or_requirement, resource_name):
|
| - pass
|
| -
|
| -def resource_filename(package_or_requirement, resource_name):
|
| - pass
|
| -
|
| -def resource_stream(package_or_requirement, resource_name):
|
| - pass
|
| -
|
| -def resource_string(package_or_requirement, resource_name):
|
| - pass
|
| -
|
| -def resource_listdir(package_or_requirement, resource_name):
|
| - pass
|
| -
|
| -def extraction_error():
|
| - pass
|
| -
|
| -def get_cache_path(archive_name, names=()):
|
| - pass
|
| -
|
| -def postprocess(tempname, filename):
|
| - pass
|
| -
|
| -def set_extraction_path(path):
|
| - pass
|
| -
|
| -def cleanup_resources(force=False):
|
| - pass
|
| -
|
| -''')
|
| -
|
| - for func_name, func in fake.locals.items():
|
| - module.locals[func_name] = func
|
| -
|
| -
|
| -def subprocess_transform(module):
|
| - if PY3K:
|
| - communicate = (bytes('string', 'ascii'), bytes('string', 'ascii'))
|
| - init = """
|
| - def __init__(self, args, bufsize=0, executable=None,
|
| - stdin=None, stdout=None, stderr=None,
|
| - preexec_fn=None, close_fds=False, shell=False,
|
| - cwd=None, env=None, universal_newlines=False,
|
| - startupinfo=None, creationflags=0, restore_signals=True,
|
| - start_new_session=False, pass_fds=()):
|
| - pass
|
| - """
|
| - else:
|
| - communicate = ('string', 'string')
|
| - init = """
|
| - def __init__(self, args, bufsize=0, executable=None,
|
| - stdin=None, stdout=None, stderr=None,
|
| - preexec_fn=None, close_fds=False, shell=False,
|
| - cwd=None, env=None, universal_newlines=False,
|
| - startupinfo=None, creationflags=0):
|
| - pass
|
| - """
|
| - if PY33:
|
| - wait_signature = 'def wait(self, timeout=None)'
|
| - else:
|
| - wait_signature = 'def wait(self)'
|
| - fake = AstroidBuilder(MANAGER).string_build('''
|
| -
|
| -class Popen(object):
|
| - returncode = pid = 0
|
| - stdin = stdout = stderr = file()
|
| -
|
| - %(init)s
|
| -
|
| - def communicate(self, input=None):
|
| - return %(communicate)r
|
| - %(wait_signature)s:
|
| - return self.returncode
|
| - def poll(self):
|
| - return self.returncode
|
| - def send_signal(self, signal):
|
| - pass
|
| - def terminate(self):
|
| - pass
|
| - def kill(self):
|
| - pass
|
| - ''' % {'init': init,
|
| - 'communicate': communicate,
|
| - 'wait_signature': wait_signature})
|
| -
|
| - for func_name, func in fake.locals.items():
|
| - module.locals[func_name] = func
|
| -
|
| -
|
| -
|
| -MODULE_TRANSFORMS['hashlib'] = hashlib_transform
|
| -MODULE_TRANSFORMS['collections'] = collections_transform
|
| -MODULE_TRANSFORMS['pkg_resources'] = pkg_resources_transform
|
| -MODULE_TRANSFORMS['subprocess'] = subprocess_transform
|
| -
|
| -# namedtuple support ###########################################################
|
| -
|
| -def infer_named_tuple(node, context=None):
|
| - """Specific inference function for namedtuple CallFunc node"""
|
| - class_node, name, attributes = infer_func_form(node, nodes.Tuple._proxied,
|
| - context=context)
|
| - fake = AstroidBuilder(MANAGER).string_build('''
|
| -class %(name)s(tuple):
|
| - _fields = %(fields)r
|
| - def _asdict(self):
|
| - return self.__dict__
|
| - @classmethod
|
| - def _make(cls, iterable, new=tuple.__new__, len=len):
|
| - return new(cls, iterable)
|
| - def _replace(_self, **kwds):
|
| - result = _self._make(map(kwds.pop, %(fields)r, _self))
|
| - if kwds:
|
| - raise ValueError('Got unexpected field names: %%r' %% list(kwds))
|
| - return result
|
| - ''' % {'name': name, 'fields': attributes})
|
| - class_node.locals['_asdict'] = fake.body[0].locals['_asdict']
|
| - class_node.locals['_make'] = fake.body[0].locals['_make']
|
| - class_node.locals['_replace'] = fake.body[0].locals['_replace']
|
| - class_node.locals['_fields'] = fake.body[0].locals['_fields']
|
| - # we use UseInferenceDefault, we can't be a generator so return an iterator
|
| - return iter([class_node])
|
| -
|
| -def infer_enum(node, context=None):
|
| - """ Specific inference function for enum CallFunc node. """
|
| - enum_meta = nodes.Class("EnumMeta", 'docstring')
|
| - class_node = infer_func_form(node, enum_meta,
|
| - context=context, enum=True)[0]
|
| - return iter([class_node.instanciate_class()])
|
| -
|
| -def infer_enum_class(node, context=None):
|
| - """ Specific inference for enums. """
|
| - names = set(('Enum', 'IntEnum', 'enum.Enum', 'enum.IntEnum'))
|
| - for basename in node.basenames:
|
| - # TODO: doesn't handle subclasses yet.
|
| - if basename not in names:
|
| - continue
|
| - if node.root().name == 'enum':
|
| - # Skip if the class is directly from enum module.
|
| - break
|
| - for local, values in node.locals.items():
|
| - if any(not isinstance(value, nodes.AssName)
|
| - for value in values):
|
| - continue
|
| - parent = values[0].parent
|
| - real_value = parent.value
|
| - new_targets = []
|
| - for target in parent.targets:
|
| - # Replace all the assignments with our mocked class.
|
| - classdef = dedent('''
|
| - class %(name)s(object):
|
| - @property
|
| - def value(self):
|
| - return %(value)s
|
| - @property
|
| - def name(self):
|
| - return %(name)r
|
| - %(name)s = %(value)s
|
| - ''' % {'name': target.name,
|
| - 'value': real_value.as_string()})
|
| - fake = AstroidBuilder(MANAGER).string_build(classdef)[target.name]
|
| - fake.parent = target.parent
|
| - for method in node.mymethods():
|
| - fake.locals[method.name] = [method]
|
| - new_targets.append(fake.instanciate_class())
|
| - node.locals[local] = new_targets
|
| - break
|
| - return node
|
| -
|
| -MANAGER.register_transform(nodes.CallFunc, inference_tip(infer_named_tuple),
|
| - AsStringRegexpPredicate('namedtuple', 'func'))
|
| -MANAGER.register_transform(nodes.CallFunc, inference_tip(infer_enum),
|
| - AsStringRegexpPredicate('Enum', 'func'))
|
| -MANAGER.register_transform(nodes.Class, infer_enum_class)
|
|
|