fix: support for new ipython 9.x theme system

Author: iloveitaly

README.rst

@@ -101,6 +101,10 @@ by adding ``c.Completer.use_jedi = False`` to the ``ipython_config.py`` file.
 Changelog
 ---------
 
+next
+~~~~
+- Fix compatibility with IPython 9's new theme system.
+
 v0.5
 ~~~~
 - Avoid erroring when exiting IPython≥8.15.

src/ipython_autoimport.py

@@ -89,9 +89,16 @@ def _report(ipython, msg):
         sys.stdout._raw = True
     except AttributeError:
         pass
-    cs = PyColorize.Parser().color_table[ipython.colors].colors
-    # Token.NUMBER: bright blue (cyan), looks reasonable.
-    print("{}Autoimport:{} {}".format(cs[token.NUMBER], cs["normal"], msg))
+    if IPython.version_info >= (9,):
+        import pygments  # Only a dependency from IPython 4.2.0.
+        print(PyColorize.theme_table[ipython.colors].format([
+            (pygments.token.Number, "Autoimport: "),
+            (pygments.token.Text, msg),
+        ]))
+    else:
+        cs = PyColorize.Parser().color_table[ipython.colors].colors
+        # Token.NUMBER: bright blue (cyan), looks reasonable.
+        print("{}Autoimport:{} {}".format(cs[token.NUMBER], cs["normal"], msg))
 
 
 class _SubmoduleAutoImporterModule(ModuleType):