| Index: third_party/logilab/astroid/__init__.py
|
| ===================================================================
|
| --- third_party/logilab/astroid/__init__.py (revision 0)
|
| +++ third_party/logilab/astroid/__init__.py (working copy)
|
| @@ -0,0 +1,118 @@
|
| +# copyright 2003-2013 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
|
| +# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
|
| +#
|
| +# This file is part of astroid.
|
| +#
|
| +# astroid is free software: you can redistribute it and/or modify it
|
| +# under the terms of the GNU Lesser General Public License as published by the
|
| +# Free Software Foundation, either version 2.1 of the License, or (at your
|
| +# option) any later version.
|
| +#
|
| +# astroid is distributed in the hope that it will be useful, but
|
| +# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
| +# FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
|
| +# for more details.
|
| +#
|
| +# You should have received a copy of the GNU Lesser General Public License along
|
| +# with astroid. If not, see <http://www.gnu.org/licenses/>.
|
| +"""Python Abstract Syntax Tree New Generation
|
| +
|
| +The aim of this module is to provide a common base representation of
|
| +python source code for projects such as pychecker, pyreverse,
|
| +pylint... Well, actually the development of this library is essentially
|
| +governed by pylint's needs.
|
| +
|
| +It extends class defined in the python's _ast module with some
|
| +additional methods and attributes. Instance attributes are added by a
|
| +builder object, which can either generate extended ast (let's call
|
| +them astroid ;) by visiting an existent ast tree or by inspecting living
|
| +object. Methods are added by monkey patching ast classes.
|
| +
|
| +Main modules are:
|
| +
|
| +* nodes and scoped_nodes for more information about methods and
|
| + attributes added to different node classes
|
| +
|
| +* the manager contains a high level object to get astroid trees from
|
| + source files and living objects. It maintains a cache of previously
|
| + constructed tree for quick access
|
| +
|
| +* builder contains the class responsible to build astroid trees
|
| +"""
|
| +__doctype__ = "restructuredtext en"
|
| +
|
| +import sys
|
| +import re
|
| +from operator import attrgetter
|
| +
|
| +# WARNING: internal imports order matters !
|
| +
|
| +# make all exception classes accessible from astroid package
|
| +from astroid.exceptions import *
|
| +
|
| +# make all node classes accessible from astroid package
|
| +from astroid.nodes import *
|
| +
|
| +# trigger extra monkey-patching
|
| +from astroid import inference
|
| +
|
| +# more stuff available
|
| +from astroid import raw_building
|
| +from astroid.bases import YES, Instance, BoundMethod, UnboundMethod
|
| +from astroid.node_classes import are_exclusive, unpack_infer
|
| +from astroid.scoped_nodes import builtin_lookup
|
| +
|
| +# make a manager instance (borg) as well as Project and Package classes
|
| +# accessible from astroid package
|
| +from astroid.manager import AstroidManager, Project
|
| +MANAGER = AstroidManager()
|
| +del AstroidManager
|
| +
|
| +# transform utilities (filters and decorator)
|
| +
|
| +class AsStringRegexpPredicate(object):
|
| + """Class to be used as predicate that may be given to `register_transform`
|
| +
|
| + First argument is a regular expression that will be searched against the `as_string`
|
| + representation of the node onto which it's applied.
|
| +
|
| + If specified, the second argument is an `attrgetter` expression that will be
|
| + applied on the node first to get the actual node on which `as_string` should
|
| + be called.
|
| + """
|
| + def __init__(self, regexp, expression=None):
|
| + self.regexp = re.compile(regexp)
|
| + self.expression = expression
|
| +
|
| + def __call__(self, node):
|
| + if self.expression is not None:
|
| + node = attrgetter(self.expression)(node)
|
| + return self.regexp.search(node.as_string())
|
| +
|
| +def inference_tip(infer_function):
|
| + """Given an instance specific inference function, return a function to be
|
| + given to MANAGER.register_transform to set this inference function.
|
| +
|
| + Typical usage
|
| +
|
| + .. sourcecode:: python
|
| +
|
| + MANAGER.register_transform(CallFunc, inference_tip(infer_named_tuple),
|
| + AsStringRegexpPredicate('namedtuple', 'func'))
|
| + """
|
| + def transform(node, infer_function=infer_function):
|
| + node._explicit_inference = infer_function
|
| + return node
|
| + return transform
|
| +
|
| +# load brain plugins
|
| +from os import listdir
|
| +from os.path import join, dirname
|
| +BRAIN_MODULES_DIR = join(dirname(__file__), 'brain')
|
| +if BRAIN_MODULES_DIR not in sys.path:
|
| + # add it to the end of the list so user path take precedence
|
| + sys.path.append(BRAIN_MODULES_DIR)
|
| +# load modules in this directory
|
| +for module in listdir(BRAIN_MODULES_DIR):
|
| + if module.endswith('.py'):
|
| + __import__(module[:-3])
|
|
|