我正在努力寻找一种减少样板装饰器的方法。我们有很多使用@decorate 的类。例如:
class MyClass(Base):
@decorate
def fun1(self):
pass
@decorate
def fun2(self):
pass
def fun3(self):
pass
我想让它默认装饰器在那里,除非有人另有说明。
我使用此代码进行自动换行
from functools import wraps
def myDecorator(func):
@wraps(func)
def decorator(self, *args, **kwargs):
try:
print 'enter'
ret = func(self, *args, **kwargs)
print 'leave'
except:
print 'exception'
ret = None
return ret
return decorator
class TestDecorateAllMeta(type):
def __new__(cls, name, bases, local):
for attr in local:
value = local[attr]
if callable(value):
local[attr] = myDecorator(value)
return type.__new__(cls, name, bases, local)
class TestClass(object):
__metaclass__ = TestDecorateAllMeta
def test_print2(self, val):
print val
def test_print(self, val):
print val
c = TestClass()
c.test_print1("print 1")
c.test_print2("print 2")
我的问题是:
- 有没有更好的方法来完成自动装饰?
- 我该如何进行覆盖?
理想情况下,我的最终解决方案是:
class TestClass(object):
__metaclass__ = TestDecorateAllMeta
def autowrap(self):
print("Auto wrap")
@dont_decorate
def test_dont_decorate(self, val):
print val
编辑
谈谈下面的评论之一,因为类是可调用的,而不是做
if callable(value):
它应该是:
if isinstance(value,types.FunctionType)