通过PyGObject调用GDK截图

9
24
2011

通过PyGObject调用GDK截图

本文来自依云's Blog，转载请注明。

Linux 下截个图挺麻烦的。最开始我想学 scrot 使用 Xlib，结果因为看不懂而放弃，转而使用GDK。搜到了TualatriX的这篇《几十行代码构造一个截屏软件》。虽然才不到50行的 C 代码，但我还是觉得有点长。

本来准备像上次的《使用Xtest模拟鼠标点击》一样写成 Python 模块的，后来从 Vayn 那里看到原来可以通过 PyGObject 来调用 GTK 及 GDK 等等（hello world 程序）。于是我也用这种方式完成了截图的代码，才十几行，原理和TualatriX的完全一样。

import mimetypes
from gi.repository import Gdk

def screenshot(filename, rect=None, filetype=None):
  screen = Gdk.Screen.get_default()
  if rect is None:
    rect = (0, 0, screen.width(), screen.height())
  if filetype is None:
    t = mimetypes.guess_type(filename)[0]
    if t is None:
      raise ValueError('cannot guess filetype for filename: %s' % filename)
    filetype = t.split('/')[1]

  rootwin = screen.get_root_window()
  pixbuf = Gdk.pixbuf_get_from_window(rootwin, *rect)
  pixbuf.savev(filename, filetype, (), ())

不过没有找到PyGObject的文档。官方说可以自己从 gir 文件生成，但是那个脚本在最新版的代码中才有，而那个代码也要求Glib非常新，我的 Arch 上都没有那么新，于是作罢。所以用法除了自己按 GDK 的文档猜就是 Google 了。那个savev的参数我都找到mono的文档去了。。。

PyGObject 默认是使用 GTK 3。也可以指定使用 GTK 2：

import gi
gi.require_version("Gdk", "2.0")
gi.require_version("Gtk", "2.0")
from gi.repository import Gdk, Gtk

在/usr/lib/girepository-1.0/下还有一些typelib文件，说明这些库都有 GObject Introspection 支持，可以用包括 Python 3 在内的任何其支持的语言访问。不过我调用 xlib 时出错了：

>>> from gi.repository import xlib
>>> d = xlib.open_display()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib/python3.2/site-packages/gi/types.py", line 44, in function
    return info.invoke(*args)
glib.GError: Could not locate XOpenDisplay: `XOpenDisplay': python3: undefined symbol: XOpenDisplay

GObject Introspection 这个东西挺好的，除了文档。文档啊，就算不能支持 Python 的 docstring，至少也弄个 HTML 版出来啊，现在只有堆 XML 文件叫我怎么情何以堪啊，现在用 PyGObject 写代码就像在猜谜。。。

最后，代码的 github 链接。

Category: python | Tags: gtk linux python | Read Count: 16205