brain_multiprocessing.py

# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
# For details: https://github.com/PyCQA/astroid/blob/main/LICENSE
# Copyright (c) https://github.com/PyCQA/astroid/blob/main/CONTRIBUTORS.txt

from astroid.bases import BoundMethod
from astroid.brain.helpers import register_module_extender
from astroid.builder import parse
from astroid.exceptions import InferenceError
from astroid.manager import AstroidManager
from astroid.nodes.scoped_nodes import FunctionDef

def _multiprocessing_transform():
    module = parse(
        """
    from multiprocessing.managers import SyncManager
    def Manager():
        return SyncManager()
    """
    )
    # Multiprocessing uses a getattr lookup inside contexts,
    # in order to get the attributes they need. Since it's extremely
    # dynamic, we use this approach to fake it.
    node = parse(
        """
    from multiprocessing.context import DefaultContext, BaseContext
    default = DefaultContext()
    base = BaseContext()
    """
    )
    try:
        context = next(node["default"].infer())
        base = next(node["base"].infer())
    except (InferenceError, StopIteration):
        return module

for node in (context, base):
        for key, value in node.locals.items():
            if key.startswith("_"):
                continue

value = value[0]
            if isinstance(value, FunctionDef):
                # We need to rebound this, since otherwise
                # it will have an extra argument (self).
                value = BoundMethod(value, node)
            module[key] = value
    return module

def _multiprocessing_managers_transform():
    return parse(
        """
    import array
    import threading
    import multiprocessing.pool as pool
    import queue

class Namespace(object):
        pass

class Value(object):
        def __init__(self, typecode, value, lock=True):
            self._typecode = typecode
            self._value = value
        def get(self):
            return self._value
        def set(self, value):
            self._value = value
        def __repr__(self):
            return '%s(%r, %r)'%(type(self).__name__, self._typecode, self._value)
        value = property(get, set)

def Array(typecode, sequence, lock=True):
        return array.array(typecode, sequence)

class SyncManager(object):
        Queue = JoinableQueue = queue.Queue
        Event = threading.Event
        RLock = threading.RLock
        Lock = threading.Lock
        BoundedSemaphore = threading.BoundedSemaphore
        Condition = threading.Condition
        Barrier = threading.Barrier
        Pool = pool.Pool
        list = list
        dict = dict
        Value = Value
        Array = Array
        Namespace = Namespace
        __enter__ = lambda self: self
        __exit__ = lambda *args: args

def start(self, initializer=None, initargs=None):
            pass
        def shutdown(self):
            pass
    """
    )

register_module_extender(
    AstroidManager(), "multiprocessing.managers", _multiprocessing_managers_transform
)
register_module_extender(
    AstroidManager(), "multiprocessing", _multiprocessing_transform
)

docroot: /home/induvedam/public_html

curdir: /opt/cloudlinux/venv/lib/python3.11/site-packages/astroid/brain

	.	4096
	..	4096
X	__init__.py	0
	__pycache__	4096
X	brain_argparse.py	1557
X	brain_attrs.py	3038
X	brain_boto3.py	1012
X	brain_builtin_inference.py	34254
X	brain_collections.py	4416
X	brain_crypt.py	863
X	brain_ctypes.py	2660
X	brain_curses.py	3477
X	brain_dataclasses.py	22089
X	brain_dateutil.py	767
X	brain_fstrings.py	2471
X	brain_functools.py	5986
X	brain_gi.py	7543
X	brain_hashlib.py	2821
X	brain_http.py	10640
X	brain_hypothesis.py	1732
X	brain_io.py	1526
X	brain_mechanize.py	2646
X	brain_multiprocessing.py	3211
X	brain_namedtuple_enum.py	22725
X	brain_nose.py	2320
X	brain_numpy_core_einsumfunc.py	825
X	brain_numpy_core_fromnumeric.py	732
X	brain_numpy_core_function_base.py	1298
X	brain_numpy_core_multiarray.py	4238
X	brain_numpy_core_numeric.py	1629
X	brain_numpy_core_numerictypes.py	8546
X	brain_numpy_core_umath.py	4893
X	brain_numpy_ma.py	896
X	brain_numpy_ndarray.py	8998
X	brain_numpy_random_mtrand.py	3436
X	brain_numpy_utils.py	2637
X	brain_pathlib.py	1542
X	brain_pkg_resources.py	2200
X	brain_pytest.py	2223
X	brain_qt.py	2808
X	brain_random.py	2890
X	brain_re.py	2870
X	brain_regex.py	3362
X	brain_responses.py	1868
X	brain_scipy_signal.py	2276
X	brain_signal.py	3880
X	brain_six.py	7594
X	brain_sqlalchemy.py	1009
X	brain_ssl.py	6554
X	brain_subprocess.py	2996
X	brain_threading.py	870
X	brain_type.py	2483
X	brain_typing.py	14315
X	brain_unittest.py	1147
X	brain_uuid.py	667
X	helpers.py	908