Python之美--Decorator深入详解
转自：http://www.cnblogs.com/SeasonLee/archive/2010/04/24/1719444.html
一些往事
在正式进入Decorator话题之前，请允许我讲一个小故事。
在最近的项目开发过程中，一位同事在读我的代码的时候，提出质疑，为什么同样的验证代码要重复出现在服务接口中呢？
如某服务接口是游戏中的玩家想建造建筑：
def build(user, build_name):
if not is_user_valid(user):
redirect("/auth/login/")
return False
#do something here
return create_building(build_name)
在build这个方法中，前3行是用来检测玩家是否已经验证过的，如果是非验证的玩家，就重定向到登陆页面；如果是验证过的玩家，就给他建造他想要的建筑。
他指出这样的3行验证代码(虽然已经很短)将会出现任意一个玩家服务接口中——升级建筑、拆建筑、甚至是频繁的聊天，我说这只是Ctrl+C、Ctrl+V的时间啊，但是那时候我深想，如果现在需求有变动，要在原来的验证失败的情况下，写日志，那原来的3行代码就变成4行：
if not is_user_valid(user):
redirect("/auth/login/")
log_warning(" %s is trying to enter without permission!" %(user["name"]))
return False
更痛苦的是，你要在每个出现if not is_user_valid(user)的地方增加log_warning这一行，这些时间足够你泡一壶好茶、听完一首《Poker Face》、tweet好几次、甚至修复了一个bug……
正如文章开头Time Peters所说的那样，总会有一个最好(合适)的方法来完成一件事。之后请教赖总，就祭出Decorator。以下是我基于Decorator改写的验证代码:
def authenticated(method):
def wrapper(user, *args):
if not is_user_valid(user):
redirect("/auth/login/")
log_warning(" %s is trying to enter without permission!" %(user["name"]))
return False
return method(user, *args)
return wrapper
@authenticated
def build(user, build_name):
return create_building(build_name)
使用Decorator的好处是玩家服务接口——升级建筑、拆建筑、聊天，需要进行验证的时候，只需要在方法前加上@authenticated就可，更重要的是因需求而对验证失败情况的处理时(如上面讲到的log)，并不会影响原有代码的结构，因为你只要在authenticated方法中加入log_warning这一行就搞掂啦！
毫无疑问Decorator对于维持代码结构起到神一样的作用，下面让我们进入Decorator之旅，PS：如果你已经是Decorator高手，请先别急着Ctrl+W，望不吝赐教，指出笔者在文中的问题和疏漏，不胜感激！
初始Decorator
Decorator，修饰符，是在Python2.4中增加的功能，也是pythoner实现元编程的最新方式，同时它也是最简单的元编程方式。为什么是“最简单”呢？是的，其实在Decorator之前就已经有classmethod()和staticmethod()内置函数，但他们的缺陷是会导致函数名的重复使用(可以看看David Mertz的Charming Python: Decorators make magic easy )，以下是摘自他本人的原文：
class C:
def foo(cls, y):
print "classmethod", cls, y
foo = classmethod(foo)
是的，classmethod做的只是函数转换，但是它却让foo这个名字另外出现了2次。记得有一句话是：人类因懒惰而进步。Decorator的诞生，让foo少出现2次。
class C:
@classmethod
def foo(cls, y):
print "classmethod", cls, y
读者也许已经想到Decorator在python中是怎么处理的了(如果还没头绪的，强烈建议先去看看limodou写的Decorator学习笔记 )。下面我列出4种用法。
| 写法 | 使用Decorator | 不使用Decorator |
| 单个Decorator，不带参数 | @dec def method(args): pass |
def method(args): pass method = dec(method) |
| 多个Decorator，不带参数 | @dec_a @dec_b @dec_c def method(args): pass |
def method(args): pass method = dec_a(dec_b(dec_c(method))) |
| 单个Decorator，带参数 | @dec(params) def method(args): pass |
def method(args): pass method = dec(params)(method) |
| 多个Decorator，带参数 | @dec_a(params1) @dec_b(params2) @dec_c(params3) def method(args): pass |
def method(args): pass method = dec_a(params1)(dec_b(params2)(dec_c(params)(method))) |
单个 Decorator，不带参数
设想一个情景，你平时去买衣服的时候，跟售货员是怎么对话的呢？售货员会先向你问好，然后你会试穿某件你喜爱的衣服。
def salesgirl(method):
def serve(*args):
print "Salesgirl:Hello, what do you want?", method.__name__
method(*args)
return serve
@salesgirl
def try_this_shirt(size):
if size < 35:
print "I: %d inches is to small to me" %(size)
else:
print "I:%d inches is just enough" %(size)
try_this_shirt(38)
结果是：
Salesgirl:Hello, what do you want? try_this_shirt
I:38 inches is just enough
这只是一个太简单的例子，以至一些“细节”没有处理好，你试穿完了好歹也告诉salesgirl到底要不要买啊。。。这样try_this_shirt方法需要改成带返回值 (假设是bool类型,True就是要买，False就是不想买)，那么salesgirl中的serve也应该带返回值，并且返回值就是 method(*args)。
修改后的salesgirl
def salesgirl(method):
def serve(*args):
print "Salesgirl:Hello, what do you want?", method.__name__
return method(*args)
return serve
@salesgirl
def try_this_shirt(size):
if size < 35:
print "I: %d inches is to small to me" %(size)
return False
else:
print "I:%d inches is just enough" %(size)
return True
result = try_this_shirt(38)
print "Mum:do you want to buy this?", result
结果是：
Salesgirl:Hello, what do you want? try_this_shirt
I:38 inches is just enough
Mum:do you want to buy this? True
现在我们的salesgirl还不算合格，她只会给客人打招呼，但是客人要是买衣服了，也不会给他报价；客人不买的话，也应该推荐其他款式！
会报价的salesgirl:
def salesgirl(method):
def serve(*args):
print "Salesgirl:Hello, what do you want?", method.__name__
result = method(*args)
if result:
print "Salesgirl: This shirt is 50$."
else:
print "Salesgirl: Well, how about trying another style?"
return result
return serve
@salesgirl
def try_this_shirt(size):
if size < 35:
print "I: %d inches is to small to me" %(size)
return False
else:
print "I:%d inches is just enough" %(size)
return True
result = try_this_shirt(38)
print "Mum:do you want to buy this?", result
结果是：
Salesgirl:Hello, what do you want? try_this_shirt
I:38 inches is just enough
Salesgirl: This shirt is 50$.
Mum:do you want to buy this? True
这样的salesgirl总算是合格了，但离出色还很远，称职的salesgirl是应该对熟客让利，老用户总得有点好处吧？
单个 Decorator，带参数
会报价并且带折扣的salesgirl：
def salesgirl(discount):
def expense(method):
def serve(*args):
print "Salesgirl:Hello, what do you want?", method.__name__
result = method(*args)
if result:
print "Salesgirl: This shirt is 50$.As an old user, we promised to discount at %d%%" %(discount)
else:
print "Salesgirl: Well, how about trying another style?"
return result
return serve
return expense
@salesgirl(50)
def try_this_shirt(size):
if size < 35:
print "I: %d inches is to small to me" %(size)
return False
else:
print "I:%d inches is just enough" %(size)
return True
result = try_this_shirt(38)
print "Mum:do you want to buy this?", result
结果是：
Salesgirl:Hello, what do you want? try_this_shirt
I:38 inches is just enough
Salesgirl: This shirt is 50$.As an old user, we promised to discount at 50%
Mum:do you want to buy this? True
这里定义的salesgirl是会给客户50%的折扣，因为salesgirl描述符是带参数，而参数就是折扣。如果你是第一次看到这个 salesgirl，会被她里面嵌套的2个方法而感到意外，没关系，当你习惯了Decorator之后，一切都变得很亲切啦。
