Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.10/site-packages/astroid/brain/brain_qt.py: 57%

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 

4 

5"""Astroid hooks for the PyQT library.""" 

6 

7from astroid import nodes 

8from astroid.brain.helpers import register_module_extender 

9from astroid.builder import AstroidBuilder, parse 

10from astroid.manager import AstroidManager 

11 

12 

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 

28 

29 

30def transform_pyqt_signal(node: nodes.FunctionDef) -> None: 

31 module = parse( 

32 """ 

33 _UNSET = object() 

34 

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"]] 

48 

49 

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"]] 

66 

67 

68def pyqt4_qtcore_transform(): 

69 return AstroidBuilder(AstroidManager()).string_build( 

70 """ 

71 

72def SIGNAL(signal_name): pass 

73 

74class QObject(object): 

75 def emit(self, signal): pass 

76""" 

77 ) 

78 

79 

80def register(manager: AstroidManager) -> None: 

81 register_module_extender(manager, "PyQt4.QtCore", pyqt4_qtcore_transform) 

82 manager.register_transform( 

83 nodes.FunctionDef, transform_pyqt_signal, _looks_like_signal 

84 ) 

85 manager.register_transform( 

86 nodes.ClassDef, 

87 transform_pyside_signal, 

88 lambda node: node.qname() in {"PySide.QtCore.Signal", "PySide2.QtCore.Signal"}, 

89 )