Question

我有一个垃圾邮件对象列表：

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

MonkeyPatching在python世界中不受欢迎。

您应该使用Mixin方法并使用Multiple inheritance。

然后，您可以动态替换（更新）父项，以达到预期的效果。

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

HTH

如何扩充Python对象的方法？

4 个答案:

完整脚本：

输出：