| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- # 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
- """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
- """
- )
- register_module_extender(AstroidManager(), "PyQt4.QtCore", pyqt4_qtcore_transform)
- AstroidManager().register_transform(
- nodes.FunctionDef, transform_pyqt_signal, _looks_like_signal
- )
- AstroidManager().register_transform(
- nodes.ClassDef,
- transform_pyside_signal,
- lambda node: node.qname() in {"PySide.QtCore.Signal", "PySide2.QtCore.Signal"},
- )
|
