use export_to_png() for screnshots

Author: b3b

pythonhere/magic_here/shortcuts.py

@@ -2,9 +2,10 @@
 # pylint: disable=invalid-name
 
 from base64 import b64decode
-from io import StringIO
+from io import BytesIO, StringIO
 
-from IPython.display import Image, display
+from PIL import Image as PILImage
+from IPython.display import display
 import click
 from herethere.there.commands import there_group, there_code_shortcut
 
@@ -65,4 +66,9 @@ def screenshot(ctx, width, output):
         output.write(data)
         output.close()
 
-    display(Image(data=data, width=width))
+    img = PILImage.open(BytesIO(data)).convert("RGB")
+    if width:
+        height = int(width * img.size[1] // img.size[0])
+        img = img.resize((width, height), PILImage.ANTIALIAS)
+
+    display(img)

pythonhere/window_here.py

@@ -1,6 +1,7 @@
 """Utilities for working with Kivy window."""
 from base64 import b64encode
 import os
+from pathlib import Path
 import time
 
 from kivy.app import App
@@ -49,7 +50,8 @@ def load_kv_string(code: str, clear_style: bool):
 
 def encoded_screenshot() -> str:
     """Return base64 encoded displayed image."""
-    path = Window.screenshot("screenshot_here.png")
+    path = str(Path(f"screenshot_{time.time()}.png").resolve())
+    Window.children[0].export_to_png(path)
     with open(path, "rb") as png_file:
         data = b64encode(png_file.read()).decode()
     os.remove(path)

setup.py

@@ -29,6 +29,7 @@
         "ipython",
         "ipywidgets",
         "nest_asyncio",
+        "Pillow",
     ],
     extras_require={
         "dev": [