zoukankan      html  css  js  c++  java
  • python 函数 装饰器的使用方法

    一、装饰器 

    首先,我们要了解到什么是开放封闭式原则?

        软件一旦上线后,对修改源代码是封闭的,对功能的扩张是开放的,所以我们应该遵循开放封闭的原则。

      也就是说:我们必须找到一种解决方案,能够在不修改一个功能源代码以及调用方式的前提下,为其加上新功能。

    总结:原则如下:

      1、不修改源代码

      2、不修改调用方式

    目的:在遵循1和2原则的基础上扩展新功能。

    二、什么是装饰器?

    器:指的是工具,

    装饰:指的是为被装饰对象添加新功能。

    完整的含义:装饰器即在不修改装饰对象源代码与调用方式的前提下,为被装饰器对象添加新功能的一种函数,这个函数的特殊之处就在于它的返回值也是一个函数。

    一般而言,我们想要拓展原来函数的代码,直接的办法就是侵入代码里面修改,例如:

    1 import time
    2 def index():
    3     start_time=time.time()
    4     time.sleep(3)
    5     print('hello word')
    6     stop_time=time.time()
    7     print('run time is %s'%(stop_time-start_time))
    8 
    9 index()

    ###输出:

    hello word
    run time is 3.000171422958374

      以上代码就是让你过三秒才打印‘hello word’,下面我们要再添加一个新功能,和上面的功能一样,但是要传参数进去,过5秒输出结果。

    修改1:

     1 import  time
     2 def index():
     3     time.sleep(3)
     4     print('hello word')
     5 
     6 def home(name):
     7     time.sleep(5)
     8     print('welcome %s to home page'%name)
     9 
    10 def wrapper(func):
    11     start_time=time.time()
    12     func('haolo')
    13     stop_time=time.time()
    14     print('run time is %s'%(stop_time-start_time))
    15 
    16 
    17 wrapper(home)

    ###输出:

    welcome haolo to home page
    run time is 5.000286102294922

    这样写感觉还是不怎么好,而且我们还修改了函数的调用方式,很不符合规矩。所以我们还是换一种方式来修改它。通过装饰器的方式。

    修改2

     1 import time
     2 def index():
     3     time.sleep(3)
     4     print('hello word')
     5 
     6 def home(name):
     7     time.sleep(5)
     8     print('welcome to %s'%name)
     9 
    10 def outter(func):  # func为最原始的inde 和home
    11     def warpper():
    12         start_time=time.time()
    13         func('yuan')
    14         stop_time=time.time()
    15         print(stop_time-start_time)
    16     return warpper
    17 
    18 home=outter(home) ###home这个变量名是新赋值的,把原来的home给覆盖了。
    19 home()

    ###输出:

    welcome to yuan
    5.000286102294922

    这种方式虽然满足了不修改源代码和不修改调用方式的条件,但还是不能够实现两个函数同时运行的功能,说到底还是不行,我们还得想个方式出来。就是让他们两个同时运行。这时,我又想到了上节课所学的知识,就是*args和**kargs,用两个函数通过可变参数形式来实现内嵌函数的形式传入,所以它支持运行是构建参数列表,这对于以上两次不能解决的办法是最有效的。下面我们来试试,看到底能不能成功。

    方式3:

     1 import time
     2 def index():
     3     time.sleep(3)
     4     print('hello word')
     5 
     6 def home(name):
     7     time.sleep(5)
     8     print('welcome %s to home page'%name)
     9 
    10 def timmer(func):   #func为最原始的home
    11     def warpper(*args,**kwargs):
    12         start_time=time.time()
    13         res=func(*args,**kwargs)  #调用了最原始的home
    14         stop_time=time.time()
    15         print(stop_time-start_time)
    16         return res
    17     return warpper
    18 
    19 index=timmer(index) #为最新的index = wrapper
    20 home=timmer(home)  #为最新的home = wrapper
    21 
    22 home(name='yuan')  #wrapper=('yuan')
    23 index()  #wrapper

    ###输出:

    welcome yuan to home page
    5.000285863876343
    hello word
    3.000171661376953

    看吧,很快就实现了两个功能并用,而且我们还没有修改原始代码,还有调用方式。

    其实很简单,我只是用了一个无参装饰器的模板,这个模板可以说是万能的,在以后很多的函数代码都可以用这种方式来套用。

    模板:

    1 def outer(func):
    2     def inner(*args,**kwargs):
    3         res=func(*args,**kwargs)
    4         return res
    5     return inner

    现在又有问题来了,我们调装饰器的时候,每调一次,又要把装饰器对象传进来,调一次又传一次,这样不会觉得很麻烦吗?那么我们又想到了一种方法,就是装饰器语法糖,在被装饰对象的上面加@timmer 用它来取代 index=timmer(index)
    并且把返回值正常的返回给它。

     1 import time
     2 def timmer(func):   #func为最原始的home
     3     def warpper(*args,**kwargs):
     4         start_time=time.time()
     5         res=func(*args,**kwargs)  #调用了最原始的home
     6         stop_time=time.time()
     7         print(stop_time-start_time)
     8         return res
     9     return warpper
    10 
    11 @timmer  #就是取代底下的 index=timmer(index)
    12 def index():
    13     time.sleep(3)
    14     print('hello word')
    15 
    16 @timmer #就是取代底下的home=timmer(home)   home(name='yuan')
    17 def home(name):
    18     time.sleep(5)
    19     print('welcome %s to home page'%name)
    20 
    21 index()
    22 home('liuyuan')

    ###输出;

    hello word
    3.000171661376953
    welcome liuyuan to home page
    5.000286102294922

    注意:这里的timmer函数就是最原始的装饰器,它的参数就是一个函数,然后返回值也是一个函数。其中作为参数的这个函数index()hemo(name)就是在返回函数的wrapper()的内部执行。然后再这两个函数的前面加上@timmerindex()home(name)函数就相当于被注入了计时功能,现在只需要调用index()home(‘yuan’),它就已经变身为‘新功能更多的函数了。’

      所以这里的装饰器就像一个注入的符号:有了它,拓展了原来函数的功能既不需要侵入函数内更改代码,也不需要重复执行原函数。

    用装饰器来实现认证功能:

     1 import time
     2 current_user={
     3     'username':None,
     4     #'login_time':None
     5 }
     6 
     7 def auth(func):
     8     # func = index
     9     def wrapper(*args,**kwargs):
    10         if current_user['username']:
    11             print('已经登录过了')
    12             res=func(*args,**kwargs)
    13             return res
    14 
    15         uname = input('输入用户名:').strip()
    16         pwd = input('密码:').strip()
    17         if uname == 'yuan' and pwd == '123':
    18             print('登录成功')
    19             current_user['username']=uname
    20             res = func(*args,**kwargs)
    21             return res
    22         else:
    23             print('用户名或密码错误')
    24     return wrapper
    25 
    26 @auth #index = auth(index)
    27 def index():
    28     time.sleep(1)
    29     print('welcom to index page')
    30 
    31 @auth
    32 def home(name):
    33     time.sleep(2)
    34     print('welcome %s to home page'%name)
    35 
    36 input()
    37 home('yuan')

    有参数的装饰器来用于用户认证

     1 import time
     2 current_user={
     3     'username':None,
     4     # 'login_time':None
     5 }
     6 
     7 def auth(func):
     8     # func=index
     9     def wrapper(*args,**kwargs):
    10         if current_user['username']:
    11             print('已经登陆过了')
    12             res=func(*args,**kwargs)
    13             return res
    14 
    15         uname=input('用户名>>: ').strip()
    16         pwd=input('密码>>: ').strip()
    17         if uname == 'yuan' and pwd == '123':
    18             print('登陆成功')
    19             current_user['username']=uname
    20             res=func(*args,**kwargs)
    21             return res
    22         else:
    23             print('用户名或密码错误')
    24     return wrapper
    25 
    26 def timmer(func):
    27     def wrapper(*args,**kwargs):
    28         start_time=time.time()
    29         res=func(*args,**kwargs)
    30         stop_time=time.time()
    31         print(stop_time-start_time)
    32         return res
    33     return wrapper
    34 
    35 @timmer # timmer 统计的是auth+index的执行时间
    36 @auth
    37 def index():
    38     time.sleep(1)
    39     print('welcome to index page')
    40     return 122
    41 
    42 index()

    叠加多个装饰器:

     1 import time
     2 current_user={
     3     'username':None,
     4     # 'login_time':None
     5 }
     6 
     7 def auth(func):
     8     # func=index
     9     def wrapper(*args,**kwargs):
    10         if current_user['username']:
    11             print('已经登陆过了')
    12             res=func(*args,**kwargs)
    13             return res
    14 
    15         uname=input('用户名>>: ').strip()
    16         pwd=input('密码>>: ').strip()
    17         if uname == 'egon' and pwd == '123':
    18             print('登陆成功')
    19             current_user['username']=uname
    20             res=func(*args,**kwargs)
    21             return res
    22         else:
    23             print('用户名或密码错误')
    24     return wrapper
    25 
    26 def timmer(func):
    27     def wrapper(*args,**kwargs):
    28         start_time=time.time()
    29         res=func(*args,**kwargs)
    30         stop_time=time.time()
    31         print(stop_time-start_time)
    32         return res
    33     return wrapper
    34 
    35 @timmer # timmer 统计的是auth+index的执行时间
    36 @auth
    37 def index():
    38     time.sleep(1)
    39     print('welcome to index page')
    40     return 122
    41 
    42 index()
  • 相关阅读:
    UML常见工具之Powerdesigner
    在webForm中WebRequest\WebClient\WebBrowser获取远程页面源码的三种方式(downmoon)
    忍不住了,我来说两句,从一道面试题说起
    《UML用户指南第二版》再次温读笔记(一)(downmoon)
    Database Project requires local SQL 2005 instance的解决方案(downmoon)
    JDBC Driver For SQL2000/2005/2008
    服务器更新dll后导致网站崩溃,重启iis也无效的一种解决方案(downmoon)
    白孩儿一个网上流传的故事[生活感悟]
    vs2008中js的语法提示及修正功能(downmoonn)
    Contoso 大学 2 – 实现基本的增删改查
  • 原文地址:https://www.cnblogs.com/lyox/p/8671098.html
Copyright © 2011-2022 走看看