import sys
import time
try:
    import gtk.gdk
    timestr = time.strftime("%Y%m%d-%H%M%S")
    pth = "/tmp"
    fnm = "screenshot"
    if (len(sys.argv) > 1):
        fnm = sys.argv[1]
    w = gtk.gdk.get_default_root_window()
    sz = w.get_size()
    fn = pth + "/" + fnm + "_" + str(sz[0]) + "_" + str(sz[1]) + "_" + timestr + ".png"
    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(fn,"png")
        print "%s" % fn
        sys.exit()
    else:
        sys.exit(2)
except:
    sys.exit()