Implementation of python screen capture in linux

Implementation of python screen capture in linux

This article describes how to implement python screen capture in linux. Share it with you for your reference. The specific implementation code is as follows:

#!/usr/bin/python'''by zevolo, 2012.12.20'''import gtk.gdkimport gtkimport glibclass MyRect():  def __init__(self, x = 0, y = 0, w = 0, h = 0):    self.x = x    self.y = y    self.w = w    self.h = h  def __init__(self, x, y):    self.x = min(int(x.x), int(y.x))    self.y = min(int(x.y), int(y.y))    self.w = abs(int(y.x - x.x))    self.h = abs(int(y.y - x.y))class MyPair():  def __init__(self, x = 0, y = 0):    self.x = x    self.y = yclass MyPoint(MyPair):  def __init__(self, x = 0, y = 0):    MyPair.__init__(self, x, y)class MySize(MyPair):  def __init__(self, w = 0, h = 0):    MyPair.__init__(self, x, y)class MyCapture():  (event_enter, event_leave) = (0, 1)  def __init__(self):    pass  def capture(self):    pass  def handleEvent(self, event):    if event == event_enter:      enterSnap()    elif event == event_leave:      leaveSnap()  def enterSnap(self):    pass  def leaveSnap(self):    pass  def snap(self, window = None, rect = None, name = None):    passclass MyCaptureGtk(MyCapture):  def __init__(self):    MyCapture.__init__(self)    self.window = gtk.Window()    self.window.set_default_size(1,1)    self.window.connect("button-press-event", self.button_press_cb)    self.first = None    self.second = None    self.window.show()    #self.window.set_events(gtk.gdk.BUTTON_PRESS_MASK)  def getWindow(self):    return self.window  def button_press_cb(self, widget, event):    #print "type is %d" % event.type    if event.type == gtk.gdk.BUTTON_PRESS:      if event.button == 1: #left button        print "(%d, %d), (%d, %d), button is %d" % (event.x_root, event.y_root, event.x, event.y, event.button)        if not self.first:          self.first = MyPoint(event.x_root, event.y_root)        else:          self.second = MyPoint(event.x_root, event.y_root)          self.snap(None, MyRect(self.first, self.second))          self.first = None      elif event.button == 3: #right button        self.uncapture()      else:        pass  def uncapture(self):    if self.first:      print "cancel"      self.first = None    else:      print "exit now"      gtk.gdk.pointer_ungrab()      gtk.mainquit()  def capture(self, time = 0L):    cursor = gtk.gdk.Cursor(gtk.gdk.display_get_default(), gtk.gdk.CROSSHAIR)    ret = gtk.gdk.pointer_grab(self.window.window, True,          gtk.gdk.BUTTON_PRESS_MASK,          None, cursor, time)    if ret == gtk.gdk.GRAB_SUCCESS:      print "left button start, end, right button cancel/exit"    else:      print "failed to capture %d, (viewable %d),(frozen %d), (already %d)" \       % (ret, gtk.gdk.GRAB_NOT_VIEWABLE, gtk.gdk.GRAB_FROZEN, gtk.gdk.GRAB_ALREADY_GRABBED)  def snap(self, window = None, rect = None, name = None):    w = window    if not window:      #w = gtk.gdk.get_default_root_window()      d = gtk.gdk.display_get_default()      w = d.get_default_screen().get_root_window()    r = rect    if not r:      sz = w.get_size()      r = MyRect(0, 0, sz[0], sz[1])    print "The size of the window is (%d, %d, %d, %d)" % (r.x, r.y, r.w, r.h)    n = name    if not n:      n = "screenshot.png"    buf = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB,False,8, r.w, r.h)    buf = buf.get_from_drawable(w,w.get_colormap(), r.x, r.y, 0, 0, r.w, r.h)    if (buf != None):      buf.save(n, "png")      print "Screenshot saved to %s." % n    else:      print "Unable to get the screenshot."def timeout(data):  #print "timeout"  data.capture()if __name__ == '__main__':  cap = MyCaptureGtk()  w = cap.getWindow()  w.show()  glib.timeout_add_seconds(1, timeout, cap)#cap.snap()  gtk.main()

I hope this article will help you with Python programming.