Fix screenshot resizing with modern Pillow

Author: b3b

pythonhere/magic_here/shortcuts.py

@@ -66,7 +66,9 @@ def screenshot(ctx, width, output):
     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)
+        resampling = getattr(PILImage, "Resampling", PILImage)
+        resample_filter = getattr(resampling, "LANCZOS", PILImage.LANCZOS)
+        img = img.resize((width, height), resample_filter)
 
     if output:
         img.save(output)

tests/test_magic.py

@@ -39,6 +39,16 @@ async def test_screenshot_saved_to_file(tmpdir, app_instance, call_there_group):
     assert os.path.exists(output)
 
 
+@pytest.mark.asyncio
+async def test_screenshot_resized_to_width(tmpdir, app_instance, call_there_group):
+    output = Path(tmpdir) / "test.png"
+
+    call_there_group(["screenshot", "--width", "100", "-o", output], "")
+
+    image = shortcuts.PILImage.open(output)
+    assert image.size == (100, 75)
+
+
 @pytest.mark.asyncio
 async def test_pin_command_pin_shortcut_called(
     mocker, capfd, mocked_android_modules, call_there_group, test_py_script