How to print Syntax without trailing whitespace?

[thatlittleboy]

Hi there, I'm using rich==13.5.2, python 3.10, and when trying to print syntax highlighted code in Jupyter, I'm getting trailing whitespaces which makes it inconvenient to copy, see image below:

from rich import print
from rich.syntax import Syntax

Syntax("select 123\nlimit 100", lexer="sql", theme="vs")

My desired outcome is for the trailing whitespaces to not exist.

---

I came across this issue while trying to fix this problem, #2647 , but the suggested solution there (WhitespaceStrippingConsole didn't work for me), here is my attempt anyway:

from rich.syntax import Syntax
from rich.console import Console

class WhitespaceStrippingConsole(Console):
    def _render_buffer(self, *args, **kwargs):
        rendered = super()._render_buffer(*args, **kwargs)
        to_render = []
        for line in rendered.splitlines():
            has_color_closing = line.endswith("\x1b[0m")
            line_new = line.removesuffix("\x1b[0m").rstrip()
            to_render.append(line_new)

        return "\n".join(to_render)

c = WhitespaceStrippingConsole()
c.print(Syntax("select 123\nlimit 100", lexer="sql", theme="vs"))

(I had to modify it a little because each line ending wasn't \n, it was \x1b[0m.

But it doesn't work, there are still trailing whitespaces as seen from the image above. I'm sure that the monkey-patched _render_buffer() function is actually running, and that the final to_render string does NOT have trailing whitespace, but yet when rendered in Jupyter the whitespaces still exist.

Can anyone provide any guidance or assistance, it would be much appreciated. Thanks!