#!/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()
希望本文所述对大家的python程序设计有所帮助。