如何在新样式的类中拦截对python“magic”方法的调用？

Question

我试图在新样式的类中拦截对python的双下划线魔术方法的调用。这是一个微不足道的例子，但它表明了目的：

class ShowMeList(object):
    def __init__(self, it):
        self._data = list(it)

    def __getattr__(self, name):
        attr = object.__getattribute__(self._data, name)
        if callable(attr):
            def wrapper(*a, **kw):
                print "before the call"
                result = attr(*a, **kw)
                print "after the call"
                return result
            return wrapper
        return attr

如果我在list周围使用代理对象，我会得到非magic方法的预期行为，但是我的包装函数从来没有为magic方法调用过。

>>> l = ShowMeList(range(8))

>>> l #call to __repr__
<__main__.ShowMeList object at 0x9640eac>

>>> l.append(9)
before the call
after the call

>> len(l._data)
9

如果我没有从对象继承（第一行class ShowMeList:），那么一切都按预期工作：

>>> l = ShowMeList(range(8))

>>> l #call to __repr__
before the call
after the call
[0, 1, 2, 3, 4, 5, 6, 7]

>>> l.append(9)
before the call
after the call

>> len(l._data)
9

如何使用新样式类完成此截取？

Answer 1

使用__getattr__和__getattribute__是类的最后一个响应获取属性的资源。

请考虑以下几点：

>>> class C:
    x = 1
    def __init__(self):
        self.y = 2
    def __getattr__(self, attr):
        print(attr)

>>> c = C()
>>> c.x
1
>>> c.y
2
>>> c.z
z

__getattr__方法只在其他方法不起作用时调用（它对运算符不起作用，您可以阅读有关该方法的here）。

在您的示例中，__repr__和许多其他magic方法已经在object类中定义。

有一件事可以做，思想，它是定义那些神奇的方法，然后使然后调用__getattr__方法。请检查我提出的另一个问题及其答案（link）以查看某些代码正在执行此操作。

Answer 2

出于性能原因，Python总是在类（和父类）中查找magic方法，而不使用常规的属性查找机制。解决方法是在创建类时使用元类自动为magic方法添加代理；例如，我使用此技术避免了通过包装类的方法编写样板调用。

class Wrapper(object):
    """Wrapper class that provides proxy access to an instance of some
       internal instance."""

    __wraps__  = None
    __ignore__ = "class mro new init setattr getattr getattribute"

    def __init__(self, obj):
        if self.__wraps__ is None:
            raise TypeError("base class Wrapper may not be instantiated")
        elif isinstance(obj, self.__wraps__):
            self._obj = obj
        else:
            raise ValueError("wrapped object must be of %s" % self.__wraps__)

    # provide proxy access to regular attributes of wrapped object
    def __getattr__(self, name):
        return getattr(self._obj, name)

    # create proxies for wrapped object's double-underscore attributes
    class __metaclass__(type):
        def __init__(cls, name, bases, dct):

            def make_proxy(name):
                def proxy(self, *args):
                    return getattr(self._obj, name)
                return proxy

            type.__init__(cls, name, bases, dct)
            if cls.__wraps__:
                ignore = set("__%s__" % n for n in cls.__ignore__.split())
                for name in dir(cls.__wraps__):
                    if name.startswith("__"):
                        if name not in ignore and name not in dct:
                            setattr(cls, name, property(make_proxy(name)))

用法：

class DictWrapper(Wrapper):
    __wraps__ = dict

wrapped_dict = DictWrapper(dict(a=1, b=2, c=3))

# make sure it worked....
assert "b" in wrapped_dict                        # __contains__
assert wrapped_dict == dict(a=1, b=2, c=3)        # __eq__
assert "'a': 1" in str(wrapped_dict)              # __str__
assert wrapped_dict.__doc__.startswith("dict()")  # __doc__

Answer 3

对于Asymmetric behavior for __getattr__, newstyle vs oldstyle classes（另请参见Python docs）的答案，使用__getattr__或__getattribute__修改对“magic”方法的访问权限对于新样式的类是不可能的。这个限制使解释器更快。