Add examples (PIL, GNU/Linux, general)

BoboTiG · BoboTiG · commit 24f0e2e8aa9c · 2016-10-05T13:04:04.000+02:00
diff --git a/.gitignore b/.gitignore
@@ -1,7 +1,8 @@
 build/
 .cache/
 dist/
-MANIFEST
+*.egg-info/
+MANIFEST*
 .DS_Store
 *.orig
 *.png
diff --git a/MANIFEST.in b/MANIFEST.in
diff --git a/docs/source/examples.rst b/docs/source/examples.rst
@@ -2,6 +2,23 @@
 Examples
 ========
 
+GNU/Linux
+---------
+
+Usage example with a specific display::
+
+    from mss.linux import MSS
+
+
+    display = ':0.0'
+    print('Screenshot of display "{0}"'.format(display))
+    output = 'monitor{0}-%d.png'.format(display)
+
+    with MSS(display=display) as screenshotter:
+        for filename in screenshotter.save(output=output):
+            print(filename)
+
+
 Using PIL
 ---------
 
diff --git a/examples/callback.py b/examples/callback.py
@@ -7,7 +7,7 @@
 from os import rename
 from os.path import isfile
 
-from mss import ScreenshotError, mss
+from mss import mss, ScreenshotError
 
 
 def main():
diff --git a/examples/linux-display_keyword.py b/examples/linux-display_keyword.py
@@ -9,12 +9,13 @@
 
 
 def main():
-    ''' Usage example with specified display. '''
+    ''' Usage example with a specific display. '''
 
     display = ':0.0'
+    print('Screenshot of display "{0}"'.format(display))
+
     try:
         with MSS(display=display) as screenshotter:
-            print('Screenshot of display "{0}"'.format(display))
             output = 'monitor{0}-%d.png'.format(display)
             for filename in screenshotter.save(output=output):
                 print(filename)
diff --git a/examples/pil.py b/examples/pil.py
@@ -0,0 +1,43 @@
+#!/usr/bin/env python
+# coding: utf-8
+''' This is part of the MSS Python's module.
+    Source: https://github.com/BoboTiG/python-mss
+'''
+
+from mss import mss, ScreenshotError
+from PIL import Image
+
+
+def main():
+    ''' PIL example using frombytes(). '''
+
+    try:
+        with mss() as screenshotter:
+            # We retrieve monitors informations:
+            monitors = screenshotter.enum_display_monitors()
+
+            # Get rid of the first, as it represents the "All in One" monitor:
+            for num, monitor in enumerate(monitors[1:], 1):
+                # Get raw pixels from the screen.
+                # This method will store screen size into `width` and `height`
+                # and raw pixels into `image`.
+                screenshotter.get_pixels(monitor)
+
+                # Create an Image:
+                size = (screenshotter.width, screenshotter.height)
+                img = Image.frombytes('RGB', size, screenshotter.image)
+
+                # And save it!
+                output = 'monitor-{0}.png'.format(num)
+                img.save(output)
+                print(output)
+
+            return 0
+    except ScreenshotError as ex:
+        print(ex)
+
+    return 1
+
+
+if __name__ == '__main__':
+    exit(main())
