diff options
| -rw-r--r-- | .pre-commit-config.yaml | 2 | ||||
| -rw-r--r-- | astroid/__init__.py | 36 | ||||
| -rw-r--r-- | astroid/nodes.py | 85 |
3 files changed, 95 insertions, 28 deletions
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index af73db8e..5021833f 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -30,7 +30,7 @@ repos: rev: 5.8.0 hooks: - id: isort - exclude: tests/testdata|astroid/__init__.py + exclude: tests/testdata - repo: https://github.com/Pierre-Sassoulas/black-disable-checker/ rev: 1.0.1 hooks: diff --git a/astroid/__init__.py b/astroid/__init__.py index 1b922bfe..f0c1c3b6 100644 --- a/astroid/__init__.py +++ b/astroid/__init__.py @@ -39,47 +39,29 @@ Main modules are: * builder contains the class responsible to build astroid trees """ - import os from importlib import import_module from pathlib import Path - -from .__pkginfo__ import __version__, version - - -# WARNING: internal imports order matters ! -# pylint: disable=wrong-import-order,wrong-import-position,redefined-builtin - -# make all exception classes accessible from astroid package +from astroid import inference, raw_building +from astroid.bases import BaseInstance, BoundMethod, Instance, UnboundMethod +from astroid.brain.helpers import register_module_extender +from astroid.builder import extract_node, parse +from astroid.const import Context, Del, Load, Store +from astroid.context import * 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.const import Context, Load, Store, Del from astroid.inference_tip import _inference_tip_cached, inference_tip -from astroid.bases import BaseInstance, Instance, BoundMethod, UnboundMethod +from astroid.manager import AstroidManager from astroid.node_classes import are_exclusive, unpack_infer +from astroid.nodes import * # pylint: disable=redefined-builtin (Ellipsis) from astroid.scoped_nodes import builtin_lookup -from astroid.builder import parse, extract_node from astroid.util import Uninferable -from astroid.brain.helpers import register_module_extender - -# make a manager instance (borg) accessible from astroid package -from astroid.manager import AstroidManager +from .__pkginfo__ import __version__, version MANAGER = AstroidManager() del AstroidManager - # load brain plugins BRAIN_MODULES_DIR = Path(__file__).with_name("brain") for module in os.listdir(BRAIN_MODULES_DIR): diff --git a/astroid/nodes.py b/astroid/nodes.py index cfe45e47..a3ce7b4d 100644 --- a/astroid/nodes.py +++ b/astroid/nodes.py @@ -23,6 +23,91 @@ All nodes inherit from :class:`~astroid.node_classes.NodeNG`. # pylint: disable=redefined-builtin # Nodes not present in the builtin ast module: DictUnpack, Unknown, and EvaluatedObject. + +__all__ = ( + "AsyncFunctionDef", + "AsyncFor", + "AsyncWith", + "Await", + "Arguments", + "AssignAttr", + "Assert", + "Assign", + "AnnAssign", + "AssignName", + "AugAssign", + "BinOp", + "BoolOp", + "Break", + "Call", + "ClassDef", + "Compare", + "Comprehension", + "Const", + "const_factory", + "Continue", + "Decorators", + "DelAttr", + "DelName", + "Delete", + "Dict", + "DictComp", + "DictUnpack", + "Expr", + "Ellipsis", + "EmptyNode", + "EvaluatedObject", + "ExceptHandler", + "ExtSlice", + "For", + "ImportFrom", + "FunctionDef", + "Attribute", + "GeneratorExp", + "Global", + "If", + "IfExp", + "Import", + "Index", + "Keyword", + "Lambda", + "List", + "ListComp", + "Match", + "MatchAs", + "MatchCase", + "MatchClass", + "MatchMapping", + "MatchOr", + "MatchSequence", + "MatchSingleton", + "MatchStar", + "MatchValue", + "Name", + "NamedExpr", + "Nonlocal", + "Module", + "Pass", + "Raise", + "Return", + "Set", + "SetComp", + "Slice", + "Starred", + "Subscript", + "TryExcept", + "TryFinally", + "Tuple", + "UnaryOp", + "Unknown", + "While", + "With", + "Yield", + "YieldFrom", + "FormattedValue", + "JoinedStr", +) + from astroid.node_classes import ( AnnAssign, Arguments, |