A simple, cross-platform screenshot module for Python 2 and 3.
Project description
PyScreeze is a simple, cross-platform screenshot module for Python 2 and 3.
About
PyScreeze can take screenshots, save them to files, and locate images within the screen. This is useful if you have a small image of, say, a button that needs to be clicked and want to locate it on the screen.
Screenshot functionality requires the Pillow module. OS X uses the screencapture command, which comes with the operating system. Linux uses the scrot command, which can be installed by running sudo apt-get install scrot.
Special Notes About Ubuntu
Unfortunately, Ubuntu seems to have several deficiencies with installing Pillow. PNG and JPEG support are not included with Pillow out of the box on Ubuntu. The following links have more information
The screenshot() Function
Calling screenshot() will return an Image object (see the Pillow or PIL module documentation for details). Passing a string of a filename will save the screenshot to a file as well as return it as an Image object.
>>> import pyscreeze >>> im1 = pyscreeze.screenshot() >>> im2 = pyscreeze.screenshot('my_screenshot.png')
On a 1920 x 1080 screen, the screenshot() function takes roughly 100 milliseconds - it’s not fast but it’s not slow.
There is also an optional region keyword argument, if you do not want a screenshot of the entire screen. You can pass a four-integer tuple of the left, top, width, and height of the region to capture:
>>> import pyscreeze >>> im = pyscreeze.screenshot(region=(0,0, 300, 400))
The Locate Functions
You can call the locateOnScreen(‘calc7key.png’) function to get the screen coordinates of the 7 button for a calculator app. The return value is a 4-integer tuple: (left, top, width, height). This tuple can be passed to center() to get the X and Y coordinates at the center of this region. If the image can’t be found on the screen, locateOnScreen() returns None.
>>> import pyscreeze >>> button7location = pyscreeze.locateOnScreen('calc7key.png') >>> button7location (1416, 562, 50, 41) >>> button7x, button7y = pyscreeze.center(button7location) >>> button7x, button7y (1441, 582) >>> pyscreeze.click(button7x, button7y) # clicks the center of where the 7 button was found
The locateCenterOnScreen() function is probably the one you want to use most often:
>>> import pyscreeze >>> x, y = pyscreeze.locateCenterOnScreen('calc7key.png') >>> pyscreeze.click(x, y)
On a 1920 x 1080 screen, the locate function calls take about 1 or 2 seconds. This may be too slow for action video games, but works for most purposes and applications.
There are several “locate” functions. They all start looking at the top-left corner of the screen (or image) and look to the left and then down. The arguments can either be a
locateOnScreen(image, grayscale=False) - Returns (left, top, width, height) coordinate of first found instance of the image on the screen. Returns None if not found on the screen.
locateCenterOnScreen(image, grayscale=False) - Returns (x, y) coordinates of the center of the first found instance of the image on the screen. Returns None if not found on the screen.
locateAllOnScreen(image, grayscale=False) - Returns a generator that yields (left, top, width, height) tuples for where the image is found on the screen.
locate(needleImage, haystackImage, grayscale=False) - Returns (left, top, width, height) coordinate of first found instance of needleImage in haystackImage. Returns None if not found on the screen.
locateAll(needleImage, haystackImage, grayscale=False) - Returns a generator that yields (left, top, width, height) tuples for where needleImage is found in haystackImage.
The “locate all” functions can be used in for loops or passed to list():
>>> import pyscreeze >>> for pos in pyscreeze.locateAllOnScreen('someButton.png') ... print(pos) ... (1101, 252, 50, 50) (59, 481, 50, 50) (1395, 640, 50, 50) (1838, 676, 50, 50) >>> list(pyscreeze.locateAllOnScreen('someButton.png')) [(1101, 252, 50, 50), (59, 481, 50, 50), (1395, 640, 50, 50), (1838, 676, 50, 50)]
Grayscale Matching
Optionally, you can pass grayscale=True to the locate functions to give a slight speedup (about 30%-ish). This desaturates the color from the images and screenshots, speeding up the locating but potentially causing false-positive matches.
>>> import pyscreeze >>> button7location = pyscreeze.locateOnScreen('calc7key.png', grayscale=True) >>> button7location (1416, 562, 50, 41)
Pixel Matching
To obtain the RGB color of a pixel in a screenshot, use the Image object’s getpixel() method:
>>> import pyscreeze >>> im = pyscreeze.screenshot() >>> im.getpixel((100, 200)) (130, 135, 144)
Or as a single function, call the pixel() PyScreeze function, which is a wrapper for the previous calls:
>>> import pyscreeze >>> pyscreeze.pixel(100, 200) (130, 135, 144)
If you just need to verify that a single pixel matches a given pixel, call the pixelMatchesColor() function, passing it the X coordinate, Y coordinate, and RGB tuple of the color it represents:
>>> import pyscreeze >>> pyscreeze.pixelMatchesColor(100, 200, (130, 135, 144)) True >>> pyscreeze.pixelMatchesColor(100, 200, (0, 0, 0)) False
The optional tolerance keyword argument specifies how much each of the red, green, and blue values can vary while still matching:
>>> import pyscreeze >>> pyscreeze.pixelMatchesColor(100, 200, (130, 135, 144)) True >>> pyscreeze.pixelMatchesColor(100, 200, (140, 125, 134)) False >>> pyscreeze.pixelMatchesColor(100, 200, (140, 125, 134), tolerance=10) True
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.