FM / brain

/opt/imh-python/lib/python3.9/site-packages/astroid/brain/

/opt/imh-python/lib/python3.9/site-packages/astroid/brain

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
brain_argparse.py	1813	0644	edit dl rm
brain_attrs.py	3352	0644	edit dl rm
brain_boto3.py	1072	0644	edit dl rm
brain_builtin_inference.py	37433	0644	edit dl rm
brain_collections.py	4787	0644	edit dl rm
brain_crypt.py	915	0644	edit dl rm
brain_ctypes.py	2720	0644	edit dl rm
brain_curses.py	3529	0644	edit dl rm
brain_dataclasses.py	22363	0644	edit dl rm
brain_datetime.py	771	0644	edit dl rm
brain_dateutil.py	819	0644	edit dl rm
brain_functools.py	6390	0644	edit dl rm
brain_gi.py	7598	0644	edit dl rm
brain_hashlib.py	2757	0644	edit dl rm
brain_http.py	10687	0644	edit dl rm
brain_hypothesis.py	1800	0644	edit dl rm
brain_io.py	1589	0644	edit dl rm
brain_mechanize.py	2698	0644	edit dl rm
brain_multiprocessing.py	3260	0644	edit dl rm
brain_namedtuple_enum.py	23912	0644	edit dl rm
brain_nose.py	2402	0644	edit dl rm
brain_numpy_core_einsumfunc.py	885	0644	edit dl rm
brain_numpy_core_fromnumeric.py	792	0644	edit dl rm
brain_numpy_core_function_base.py	1408	0644	edit dl rm
brain_numpy_core_multiarray.py	4400	0644	edit dl rm
brain_numpy_core_numeric.py	1740	0644	edit dl rm
brain_numpy_core_numerictypes.py	8606	0644	edit dl rm
brain_numpy_core_umath.py	4939	0644	edit dl rm
brain_numpy_ma.py	948	0644	edit dl rm
brain_numpy_ndarray.py	9066	0644	edit dl rm
brain_numpy_random_mtrand.py	3496	0644	edit dl rm
brain_numpy_utils.py	2553	0644	edit dl rm
brain_pathlib.py	1705	0644	edit dl rm
brain_pkg_resources.py	2252	0644	edit dl rm
brain_pytest.py	2270	0644	edit dl rm
brain_qt.py	2874	0644	edit dl rm
brain_random.py	3170	0644	edit dl rm
brain_re.py	2966	0644	edit dl rm
brain_regex.py	3433	0644	edit dl rm
brain_responses.py	1920	0644	edit dl rm
brain_scipy_signal.py	2328	0644	edit dl rm
brain_signal.py	3932	0644	edit dl rm
brain_six.py	7666	0644	edit dl rm
brain_sqlalchemy.py	1061	0644	edit dl rm
brain_ssl.py	6716	0644	edit dl rm
brain_subprocess.py	2971	0644	edit dl rm
brain_threading.py	922	0644	edit dl rm
brain_type.py	2481	0644	edit dl rm
brain_typing.py	16717	0644	edit dl rm
brain_unittest.py	1136	0644	edit dl rm
brain_uuid.py	727	0644	edit dl rm
helpers.py	4303	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /opt/imh-python/lib/python3.9/site-packages/astroid/brain/brain_numpy_utils.py (2553B)

"""Different utilities for the numpy brains."""

from __future__ import annotations

from astroid.builder import extract_node
from astroid.context import InferenceContext
from astroid.nodes.node_classes import Attribute, Import, Name

# Class subscript is available in numpy starting with version 1.20.0
NUMPY_VERSION_TYPE_HINTS_SUPPORT = ("1", "20", "0")

def numpy_supports_type_hints() -> bool:
    """Returns True if numpy supports type hints."""
    np_ver = _get_numpy_version()
    return np_ver and np_ver > NUMPY_VERSION_TYPE_HINTS_SUPPORT

def _get_numpy_version() -> tuple[str, str, str]:
    """
    Return the numpy version number if numpy can be imported.

Otherwise returns ('0', '0', '0')
    """
    try:
        import numpy  # pylint: disable=import-outside-toplevel

return tuple(numpy.version.version.split("."))
    except (ImportError, AttributeError):
        return ("0", "0", "0")

def infer_numpy_member(src, node, context: InferenceContext | None = None):
    node = extract_node(src)
    return node.infer(context=context)

def _is_a_numpy_module(node: Name) -> bool:
    """
    Returns True if the node is a representation of a numpy module.

For example in :
        import numpy as np
        x = np.linspace(1, 2)
    The node <Name.np> is a representation of the numpy module.

:param node: node to test
    :return: True if the node is a representation of the numpy module.
    """
    module_nickname = node.name
    potential_import_target = [
        x for x in node.lookup(module_nickname)[1] if isinstance(x, Import)
    ]
    return any(
        ("numpy", module_nickname) in target.names or ("numpy", None) in target.names
        for target in potential_import_target
    )

def name_looks_like_numpy_member(member_name: str, node: Name) -> bool:
    """
    Returns True if the Name is a member of numpy whose
    name is member_name.
    """
    return node.name == member_name and node.root().name.startswith("numpy")

def attribute_looks_like_numpy_member(member_name: str, node: Attribute) -> bool:
    """
    Returns True if the Attribute is a member of numpy whose
    name is member_name.
    """
    return (
        node.attrname == member_name
        and isinstance(node.expr, Name)
        and _is_a_numpy_module(node.expr)
    )

# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html # For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE # Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt """Different utilities for the numpy brains.""" from __future__ import annotations from astroid.builder import extract_node from astroid.context import InferenceContext from astroid.nodes.node_classes import Attribute, Import, Name # Class subscript is available in numpy starting with version 1.20.0 NUMPY_VERSION_TYPE_HINTS_SUPPORT = ("1", "20", "0") def numpy_supports_type_hints() -> bool: """Returns True if numpy supports type hints.""" np_ver = _get_numpy_version() return np_ver and np_ver > NUMPY_VERSION_TYPE_HINTS_SUPPORT def _get_numpy_version() -> tuple[str, str, str]: """ Return the numpy version number if numpy can be imported. Otherwise returns ('0', '0', '0') """ try: import numpy # pylint: disable=import-outside-toplevel return tuple(numpy.version.version.split(".")) except (ImportError, AttributeError): return ("0", "0", "0") def infer_numpy_member(src, node, context: InferenceContext | None = None): node = extract_node(src) return node.infer(context=context) def _is_a_numpy_module(node: Name) -> bool: """ Returns True if the node is a representation of a numpy module. For example in : import numpy as np x = np.linspace(1, 2) The node is a representation of the numpy module. :param node: node to test :return: True if the node is a representation of the numpy module. """ module_nickname = node.name potential_import_target = [ x for x in node.lookup(module_nickname)[1] if isinstance(x, Import) ] return any( ("numpy", module_nickname) in target.names or ("numpy", None) in target.names for target in potential_import_target ) def name_looks_like_numpy_member(member_name: str, node: Name) -> bool: """ Returns True if the Name is a member of numpy whose name is member_name. """ return node.name == member_name and node.root().name.startswith("numpy") def attribute_looks_like_numpy_member(member_name: str, node: Attribute) -> bool: """ Returns True if the Attribute is a member of numpy whose name is member_name. """ return ( node.attrname == member_name and isinstance(node.expr, Name) and _is_a_numpy_module(node.expr) )