本文实例讲述了linux下python抓屏实现方法。分享给大家供大家参考。具体实现代码如下：
#!/usr/bin/python
”’by zevolo, 2012.12.20
”’
import gtk.gdk
import gtk
import glib
class 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 = y
class 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):
pass
class 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()
希望本文所述对大家的python程序设计有所帮助。