python中的decorator

 在正式进入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之后,一切都变得很亲切啦。 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值