it-swarm-eu.dev

Machen Sie einen Screenshot über ein Python-Skript. [Linux]

Ich möchte einen Screenshot über ein Python-Skript aufnehmen und unauffällig speichern.

Ich bin nur an der Linux-Lösung interessiert und sollte jede X-basierte Umgebung unterstützen.

75
skyronic

Dies funktioniert ohne Scrot oder ImageMagick.

import gtk.gdk

w = gtk.gdk.get_default_root_window()
sz = w.get_size()
print "The size of the window is %d x %d" % sz
pb = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB,False,8,sz[0],sz[1])
pb = pb.get_from_drawable(w,w.get_colormap(),0,0,0,0,sz[0],sz[1])
if (pb != None):
    pb.save("screenshot.png","png")
    print "Screenshot saved to screenshot.png."
else:
    print "Unable to get the screenshot."

Ausgeliehen von http://ubuntuforums.org/showpost.php?p=2681009&postcount=5

64
Rusty

Kompilieren Sie alle Antworten in einer Klasse . Gibt das PIL-Bild aus.

#!/usr/bin/env python
# encoding: utf-8
"""
screengrab.py

Created by Alex Snet on 2011-10-10.
Copyright (c) 2011 CodeTeam. All rights reserved.
"""

import sys
import os

import Image


class screengrab:
    def __init__(self):
        try:
            import gtk
        except ImportError:
            pass
        else:
            self.screen = self.getScreenByGtk

        try:
            import PyQt4
        except ImportError:
            pass
        else:
            self.screen = self.getScreenByQt

        try:
            import wx
        except ImportError:
            pass
        else:
            self.screen = self.getScreenByWx

        try:
            import ImageGrab
        except ImportError:
            pass
        else:
            self.screen = self.getScreenByPIL


    def getScreenByGtk(self):
        import gtk.gdk      
        w = gtk.gdk.get_default_root_window()
        sz = w.get_size()
        pb = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB,False,8,sz[0],sz[1])
        pb = pb.get_from_drawable(w,w.get_colormap(),0,0,0,0,sz[0],sz[1])
        if pb is None:
            return False
        else:
            width,height = pb.get_width(),pb.get_height()
            return Image.fromstring("RGB",(width,height),pb.get_pixels() )

    def getScreenByQt(self):
        from PyQt4.QtGui import QPixmap, QApplication
        from PyQt4.Qt import QBuffer, QIODevice
        import StringIO
        app = QApplication(sys.argv)
        buffer = QBuffer()
        buffer.open(QIODevice.ReadWrite)
        QPixmap.grabWindow(QApplication.desktop().winId()).save(buffer, 'png')
        strio = StringIO.StringIO()
        strio.write(buffer.data())
        buffer.close()
        del app
        strio.seek(0)
        return Image.open(strio)

    def getScreenByPIL(self):
        import ImageGrab
        img = ImageGrab.grab()
        return img

    def getScreenByWx(self):
        import wx
        wx.App()  # Need to create an App instance before doing anything
        screen = wx.ScreenDC()
        size = screen.GetSize()
        bmp = wx.EmptyBitmap(size[0], size[1])
        mem = wx.MemoryDC(bmp)
        mem.Blit(0, 0, size[0], size[1], screen, 0, 0)
        del mem  # Release bitmap
        #bmp.SaveFile('screenshot.png', wx.BITMAP_TYPE_PNG)
        myWxImage = wx.ImageFromBitmap( myBitmap )
        PilImage = Image.new( 'RGB', (myWxImage.GetWidth(), myWxImage.GetHeight()) )
        PilImage.fromstring( myWxImage.GetData() )
        return PilImage

if __== '__main__':
    s = screengrab()
    screen = s.screen()
    screen.show()
45
Alex Snet

Nur der Vollständigkeit halber: Xlib - Beim Erfassen des gesamten Bildschirms ist es jedoch etwas langsam:

from Xlib import display, X
import Image #PIL

W,H = 200,200
dsp = display.Display()
root = dsp.screen().root
raw = root.get_image(0, 0, W,H, X.ZPixmap, 0xffffffff)
image = Image.fromstring("RGB", (W, H), raw.data, "raw", "BGRX")
image.show()

Man könnte versuchen, einige Typen in den Engpassdateien in PyXlib zu finden und diese dann mit Cython zu kompilieren. Das könnte die Geschwindigkeit etwas erhöhen.


Edit: Wir können den Kern der Funktion in C schreiben und ihn dann in Python von ctypes verwenden. Hier ist etwas, was ich zusammengehackt habe:

#include <stdio.h>
#include <X11/X.h>
#include <X11/Xlib.h>
//Compile hint: gcc -shared -O3 -lX11 -fPIC -Wl,-soname,prtscn -o prtscn.so prtscn.c

void getScreen(const int, const int, const int, const int, unsigned char *);
void getScreen(const int xx,const int yy,const int W, const int H, /*out*/ unsigned char * data) 
{
   Display *display = XOpenDisplay(NULL);
   Window root = DefaultRootWindow(display);

   XImage *image = XGetImage(display,root, xx,yy, W,H, AllPlanes, ZPixmap);

   unsigned long red_mask   = image->red_mask;
   unsigned long green_mask = image->green_mask;
   unsigned long blue_mask  = image->blue_mask;
   int x, y;
   int ii = 0;
   for (y = 0; y < H; y++) {
       for (x = 0; x < W; x++) {
         unsigned long pixel = XGetPixel(image,x,y);
         unsigned char blue  = (pixel & blue_mask);
         unsigned char green = (pixel & green_mask) >> 8;
         unsigned char red   = (pixel & red_mask) >> 16;

         data[ii + 2] = blue;
         data[ii + 1] = green;
         data[ii + 0] = red;
         ii += 3;
      }
   }
   XDestroyImage(image);
   XDestroyWindow(display, root);
   XCloseDisplay(display);
}

