python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-08-23 05:34:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Raise error if unable to open X server

Browse Source
This commit is contained in:
Andrew Murray 2020-03-24 20:20:08 +11:00
parent 13ab784c51
commit b182621c59
2 changed files with 23 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
Tests/test_imagegrab.py
View File

@ -2,24 +2,29 @@ import subprocess
import sys
import pytest
from PIL import ImageGrab
from .helper import assert_image
from PIL import ImageGrab
class TestImageGrab:
def test_grab(self):
if sys.platform in ["darwin", "win32"] or ImageGrab._has_imagemagick():
for im in [
ImageGrab.grab(),
ImageGrab.grab(include_layered_windows=True),
ImageGrab.grab(all_screens=True),
native_support = sys.platform in ["darwin", "win32"]
if native_support or ImageGrab._has_imagemagick():
for args in [
{},
{"include_layered_windows": True},
{"all_screens": True},
{"bbox": (10, 20, 50, 80)},
]:
assert_image(im, im.mode, im.size)
im = ImageGrab.grab(bbox=(10, 20, 50, 80))
assert_image(im, im.mode, (40, 60))
try:
im = ImageGrab.grab(**args)
except IOError as e:
if not native_support and str(e) == "Unable to open X server":
continue
else:
raise
assert_image(im, im.mode, (40, 60) if "bbox" in args else im.size)
else:
pytest.raises(IOError, ImageGrab.grab)
@ -36,7 +41,7 @@ $bmp = New-Object Drawing.Bitmap 200, 200
)
p.communicate()
else:
self.assertRaises(NotImplementedError, ImageGrab.grabclipboard)
pytest.raises(NotImplementedError, ImageGrab.grabclipboard)
return
im = ImageGrab.grabclipboard()

7
src/PIL/ImageGrab.py
View File

@ -68,7 +68,12 @@ def grab(bbox=None, include_layered_windows=False, all_screens=False):
raise IOError("grab requires ImageMagick unless used on macOS or Windows")
fh, filepath = tempfile.mkstemp(".png")
os.close(fh)
subprocess.call(["import", "-window", "root", filepath])
p = subprocess.Popen(
["import", "-window", "root", filepath], stderr=subprocess.PIPE
)
if b"unable to open X server" in p.communicate()[1]:
raise IOError("Unable to open X server")
im = Image.open(filepath)
im.load()
os.unlink(filepath)