假设我有一堂课
class Foo:
def __init__(self):
pass
def meth1(self, *args):
do_stuff()
def meth2(self, **kwargs):
more_stuff()
def callme(self):
print "I'm called!"
我想巧妙地修改它,以便callme()
在执行之后调用它meth1
,meth2
以及其他所有内容。我怎么做?我从来没有使用过类装饰器或元类,所以我不确定哪种方法是正确的,以及如何正确使用它们。
对于元类,我整理了以下内容:
from types import FunctionType
def addcall_meta(name, bases, dct):
newdct = dict(dct)
for k, v in dct.items():
if isinstance(v, FunctionType) and k not in ('__metaclass__', 'callme'):
# the above can include more filters such as not startswith, etc.
print k, 'is a function!'
def newmethod(self, *args, **kwargs):
v(self, *args, **kwargs)
dct['callme'](self)
newdct[k] = newmethod
return type(name, bases, newdct)
使用类装饰器:
def addcall_deco(cls):
dct = {}
for k, v in cls.__dict__.items():
if isinstance(v, FunctionType) and k != 'callme':
def newv(self, *args, **kwargs):
v(self, *args, **kwargs)
self.callme()
cls.k = newv
return cls
两者似乎都适用于我的超级简单示例:
In [75]: Foo()
I'm called!
Out[75]: <__main__.Foo instance at 0x21ed680>
是否有明显的理由使用一个而不是另一个(例如特定的极端情况等)?我也认为我可以以某种方式使用__new__
,但从常识的角度来看,我想修改类,而不是它的所有实例。