AlkantarClanX12

Your IP : 18.223.237.246

Current Path : /opt/cloudlinux/venv/lib64/python3.11/site-packages/pylint_celery/
Upload File :
Current File : //opt/cloudlinux/venv/lib64/python3.11/site-packages/pylint_celery/__init__.py

from astroid import MANAGER
from astroid.builder import AstroidBuilder
from astroid import nodes


def register(_):
    # this method is expected by pylint for plugins, however we don't
    # want to register any checkers
    pass


MODULE_TRANSFORMS = {}


def transform(module):
    try:
        tr = MODULE_TRANSFORMS[module.name]
    except KeyError:
        pass
    else:
        tr(module)
MANAGER.register_transform(nodes.Module, transform)


def celery_transform(module):
    fake = AstroidBuilder(MANAGER).string_build('''
class task_dummy(object):
    def __call__(self):
        pass
''')
    module.locals['task'] = fake.locals['task_dummy']


MODULE_TRANSFORMS['celery'] = celery_transform