hotfix for screenshot().save(s.jpg) error

openatx · May 25, 2021 · 26f21e1 · 26f21e1
1 parent 9585e2e
commit 26f21e1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/wda/__init__.py b/wda/__init__.py
@@ -562,7 +562,8 @@ def screenshot(self, png_filename=None, format='pillow'):
         elif format == 'pillow':
             from PIL import Image
             buff = io.BytesIO(raw_value)
-            return Image.open(buff)
+            im = Image.open(buff)
+            return im.convert("RGB") # convert to RGB to fix save jpeg error
         else:
             raise ValueError("unknown format")