How to install Rich into code stdlib module?

[itsthejoker]

I have a program that uses the code module to create a "repl-alike", and I'd like to use Rich in it. However, I'm not sure how to properly do this and things don't seem to be working. Here's what I have so far:

import code

from rich import pretty
from rich import print as rprint


# Note: this is basically the source code to `code.interact`, just pulled
# out so that I can have a console object to pass to `pretty.install()`.
# https://docs.python.org/3/library/code.html#code.interact

# without modifying the console object further, we get
# AttributeError: 'InteractiveConsole' object has no attribute 'print'
console = code.InteractiveConsole(globals())

# TypeError: 'crop' is an invalid keyword argument for print()
# console.print = print

# TypeError: print() got an unexpected keyword argument 'crop'
# console.print = rprint

pretty.install(console)  # expected type: console, got interactiveconsole
try:
    import readline
except ImportError:
    pass
console.interact("this is a banner", None)

This launches a REPL as expected, but trying to print anything makes it explode. Anyone have a suggestion on how to implement this properly (or even if it's possible)?

Edit: I have resorted to possibly the dumbest solution.

def print_wrapper(*args, **kwargs):
    if "crop" in kwargs:
        del kwargs["crop"]
    rprint(*args, **kwargs)

console.print = print_wrapper
pretty.install(console)

I feel like there has to be a better way to do this, but at least it works...?