基本上,我想要的是这样做:
class B:
def fn(self):
print 'B'
class A:
def fn(self):
print 'A'
@extendInherit
class C(A,B):
pass
c=C()
c.fn()
并有输出
A
B
我将如何实现 extendInherit 装饰器?
基本上,我想要的是这样做:
class B:
def fn(self):
print 'B'
class A:
def fn(self):
print 'A'
@extendInherit
class C(A,B):
pass
c=C()
c.fn()
并有输出
A
B
我将如何实现 extendInherit 装饰器?
这不是装饰师的工作。你想彻底改变一个类的正常行为,所以这实际上是一个元类的工作。
import types
class CallAll(type):
""" MetaClass that adds methods to call all superclass implementations """
def __new__(meta, clsname, bases, attrs):
## collect a list of functions defined on superclasses
funcs = {}
for base in bases:
for name, val in vars(base).iteritems():
if type(val) is types.FunctionType:
if name in funcs:
funcs[name].append( val )
else:
funcs[name] = [val]
## now we have all methods, so decorate each of them
for name in funcs:
def caller(self, *args,**kwargs):
""" calls all baseclass implementations """
for func in funcs[name]:
func(self, *args,**kwargs)
attrs[name] = caller
return type.__new__(meta, clsname, bases, attrs)
class B:
def fn(self):
print 'B'
class A:
def fn(self):
print 'A'
class C(A,B, object):
__metaclass__=CallAll
c=C()
c.fn()
我个人不会尝试使用装饰器执行此操作,因为使用了新样式的类 and super()
,可以实现以下目标:
>>> class A(object):
... def __init__(self):
... super(A, self).__init__()
... print "A"
...
>>> class B(object):
... def __init__(self):
... super(B, self).__init__()
... print "B"
...
>>> class C(A, B):
... def __init__(self):
... super(C, self).__init__()
...
>>> foo = C()
B
A
我想方法调用会以同样的方式工作。
元类是一种可能的解决方案,但有些复杂。 super
可以非常简单地做到这一点(当然,使用新样式类:没有理由在新代码中使用旧类!):
class B(object):
def fn(self):
print 'B'
try: super(B, self).fn()
except AttributeError: pass
class A(object):
def fn(self):
print 'A'
try: super(A, self).fn()
except AttributeError: pass
class C(A, B): pass
c = C()
c.fn()
您需要 try/except 来支持单继承或多继承的任何顺序(因为在某些时候将不会沿着方法解析顺序 MRO 进一步基础,定义名为 的方法fn
,您需要捕获并忽略结果AttributeError
) . 但是正如您所看到的,与您根据您对不同答案的评论所认为的不同,您不一定需要在最叶的类中覆盖fn
,除非您需要在这样的覆盖中对该类执行特定的操作——super
有效对纯粹继承(而不是覆盖)的方法也很好!