Skip to content

Commit 05318b1

Browse files
iloveitalyanntzer
authored andcommitted
fix: support for new ipython 9.x theme system
1 parent fb83a21 commit 05318b1

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

src/ipython_autoimport.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
argument, magic_arguments, parse_argstring)
2424
from IPython.core.magics.execution import ExecutionMagics
2525
from IPython.utils import PyColorize
26+
import pygments
2627

2728
try:
2829
import importlib.metadata as _im
@@ -89,9 +90,15 @@ def _report(ipython, msg):
8990
sys.stdout._raw = True
9091
except AttributeError:
9192
pass
92-
cs = PyColorize.Parser().color_table[ipython.colors].colors
93-
# Token.NUMBER: bright blue (cyan), looks reasonable.
94-
print("{}Autoimport:{} {}".format(cs[token.NUMBER], cs["normal"], msg))
93+
if IPython.version_info >= (9,):
94+
print(PyColorize.theme_table[ipython.colors].format([
95+
(pygments.token.Number, "Autoimport: "),
96+
(pygments.token.Text, msg),
97+
]))
98+
else:
99+
cs = PyColorize.Parser().color_table[ipython.colors].colors
100+
# Token.NUMBER: bright blue (cyan), looks reasonable.
101+
print("{}Autoimport:{} {}".format(cs[token.NUMBER], cs["normal"], msg))
95102

96103

97104
class _SubmoduleAutoImporterModule(ModuleType):

0 commit comments

Comments
 (0)