Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/astroid/brain/brain_qt.py: 55%
29 statements
« prev ^ index » next coverage.py v7.2.7, created at 2023-06-07 06:53 +0000
« prev ^ index » next coverage.py v7.2.7, created at 2023-06-07 06:53 +0000
1# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
2# For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE
3# Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt
5"""Astroid hooks for the PyQT library."""
7from astroid import nodes, parse
8from astroid.brain.helpers import register_module_extender
9from astroid.builder import AstroidBuilder
10from astroid.manager import AstroidManager
13def _looks_like_signal(
14 node: nodes.FunctionDef, signal_name: str = "pyqtSignal"
15) -> bool:
16 """Detect a Signal node."""
17 klasses = node.instance_attrs.get("__class__", [])
18 # On PySide2 or PySide6 (since Qt 5.15.2) the Signal class changed locations
19 if node.qname().partition(".")[0] in {"PySide2", "PySide6"}:
20 return any(cls.qname() == "Signal" for cls in klasses) # pragma: no cover
21 if klasses:
22 try:
23 return klasses[0].name == signal_name
24 except AttributeError: # pragma: no cover
25 # return False if the cls does not have a name attribute
26 pass
27 return False
30def transform_pyqt_signal(node: nodes.FunctionDef) -> None:
31 module = parse(
32 """
33 _UNSET = object()
35 class pyqtSignal(object):
36 def connect(self, slot, type=None, no_receiver_check=False):
37 pass
38 def disconnect(self, slot=_UNSET):
39 pass
40 def emit(self, *args):
41 pass
42 """
43 )
44 signal_cls: nodes.ClassDef = module["pyqtSignal"]
45 node.instance_attrs["emit"] = [signal_cls["emit"]]
46 node.instance_attrs["disconnect"] = [signal_cls["disconnect"]]
47 node.instance_attrs["connect"] = [signal_cls["connect"]]
50def transform_pyside_signal(node: nodes.FunctionDef) -> None:
51 module = parse(
52 """
53 class NotPySideSignal(object):
54 def connect(self, receiver, type=None):
55 pass
56 def disconnect(self, receiver):
57 pass
58 def emit(self, *args):
59 pass
60 """
61 )
62 signal_cls: nodes.ClassDef = module["NotPySideSignal"]
63 node.instance_attrs["connect"] = [signal_cls["connect"]]
64 node.instance_attrs["disconnect"] = [signal_cls["disconnect"]]
65 node.instance_attrs["emit"] = [signal_cls["emit"]]
68def pyqt4_qtcore_transform():
69 return AstroidBuilder(AstroidManager()).string_build(
70 """
72def SIGNAL(signal_name): pass
74class QObject(object):
75 def emit(self, signal): pass
76"""
77 )
80register_module_extender(AstroidManager(), "PyQt4.QtCore", pyqt4_qtcore_transform)
81AstroidManager().register_transform(
82 nodes.FunctionDef, transform_pyqt_signal, _looks_like_signal
83)
84AstroidManager().register_transform(
85 nodes.ClassDef,
86 transform_pyside_signal,
87 lambda node: node.qname() in {"PySide.QtCore.Signal", "PySide2.QtCore.Signal"},
88)