Python动态添加到函数

Question

如何在之前或之后向现有功能添加代码？

例如，我有一堂课：

 class A(object):
     def test(self):
         print "here"

我该如何编辑类元编程，以便执行此操作

 class A(object):
     def test(self):
         print "here"

         print "and here"

也许有某种方法可以附加另一个功能进行测试？

添加另一个功能，例如

 def test2(self):
      print "and here"

并将原件更改为

 class A(object):
     def test(self):
         print "here"
         self.test2()

有没有办法做到这一点？

Answer 1

如果需要，可以使用装饰器来修改功能。但是，由于它不是在函数初始定义时应用的装饰器，因此您将无法使用@语法糖来应用它。

>>> class A(object):
...     def test(self):
...         print "orig"
...
>>> first_a = A()
>>> first_a.test()
orig
>>> def decorated_test(fn):
...     def new_test(*args, **kwargs):
...         fn(*args, **kwargs)
...         print "new"
...     return new_test
...
>>> A.test = decorated_test(A.test)
>>> new_a = A()
>>> new_a.test()
orig
new
>>> first_a.test()
orig
new

请注意，它也会修改现有实例的方法。

编辑 ：使用和将装饰器的args列表修改为 更好的 版本args``kwargs