Skip to content

Supporting maim for screenshots on Linux #69

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
gacou54 wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Supporting maim for screenshots on Linux #69

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
31 changes: 25 additions & 6 deletions pyscreeze/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -93,6 +93,20 @@
else:
raise

maimExists = False
try:
if sys.platform not in ('java', 'darwin', 'win32'):
whichProc = subprocess.Popen(
['which', 'maim'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
maimExists = whichProc.wait() == 0
except OSError as ex:
if ex.errno == errno.ENOENT:
# if there is no "which" program to find maim, then assume there
# is no maim.
pass
else:
raise


if sys.platform == 'win32':
from ctypes import windll
Expand Down Expand Up @@ -465,21 +479,25 @@ def _screenshot_linux(imageFilename=None, region=None):
"""
TODO
"""
if not scrotExists:
raise NotImplementedError('"scrot" must be installed to use screenshot functions in Linux. Run: sudo apt-get install scrot')
if not (scrotExists or maimExists):
raise NotImplementedError('"scrot" or "maim" must be installed to use screenshot functions in Linux. Run: sudo apt-get install scrot ')
if imageFilename is None:
tmpFilename = '.screenshot%s.png' % (datetime.datetime.now().strftime('%Y-%m%d_%H-%M-%S-%f'))
else:
tmpFilename = imageFilename
if scrotExists:
subprocess.call(['scrot', '-z', tmpFilename])
if scrotExists or maimExists:
if scrotExists:
subprocess.call(['scrot', '-z', tmpFilename])
elif maimExists:
subprocess.call(['maim', tmpFilename])

im = Image.open(tmpFilename)

if region is not None:
assert len(region) == 4, 'region argument must be a tuple of four ints'
region = [int(x) for x in region]
im = im.crop((region[0], region[1], region[2] + region[0], region[3] + region[1]))
os.unlink(tmpFilename) # delete image of entire screen to save cropped version
os.unlink(tmpFilename) # delete image of entire screen to save cropped version
im.save(tmpFilename)
else:
# force loading before unlinking, Image.open() is lazy
Expand All @@ -488,8 +506,9 @@ def _screenshot_linux(imageFilename=None, region=None):
if imageFilename is None:
os.unlink(tmpFilename)
return im

else:
raise Exception('The scrot program must be installed to take a screenshot with PyScreeze on Linux. Run: sudo apt-get install scrot')
raise Exception('The scrot program or the maim program must be installed to take a screenshot with PyScreeze on Linux. Run: sudo apt-get install scrot')



Expand Down