如何使用另一个类的方法装饰（monkeypatch...）Python 类？

Question

httplib.HTTPMessage 和 email.message.Message 类[1] 都实现了 RFC822 标头解析的方法。不幸的是，它们有不同的实现[2]，并且不提供相同级别的功能。

困扰我的一个例子是：

• httplib.HTTPMessage 缺少 email.Message 中存在的 get_filename 方法，该方法允许您轻松地从 Content-disposition: 附件中检索文件名； filename="fghi.xyz" header;

• httplib.HTTPMessage 有 getparam、getplist 和 parseplist 方法，但据我所知，它们不是也不可能是在 content-type 标头解析之外使用；

• email.Message 有一个通用的 get_param 方法来解析带有参数的任何 RFC822 标头，例如 content-disposition 或 content -type。

因此，我想要 httplib.HTTPMessage 中 email.message.Message 的 get_filename 或 get_param 方法，但是，当然，我无法修补 httplib.HTTPMessage 因为它在标准库中... :-q

最后，这里是装饰器主题... :-)

我成功创建了一个 Monkeypatch_http_message 函数用我缺少的解析方法来装饰 httplib.HTTPMessage：

def monkeypatch_http_message(obj):
    from email import utils
    from email.message import (
        _parseparam,
        _unquotevalue,
    )
    cls = obj.__class__

    # methods **copied** from email.message.Message source code
    def _get_params_preserve(self, failobj, header): ...
    def get_params(self, failobj=None, header='content-type', 
                   unquote=True): ...
    def get_param(self, param, failobj=None, header='content-type', 
                  unquote=True): ...
    def get_filename(self, failobj=None): ...

    # monkeypatching httplib.Message
    cls._get_params_preserve = _get_params_preserve
    cls.get_params = get_params
    cls.get_param = get_param
    cls.get_filename = get_filename

现在我可以做：

import mechanize
from some.module import monkeypatch_http_message
browser = mechanize.Browser()

# in that form, browser.retrieve returns a temporary filename 
# and an httplib.HTTPMessage instance
(tmp_filename, headers) = browser.retrieve(someurl) 

# monkeypatch the httplib.HTTPMessage instance
monkeypatch_http_message(headers)

# yeah... my original filename, finally
filename = headers.get_filename()

这里的问题是我从字面上复制了源类中的装饰方法代码，我'我想避免。

所以，我尝试通过引用源方法来装饰：

def monkeypatch_http_message(obj):
    from email import utils
    from email.message import (
        _parseparam,
        _unquotevalue,
        Message    # XXX added
    )
    cls = obj.__class__

    # monkeypatching httplib.Message
    cls._get_params_preserve = Message._get_params_preserve
    cls.get_params = Message.get_params
    cls.get_param = Message.get_param
    cls.get_filename = Message.get_filename

但这给了我：

Traceback (most recent call last):
  File "client.py", line 224, in <module>
    filename = headers.get_filename()
TypeError: unbound method get_filename() must be called with Message instance as first argument (got nothing instead)

我现在正在抓耳挠腮......如何在不直接复制源方法的情况下装饰我的类？

有什么建议吗？ :-)

问候，

乔治·马丁

---

1. 在 Python 2.6 中。我无法在生产中使用 2.7 或 3.x。

2. httplib.HTTPMessage继承自mimetools.Message和rfc822.Message，而email.Message有自己的

Answer 1

在 Python 3.x 中，未绑定方法消失了，因此在这种情况下您只需获取文件对象，并且您的第二个示例将起作用：

>>> class C():
...   def demo(): pass
... 
>>> C.demo
<function demo at 0x1fed6d8>

在 Python 2.x 中，您可以通过未绑定方法访问底层函数，也可以通过检索它来访问底层函数直接从类字典中获取（从而绕过将其转换为未绑定方法的正常查找过程）：

>>> class C():
...   def demo(): pass
... 
>>> C.demo.im_func                  # Retrieve it from the unbound method
<function demo at 0x7f463486d5f0>
>>> C.__dict__["demo"]              # Retrieve it directly from the class dict
<function demo at 0x7f463486d5f0>

后一种方法的优点是与 Python 3.x 向前兼容。

Answer 2

@ncoghlan：我无法在注释中放入缩进代码，所以这里再次：

def monkeypatch_http_message(obj):
    import httplib
    assert isinstance(obj, httplib.HTTPMessage)
    cls = obj.__class__

    from email import utils
    from email.message import (_parseparam, _unquotevalue, Message)
    funcnames = ('_get_params_preserve', 'get_params', 'get_param', 'get_filename')
    for funcname in funcnames:
        cls.__dict__[funcname] = Message.__dict__[funcname]

谢谢！ :-)