import gtk.gdk


def main():
    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:
        pb.save("screenshot.png", "png")
        print "Screenshot saved to screenshot.png."
    else:
        print "Unable to get the screenshot."


if __name__ == '__main__':
    main()
