| 12345678910111213141516171819202122232425262728293031323334 |
- 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
|
