##// END OF EJS Templates
Address @ccordoba12's comments
Emilio Graff -
Show More
@@ -1,132 +1,134 b''
1 1 import importlib
2 2 import os
3 3
4 4 aliases = {
5 5 'qt4': 'qt',
6 6 'gtk2': 'gtk',
7 7 }
8 8
9 9 backends = [
10 10 "qt",
11 11 "qt5",
12 12 "qt6",
13 13 "gtk",
14 14 "gtk2",
15 15 "gtk3",
16 16 "gtk4",
17 17 "tk",
18 18 "wx",
19 19 "pyglet",
20 20 "glut",
21 21 "osx",
22 22 "asyncio",
23 23 ]
24 24
25 25 registered = {}
26 26
27 27 def register(name, inputhook):
28 28 """Register the function *inputhook* as an event loop integration."""
29 29 registered[name] = inputhook
30 30
31 31
32 32 class UnknownBackend(KeyError):
33 33 def __init__(self, name):
34 34 self.name = name
35 35
36 36 def __str__(self):
37 37 return ("No event loop integration for {!r}. "
38 38 "Supported event loops are: {}").format(self.name,
39 39 ', '.join(backends + sorted(registered)))
40 40
41 41
42 42 def set_qt_api(gui):
43 43 """Sets the `QT_API` environment variable if it isn't already set."""
44 44
45 45 qt_api = os.environ.get("QT_API", None)
46 46
47 47 from IPython.external.qt_loaders import (
48 48 QT_API_PYQT,
49 49 QT_API_PYQT5,
50 50 QT_API_PYQT6,
51 51 QT_API_PYSIDE,
52 52 QT_API_PYSIDE2,
53 53 QT_API_PYSIDE6,
54 54 QT_API_PYQTv1,
55 55 loaded_api,
56 56 )
57 57
58 58 loaded = loaded_api()
59 59
60 60 qt_env2gui = {
61 61 QT_API_PYSIDE: "qt4",
62 62 QT_API_PYQTv1: "qt4",
63 63 QT_API_PYQT: "qt4",
64 64 QT_API_PYSIDE2: "qt5",
65 65 QT_API_PYQT5: "qt5",
66 66 QT_API_PYSIDE6: "qt6",
67 67 QT_API_PYQT6: "qt6",
68 68 }
69 69 if loaded is not None and gui != "qt":
70 70 if qt_env2gui[loaded] != gui:
71 raise ImportError(
71 print(
72 72 f"Cannot switch Qt versions for this session; must use {qt_env2gui[loaded]}."
73 73 )
74 return
74 75
75 76 if qt_api is not None and gui != "qt":
76 77 if qt_env2gui[qt_api] != gui:
77 78 print(
78 79 f'Request for "{gui}" will be ignored because `QT_API` '
79 80 f'environment variable is set to "{qt_api}"'
80 81 )
81 82 else:
82 83 if gui == "qt5":
83 84 try:
84 85 import PyQt5 # noqa
85 86
86 87 os.environ["QT_API"] = "pyqt5"
87 88 except ImportError:
88 89 try:
89 90 import PySide2 # noqa
90 91
91 92 os.environ["QT_API"] = "pyside2"
92 93 except ImportError:
93 94 os.environ["QT_API"] = "pyqt5"
94 95 elif gui == "qt6":
95 96 try:
96 97 import PyQt6 # noqa
97 98
98 99 os.environ["QT_API"] = "pyqt6"
99 100 except ImportError:
100 101 try:
101 102 import PySide6 # noqa
102 103
103 104 os.environ["QT_API"] = "pyside6"
104 105 except ImportError:
105 106 os.environ["QT_API"] = "pyqt6"
106 107 elif gui == "qt":
107 108 # Don't set QT_API; let IPython logic choose the version.
108 109 if "QT_API" in os.environ.keys():
109 110 del os.environ["QT_API"]
110 111 else:
111 raise ValueError(
112 print(
112 113 f'Unrecognized Qt version: {gui}. Should be "qt5", "qt6", or "qt".'
113 114 )
115 return
114 116
115 117
116 118 def get_inputhook_name_and_func(gui):
117 119 if gui in registered:
118 120 return gui, registered[gui]
119 121
120 122 if gui not in backends:
121 123 raise UnknownBackend(gui)
122 124
123 125 if gui in aliases:
124 126 return get_inputhook_name_and_func(aliases[gui])
125 127
126 128 gui_mod = gui
127 129 if gui.startswith("qt"):
128 130 set_qt_api(gui)
129 131 gui_mod = "qt"
130 132
131 133 mod = importlib.import_module("IPython.terminal.pt_inputhooks." + gui_mod)
132 134 return gui, mod.inputhook
General Comments 0
You need to be logged in to leave comments. Login now