如何增强 Python 对象的方法？

Question

我有一个 Spam 对象列表：

class Spam:
    def update(self):
        print('updating spam!')

其中一些可能是 SpamLite 对象：

class SpamLite(Spam):
    def update(self):
        print('this spam is lite!')
        Spam.update(self)

我希望能够从列表中获取任意对象，并向其更新方法添加一些内容，例如：

def poison(spam):
    tmp = spam.update 
    def newUpdate(self):
        print 'this spam has been poisoned!'
        tmp(self)
    spam.update = newUpdate

我想要 spam.update()现在要么打印：

this spam has been poisoned!
updating spam!

要么

this spam has been poisoned!
this spam is lite!
updating spam!

取决于它是垃圾邮件还是垃圾邮件。

但这不起作用，因为 spam.update() 不会自动传递 self 参数，并且因为如果 tmp 离开范围或发生更改，那么它不会调用旧的更新。我有办法做到这一点吗？

Answer 1

def poison(spam):
    tmp = spam.update
    def newUpdate():
        print 'this spam has been poisoned!'
        tmp()
    spam.update = newUpdate

完整脚本：

class Spam:
    def update(self):
        print('updating spam!')

class SpamLite(Spam):
    def update(self):
        print('this spam is lite!')
        Spam.update(self)

def poison(spam):
    tmp = spam.update # it is a bound method that doesn't take any arguments
    def newUpdate():
        print 'this spam has been poisoned!'
        tmp()
    spam.update = newUpdate


from operator import methodcaller    
L = [Spam(), SpamLite()]
map(methodcaller('update'), L)
map(poison, L)
print "*"*79
map(methodcaller('update'), L)

输出：

updating spam!
this spam is lite!
updating spam!
*******************************************************************************
this spam has been poisoned!
updating spam!
this spam has been poisoned!
this spam is lite!
updating spam!

Answer 2

另一种方法，使用 MethodType：

class Spam:
    def update(self):
        print('updating spam!')

class SpamLite(Spam):
    def update(self):
        print('this spam is lite!')
        Spam.update(self)

def poison(spam):
    import types
    tmp = spam.update 
    def newUpdate(self):
        print 'this spam has been poisoned!'
        tmp()
    newUpdate = types.MethodType(newUpdate, spam, Spam)
    spam.update = newUpdate

spam = Spam()
spam_lite = SpamLite()
poison(spam)
poison(spam_lite)
spam.update()
print
spam_lite.update()

Answer 3

在 Python 世界中，MonkeyPatching 是不受欢迎的。

您确实应该使用 Mixin 方法并使用多重继承。

然后您可以动态替换（更新）父级以达到所需的效果。

Answer 4

使用这样的装饰器：

def method_decorator(f):
    def wrapper(self, *args, **kwargs):
        print('this spam has been poisoned!')
        return f(self)
    return wrapper

class Spam:
    def update(self):
        print('updating spam!')

    @method_decorator
    def update2(self):
        print('updating spam!')

Spam().update()
Spam().update2()

此打印：

updating spam!
this spam has been poisoned!
updating spam!

如果您想了解有关装饰器的更多信息，请阅读以下内容： http:// www.drdobbs.com/web-development/184406073

以上并不是一个“好公民”装饰器，阅读文章就知道如何写一个。请务必检查装饰器库： http://pypi.python.org/pypi/decorator

华泰