90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
|
# 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
|
||
|
|
||
|
"""Astroid hooks for the PyQT library."""
|
||
|
|
||
|
from astroid import nodes, parse
|
||
|
from astroid.brain.helpers import register_module_extender
|
||
|
from astroid.builder import AstroidBuilder
|
||
|
from astroid.manager import AstroidManager
|
||
|
|
||
|
|
||
|
def _looks_like_signal(
|
||
|
node: nodes.FunctionDef, signal_name: str = "pyqtSignal"
|
||
|
) -> bool:
|
||
|
"""Detect a Signal node."""
|
||
|
klasses = node.instance_attrs.get("__class__", [])
|
||
|
# On PySide2 or PySide6 (since Qt 5.15.2) the Signal class changed locations
|
||
|
if node.qname().partition(".")[0] in {"PySide2", "PySide6"}:
|
||
|
return any(cls.qname() == "Signal" for cls in klasses) # pragma: no cover
|
||
|
if klasses:
|
||
|
try:
|
||
|
return klasses[0].name == signal_name
|
||
|
except AttributeError: # pragma: no cover
|
||
|
# return False if the cls does not have a name attribute
|
||
|
pass
|
||
|
return False
|
||
|
|
||
|
|
||
|
def transform_pyqt_signal(node: nodes.FunctionDef) -> None:
|
||
|
module = parse(
|
||
|
"""
|
||
|
_UNSET = object()
|
||
|
|
||
|
class pyqtSignal(object):
|
||
|
def connect(self, slot, type=None, no_receiver_check=False):
|
||
|
pass
|
||
|
def disconnect(self, slot=_UNSET):
|
||
|
pass
|
||
|
def emit(self, *args):
|
||
|
pass
|
||
|
"""
|
||
|
)
|
||
|
signal_cls: nodes.ClassDef = module["pyqtSignal"]
|
||
|
node.instance_attrs["emit"] = [signal_cls["emit"]]
|
||
|
node.instance_attrs["disconnect"] = [signal_cls["disconnect"]]
|
||
|
node.instance_attrs["connect"] = [signal_cls["connect"]]
|
||
|
|
||
|
|
||
|
def transform_pyside_signal(node: nodes.FunctionDef) -> None:
|
||
|
module = parse(
|
||
|
"""
|
||
|
class NotPySideSignal(object):
|
||
|
def connect(self, receiver, type=None):
|
||
|
pass
|
||
|
def disconnect(self, receiver):
|
||
|
pass
|
||
|
def emit(self, *args):
|
||
|
pass
|
||
|
"""
|
||
|
)
|
||
|
signal_cls: nodes.ClassDef = module["NotPySideSignal"]
|
||
|
node.instance_attrs["connect"] = [signal_cls["connect"]]
|
||
|
node.instance_attrs["disconnect"] = [signal_cls["disconnect"]]
|
||
|
node.instance_attrs["emit"] = [signal_cls["emit"]]
|
||
|
|
||
|
|
||
|
def pyqt4_qtcore_transform():
|
||
|
return AstroidBuilder(AstroidManager()).string_build(
|
||
|
"""
|
||
|
|
||
|
def SIGNAL(signal_name): pass
|
||
|
|
||
|
class QObject(object):
|
||
|
def emit(self, signal): pass
|
||
|
"""
|
||
|
)
|
||
|
|
||
|
|
||
|
def register(manager: AstroidManager) -> None:
|
||
|
register_module_extender(manager, "PyQt4.QtCore", pyqt4_qtcore_transform)
|
||
|
manager.register_transform(
|
||
|
nodes.FunctionDef, transform_pyqt_signal, _looks_like_signal
|
||
|
)
|
||
|
manager.register_transform(
|
||
|
nodes.ClassDef,
|
||
|
transform_pyside_signal,
|
||
|
lambda node: node.qname() in {"PySide.QtCore.Signal", "PySide2.QtCore.Signal"},
|
||
|
)
|