Merge pull request #1 from radarhere/image_grab_wayland_kde

AdianKozlica · web-flow · commit 2f672935eb35 · 2025-03-29T23:44:12.000+01:00
Do not create temporary file if no utility is available
diff --git a/src/PIL/ImageGrab.py b/src/PIL/ImageGrab.py
@@ -78,15 +78,15 @@ def grab(
         size, data = Image.core.grabscreen_x11(display_name)
     except OSError:
         if display_name is None and sys.platform not in ("darwin", "win32"):
-            fh, filepath = tempfile.mkstemp(".png")
-            os.close(fh)
             if shutil.which("gnome-screenshot"):
-                subprocess.call(["gnome-screenshot", "-f", filepath])
+                args = ["gnome-screenshot", "-f"]
             elif shutil.which("spectacle"):
-                subprocess.call(["spectacle", "-n", "-b", "-f", "-o", filepath])
+                args = ["spectacle", "-n", "-b", "-f", "-o"]
             else:
-                os.unlink(filepath)
                 raise
+            fh, filepath = tempfile.mkstemp(".png")
+            os.close(fh)
+            subprocess.call(args + [filepath])
             im = Image.open(filepath)
             im.load()
             os.unlink(filepath)