Und dann die Python-Datei:

import ctypes
import os
from PIL import Image

LibName = 'prtscn.so'
AbsLibPath = os.path.dirname(os.path.abspath(__file__)) + os.path.sep + LibName
grab = ctypes.CDLL(AbsLibPath)

def grab_screen(x1,y1,x2,y2):
    w, h = x2-x1, y2-y1
    size = w * h
    objlength = size * 3

    grab.getScreen.argtypes = []
    result = (ctypes.c_ubyte*objlength)()

    grab.getScreen(x1,y1, w, h, result)
    return Image.frombuffer('RGB', (w, h), result, 'raw', 'RGB', 0, 1)

if __== '__main__':
  im = grab_screen(0,0,1440,900)
  im.show()
34
JHolta

Dies funktioniert auf X11 und vielleicht auch auf Windows (jemandem, bitte überprüfen). Bedürfnisse PyQt4 :

import sys
from PyQt4.QtGui import QPixmap, QApplication
app = QApplication(sys.argv)
QPixmap.grabWindow(QApplication.desktop().winId()).save('test.png', 'png')
18
Juliano

Ich habe ein Wrapperprojekt ( pyscreenshot ) für scrot, imagemagick, pyqt, wx und pygtk . Wenn Sie eines davon haben, können Sie es verwenden.

Installieren:

easy_install pyscreenshot

Beispiel:

import pyscreenshot as ImageGrab

# fullscreen
im=ImageGrab.grab()
im.show()

# part of the screen
im=ImageGrab.grab(bbox=(10,10,500,500))
im.show()

# to file
ImageGrab.grab_to_file('im.png')
15
ponty

Plattformübergreifende Lösung mit wxPython :

import wx
wx.App()  # Need to create an App instance before doing anything
screen = wx.ScreenDC()
size = screen.GetSize()
bmp = wx.EmptyBitmap(size[0], size[1])
mem = wx.MemoryDC(bmp)
mem.Blit(0, 0, size[0], size[1], screen, 0, 0)
del mem  # Release bitmap
bmp.SaveFile('screenshot.png', wx.BITMAP_TYPE_PNG)
8
Snowball
import ImageGrab
img = ImageGrab.grab()
img.save('test.jpg','JPEG')

dies erfordert die Python Imaging Library

7
Slava V

Eine kurze Suche nach gtkShots sieht aus, als könnte es Ihnen helfen, da es ein GPL-Python-Screenshot-Programm ist.

3
Douglas Leeder

etwas spät aber nie leicht ist man

import autopy
import time
time.sleep(2)
b = autopy.bitmap.capture_screen()
b.save("C:/Users/mak/Desktop/m.png")
2
pkm

Es gibt ein Python-Paket für diese Autopy

Das Bitmap-Modul kann zum Scannen (bitmap.capture_screen) .__ verwendet werden.

2
ouille

Sie können dies verwenden

import os
os.system("import -window root screen_shot.png")
1
Ahir Kamlesh

Ich konnte unter Linux keinen Screenshot mit "pyscreenshot" oder "scrot" erstellen, da die Ausgabe von pyscreenshot nur eine schwarze Bilddatei war.

gott sei Dank gab es jedoch eine andere, sehr einfache Möglichkeit, Screenshots in Linux aufzunehmen, ohne etwas zu installieren. Geben Sie einfach den Code in Ihr Verzeichnis ein und führen Sie den Code mit python demo.py aus.

import os
os.system("gnome-screenshot --file=this_directory.png")

es gibt auch viele Optionen für gnome-screenshot --help

Application Options:
  -c, --clipboard                Send the grab directly to the clipboard
  -w, --window                   Grab a window instead of the entire screen
  -a, --area                     Grab an area of the screen instead of the entire screen
  -b, --include-border           Include the window border with the screenshot
  -B, --remove-border            Remove the window border from the screenshot
  -p, --include-pointer          Include the pointer with the screenshot
  -d, --delay=seconds            Take screenshot after specified delay [in seconds]
  -e, --border-effect=effect     Effect to add to the border (shadow, border, vintage or none)
  -i, --interactive              Interactively set options
  -f, --file=filename            Save screenshot directly to this file
  --version                      Print version information and exit
  --display=DISPLAY              X display to use
1
cyera

Aus diesem Thread :

 import os
 os.system("import -window root temp.png")
1
anderstood

Ich habe kürzlich ein Paket geschrieben, das einen Screenshot mit der X11-Bibliothek erstellt und das Bild als numpy-Array zurückgibt. Ich habe tatsächlich einige Vorschläge verwendet, die in diesem Thread erwähnt und verbessert wurden. Auf einer modernen Maschine ist eine typische Bildrate von 60+ fps für eine Auflösung von 1080p möglich. Eigentlich habe ich auf meiner Entwicklungsmaschine, nämlich ~ 3 Jahre, 200 fps gemacht. Hier ist der Link zum Projekt https://github.com/mherkazandjian/fastgrab

0
mher

Es ist eine alte Frage. Ich möchte es mit neuen Tools beantworten.

Funktioniert mit Python 3 (sollte mit Python 2 funktionieren, aber ich habe es nicht getestet) und PyQt5.

Minimales Arbeitsbeispiel. Kopieren Sie es in die Python-Shell und erhalten Sie das Ergebnis.

from PyQt5.QtWidgets import QApplication
app = QApplication([])
screen = app.primaryScreen()
screenshot = screen.grabWindow(QApplication.desktop().winId())
screenshot.save('/tmp/screenshot.png')
0
rominf