Question

我想在python中包装特定类的每个方法，并且我想通过最低限度地编辑类的代码来实现。我应该怎么做呢？

Answer 1

Michael Foord的Voidspace博客在一个条目中描述了一种优雅的方法，该条目关于哪些元类以及如何在标题为A Method Decorating Metaclass的部分中使用它们。稍微简化并将其应用于您的情况导致：

from types import FunctionType
from functools import wraps

def wrapper(method):
    @wraps(method)
    def wrapped(*args, **kwrds):
    #   ... <do something to/with "method" or the result of calling it>
    return wrapped

class MetaClass(type):
    def __new__(meta, classname, bases, classDict):
        newClassDict = {}
        for attributeName, attribute in classDict.items():
            if isinstance(attribute, FunctionType):
                # replace it with a wrapped version
                attribute = wrapper(attribute)
            newClassDict[attributeName] = attribute
        return type.__new__(meta, classname, bases, newClassDict)

class MyClass(object):
    __metaclass__ = MetaClass  # wrap all the methods
    def method1(self, ...):
        # ...etc ...

在Python中，函数/方法装饰器只是函数包装器加上一些语法糖，使它们易于使用（而且更漂亮）。

Python 3兼容性更新

以前的代码使用Python 2.x元类语法，需要翻译才能在Python 3.x中使用，但是它在以前的版本中不再有效。这意味着它需要使用：

class MyBase(metaclass=MetaClass)
    ...

而不是：

class MyBase(object): 
    __metaclass__ = MetaClass"
    ...

如果需要，可以编写兼容Python 2.x 和 3.x的代码，但这样做需要使用稍微复杂的技术，动态创建一个新的基类，继承了所需的元类，从而避免了由于两个版本的Python之间的语法差异而导致的错误。这基本上是本杰明彼得森的six模块的with_metaclass()函数所做的。

from types import FunctionType
from functools import wraps

def wrapper(method):
    @wraps(method)
    def wrapped(*args, **kwrds):
        print('{!r} executing'.format(method.__name__))
        return method(*args, **kwrds)
    return wrapped


class MetaClass(type):
    def __new__(meta, classname, bases, classDict):
        newClassDict = {}
        for attributeName, attribute in classDict.items():
            if isinstance(attribute, FunctionType):
                # replace it with a wrapped version
                attribute = wrapper(attribute)
            newClassDict[attributeName] = attribute
        return type.__new__(meta, classname, bases, newClassDict)


def with_metaclass(meta):
    """ Create an empty class with the supplied bases and metaclass. """
    return type.__new__(meta, "TempBaseClass", (object,), {})


if __name__ == '__main__':

    # Inherit metaclass from a dynamically-created base class.
    class MyClass(with_metaclass(MetaClass)):
        @staticmethod
        def a_static_method():
            pass

        @classmethod
        def a_class_method(cls):
            pass

        def a_method(self):
            pass

    instance = MyClass()
    instance.a_static_method()  # Not decorated.
    instance.a_class_method()   # Not decorated.
    instance.a_method()         # -> 'a_method' executing

Answer 2

你的意思是以编程方式为一个类的方法设置一个包装器？嗯，这可能是一个非常糟糕的做法，但是你可以这样做：

def wrap_methods( cls, wrapper ):
    for key, value in cls.__dict__.items( ):
        if hasattr( value, '__call__' ):
            setattr( cls, key, wrapper( value ) )

如果你有课，例如

class Test( ):
    def fire( self ):
        return True
    def fire2( self ):
        return True

和包装

def wrapper( fn ):
    def result( *args, **kwargs ):
        print 'TEST'
        return fn( *args, **kwargs )
    return result

然后调用

wrap_methods( Test, wrapper )

会将wrapper应用于课程Test中定义的所有方法。 谨慎使用！实际上，根本不要使用它！

Answer 3

如果需要广泛修改默认类行为，那么MetaClasses就是您的选择。这是另一种方法。

如果您的用例仅限于包装类的实例方法，您可以尝试覆盖__getattribute__魔术方法。

from functools import wraps
def wrapper(func):
    @wraps(func)
    def wrapped(*args, **kwargs):
        print "Inside Wrapper. calling method %s now..."%(func.__name__)
        return func(*args, **kwargs)
    return wrapped

确保在创建包装器时使用functools.wraps，如果包装器用于调试，则更是如此，因为它提供了合理的TraceBack。

import types
class MyClass(object): # works only for new-style classes
    def method1(self):
        return "Inside method1"
    def __getattribute__(self, name):
        attr = super(MyClass, self).__getattribute__(name)
        if type(attr) == types.MethodType:
            attr = wrapper(attr)
        return attr

Answer 4

使用python decorators是最干净的方法，因为它看起来像你想调试或至少跟踪它出现的代码。

如何包装一个类的每个方法？

4 个答案: