You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
"""Astroid hooks for the PyQT library."""
|
|
|
|
from astroid import MANAGER, register_module_extender
|
|
from astroid.builder import AstroidBuilder
|
|
from astroid import nodes
|
|
from astroid import parse
|
|
|
|
|
|
def _looks_like_signal(node, signal_name='pyqtSignal'):
|
|
if '__class__' in node._instance_attrs:
|
|
cls = node._instance_attrs['__class__'][0]
|
|
return cls.name == signal_name
|
|
return False
|
|
|
|
|
|
def transform_pyqt_signal(node):
|
|
module = parse('''
|
|
class pyqtSignal(object):
|
|
def connect(self, slot, type=None, no_receiver_check=False):
|
|
pass
|
|
def disconnect(self, slot):
|
|
pass
|
|
def emit(self, *args):
|
|
pass
|
|
''')
|
|
signal_cls = module['pyqtSignal']
|
|
node._instance_attrs['emit'] = signal_cls['emit']
|
|
node._instance_attrs['disconnect'] = signal_cls['disconnect']
|
|
node._instance_attrs['connect'] = signal_cls['connect']
|
|
|
|
|
|
def pyqt4_qtcore_transform():
|
|
return AstroidBuilder(MANAGER).string_build('''
|
|
|
|
def SIGNAL(signal_name): pass
|
|
|
|
class QObject(object):
|
|
def emit(self, signal): pass
|
|
''')
|
|
|
|
|
|
register_module_extender(MANAGER, 'PyQt4.QtCore', pyqt4_qtcore_transform)
|
|
MANAGER.register_transform(nodes.FunctionDef, transform_pyqt_signal,
|
|
_looks_like_signal) |