If available, use wl-paste for grabclipboard() on Linux

python-pillow · Dec 7, 2022 · ccac854 · ccac854
1 parent 2ea9497
commit ccac854
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/Tests/test_imagegrab.py b/Tests/test_imagegrab.py
@@ -64,9 +64,13 @@ def test_grabclipboard(self):
             )
             p.communicate()
         else:
-            with pytest.raises(NotImplementedError) as e:
-                ImageGrab.grabclipboard()
-            assert str(e.value) == "ImageGrab.grabclipboard() is macOS and Windows only"
+            if not shutil.which("wl-paste"):
+                with pytest.raises(NotImplementedError) as e:
+                    ImageGrab.grabclipboard()
+                assert (
+                    str(e.value)
+                    == "wl-paste is required for ImageGrab.grabclipboard() on Linux"
+                )
             return
 
         ImageGrab.grabclipboard()

diff --git a/src/PIL/ImageGrab.py b/src/PIL/ImageGrab.py
@@ -132,4 +132,14 @@ def grabclipboard():
                 return BmpImagePlugin.DibImageFile(data)
         return None
     else:
-        raise NotImplementedError("ImageGrab.grabclipboard() is macOS and Windows only")
+        if not shutil.which("wl-paste"):
+            raise NotImplementedError(
+                "wl-paste is required for ImageGrab.grabclipboard() on Linux"
+            )
+        fh, filepath = tempfile.mkstemp()
+        subprocess.call(["wl-paste"], stdout=fh)
+        os.close(fh)
+        im = Image.open(filepath)
+        im.load()
+        os.unlink(filepath)
+        return im