创建包装器类以围绕现有函数调用pre和post函数吗？

Question

我想创建一个包装另一个类的类，以便在通过包装类运行某个函数时，也可以运行pre和post函数。我希望包装器类无需修改即可与任何类一起使用。

例如，如果我有此类。

class Simple(object):
    def one(self):
        print "one"

    def two(self,two):
        print "two" + two

    def three(self):
        print "three"

我可以这样使用…

number = Simple()
number.one()
number.two("2")

到目前为止，我已经编写了这个包装器类…

class Wrapper(object):
    def __init__(self,wrapped_class):
        self.wrapped_class = wrapped_class()

    def __getattr__(self,attr):
        return self.wrapped_class.__getattribute__(attr)

    def pre():
        print "pre"

    def post():
        print "post"

我可以这样称呼…

number = Wrapper(Simple)
number.one()
number.two("2")

除了更改第一行之外，可以使用与上述相同的方法。

我想发生的事情是，通过包装类调用函数时，先调用包装类中的pre函数，然后再调用包装类中的所需函数，然后再调用post函数。我希望能够做到这一点，而无需更改包装的类，也无需更改函数的调用方式，而仅更改类实例的创建方式。例如number
= Simple（）vs number = Wrapper（Simple）

Answer 1

差不多了，您只需要在内部进行一些自省__getattr__，就可以在原始属性可调用时返回一个新的包装函数：

class Wrapper(object):
    def __init__(self,wrapped_class):
        self.wrapped_class = wrapped_class()

    def __getattr__(self,attr):
        orig_attr = self.wrapped_class.__getattribute__(attr)
        if callable(orig_attr):
            def hooked(*args, **kwargs):
                self.pre()
                result = orig_attr(*args, **kwargs)
                # prevent wrapped_class from becoming unwrapped
                if result == self.wrapped_class:
                    return self
                self.post()
                return result
            return hooked
        else:
            return orig_attr

    def pre(self):
        print ">> pre"

    def post(self):
        print "<< post"

现在使用以下代码：

number = Wrapper(Simple)

print "\nCalling wrapped 'one':"
number.one()

print "\nCalling wrapped 'two':"
number.two("2")

结果是：

Calling wrapped 'one':
>> pre
one
<< post

Calling wrapped 'two':
>> pre
two2
<< post