小白的Python之路 day4 装饰器高潮

首先装饰器实现的条件:

高阶函数+嵌套函数 =》装饰器

1.首先,我们先定义一个高级函数,去装饰test1函数,得不到我们想要的操作方式

 import time

 #定义高阶函数
 def deco(func):
     start_time = time.time()
     func()
     stop_time = time.time()
     print("the func run time is %s"%(stop_time-start_time))

 #装饰test1函数

 def test1():
     time.sleep(3)
     print("in the test1")
 #deco(test1)  #这样操作就改变函数的调用方式 , 怎样操作不改变调用方式
 test1 = deco(test1)  #想想前提是用return 返回值

 #输出
 in the test1
 the func run time is 3.0002999305725098

2 .用高阶函数的返回值方法,得到了想要的调用方式,但是结果不是想要的,一直到现在我们一直在用高阶函数

 import time

 #定义高阶函数
 def deco(func):
     start_time = time.time()
     return func()
     stop_time = time.time()
     print("the func run time is %s"%(stop_time-start_time))

 #装饰test1函数

 def test1():
      time.sleep(3)
      print("in the test1")

 test1()     #调用方式问题解决了,但是结果不是想要的结果

 结果
 in the test1

3 . 再用嵌套函数得到了我们想要的结果和不变的调用方式(哇喔,是不是好牛X)

 import time

 #定义内置函数
 def timmer(func):  #timmer(test1) func=test1
     def deco():
         start_time = time.time()
         func()   #run test1()
         stop_time = time.time()
         print("the func run time is %s"%(stop_time-start_time))

     return deco  #重点的重点

 #装饰test1函数
 @timmer  # 相当于test1 = timmer(test1)   牛X的方法
 def test1():
     time.sleep(3)
     print("in the test1")

 #直接执行test1函数
 test1()

 #输出
 in the test1
 the func run time is 3.0002999305725098

4.  我们用装饰器去装饰一个带参数的函数会怎么样?

 import time

 def timmer(func):
     def deco():
         start_time = time.time()
         func()   #run test2()  这里没有参数,报错
         stop_time = time.time()
         print("the func run time is %s"%(stop_time-start_time))

     return deco  

 @timmer
 def test2(name,age):
     print("name:%s,age:%s"%(name,age))

 test2()

 #输出
 Traceback (most recent call last):
   File "D:/PycharmProjects/pyhomework/day4/装饰器/装饰器高潮.py", line 16, in <module>
     test2()
   File "D:/PycharmProjects/pyhomework/day4/装饰器/装饰器高潮.py", line 6, in deco
     func()   #run test2()
 TypeError: test2() missing 2 required positional arguments: 'name' and 'age' #缺少传入name和age参数

结果:很显然是错误的。因为这边执行的test2函数其实就是执行的deco函数,deco函数体内的func()其实就是执行test2函数,但是,test2需要传入name和age两个参数,所以报错。那怎么解决呢?

5. 我们只能传入参数了,但是你又不能确定传入几个参数,所以我们只能用非固定参数传参。代码如下:(哇喔,这个更牛X)

 import time

 def timmer(func):  #timmer(test1) func=test1
     def deco(*args,**kwargs):  #传入非固定参数  可以面对各种参数
         start_time = time.time()
         func(*args,**kwargs)   #传入非固定参数
         stop_time = time.time()
         print("the func run time is %s"%(stop_time-start_time))

     return deco

 #不带参数
 @timmer  # 相当于test1 = timmer(test1)
 def test1():
     time.sleep(3)
     print("in the test1")

 #带参数
 @timmer
 def test2(name,age):
     print("name:%s,age:%s"%(name,age))
 #调用
 test1()
 test2("qianduoduo",22)

 #输出
 #test1
 in the test1
 the func run time is 3.0010883808135986
 #test2
 name:qianduoduo,age:22
 the func run time is 0.0  #test2

6.我就问还有没有终极版的,答案是:有

前面的都是小高潮(基本上也能满足百分之九十的需求),现在正式介绍终极的高潮版的 语法糖(我也不知道为什么取这个名字)也叫装饰器

 import time
 user,passwd = 'qian','zxc123'
 def auth(func):
     def wrapper(*args,**kwargs):   #包装
         username = input("Username:").strip()
         password = input("Password:").strip()

         if user == username and passwd == password:
             print("\033[32;1mUser has passed authentication\033[0m")
             func(*args,**kwargs)
         else:
             exit("\033[31;1mInvalid username or password\033[0m")
     return wrapper

 def index():  #建立网站的首页  首页不需要登入
     print("welcome to index page")
 @auth    #加上装饰器
 def home():  #自己的页面,需要登入
     print("welcome to home page")
 @auth
 def bbs():  #论坛区的页面,需要登入
     print("welcome to bbs page")

 index()

 home()

 bbs()
 运行结果:  #运行很完美,堪称经典之作
 welcome to index page
 Username:qian
 Password:zxc123
 User has passed authentication
 welcome to home page
 Username:qian
 Password:zxc123
 User has passed authentication
 welcome to bbs page

7.走到这地方,我的问题来了,我的home执行完没有任何数据,我现在在home中return一个数据,看代码:

 import time
 user,passwd = 'qian','zxc123'
 def auth(func):
     def wrapper(*args,**kwargs):   #包装
         username = input("Username:").strip()
         password = input("Password:").strip()

         if user == username and passwd == password:
             print("\033[32;1mUser has passed authentication\033[0m")
             func(*args,**kwargs)
         else:
             exit("\033[31;1mInvalid username or password\033[0m")
     return wrapper

 def index():  #建立网站的首页  首页不需要登入
     print("welcome to index page")
 @auth    #加上装饰器
 def home():  #自己的页面,需要登入
     print("welcome to home page")
     return "from home"  #假设这就是我返回的数据
 @auth
 def bbs():  #论坛区的页面,需要登入
     print("welcome to bbs page")

 index()

 print(home())  #home的执行结果

 bbs()

 结果:
 welcome to index page
 Username:qian
 Password:zxc123
 User has passed authentication
 welcome to home page
 None   # home的执行结果,没有执行结果
 #(虽然装饰器没改变源代码,没改变调用方式,但是改变了执行结果)
 # 这时候我们怎么办,怎么获得home的返回结果?
 Username:

8 我们来分析一下,按照顺序①②③

 import time
 user,passwd = 'qian','zxc123'
 def auth(func):
     def wrapper(*args,**kwargs):   #包装
         username = input("Username:").strip()
         password = input("Password:").strip()

         if user == username and passwd == password:
             print("\033[32;1mUser has passed authentication\033[0m")
             func(*args,**kwargs)  #from home
             #   ①  func的执行结果没有传给谁,那就没了,就是None
         else:
             exit("\033[31;1mInvalid username or password\033[0m")
     return wrapper #③wrapper执行了,没有返回值

 def index():  #建立网站的首页  首页不需要登入
     print("welcome to index page")
 @auth    #加上装饰器
 def home():  #自己的页面,需要登入
     print("welcome to home page")
     return "from home"
 @auth
 def bbs():  #论坛区的页面,需要登入
     print("welcome to bbs page")

 index()

 print(home())  #②调用home 时候,就是调用wrapper()
 bbs()
                 

9 分析完得出以下代码:

 import time
 user,passwd = 'qian','zxc123'
 def auth(func):
     def wrapper(*args,**kwargs):   #包装
         username = input("Username:").strip()
         password = input("Password:").strip()

         if user == username and passwd == password:
             print("\033[32;1mUser has passed authentication\033[0m")
            # return   func(*args,**kwargs)  函数到这里就结束了
             res = func(*args,**kwargs) #增加变量
             print("---after authenticaion")  #再装饰点其他东西
             return res
         else:
             exit("\033[31;1mInvalid username or password\033[0m")
     return wrapper

 def index():  #建立网站的首页  首页不需要登入
     print("welcome to index page")
 @auth    #加上装饰器
 def home():  #自己的页面,需要登入
     print("welcome to home page")
     return "from home"
 @auth
 def bbs():  #论坛区的页面,需要登入
     print("welcome to bbs page")

 index()

 print(home())

 bbs()

 结果
 welcome to index page
 Username:qian
 Password:zxc123
 User has passed authentication
 welcome to home page
 ---after authenticaion
 from home
 Username:qian
 Password:zxc123
 User has passed authentication
 welcome to bbs page
 ---after authenticaion

10不过这不是我要讲的高潮部分,而是补充前面的前奏

同志们在真正的现实生活中,账号认证方式有多种,现在的认证方式是本地认证,下面才是真正的终极

 import time
 user,passwd = 'qian','zxc123'
 def auth(auth_type):
     print("auth func:",auth_type)
     def outer_wrapper(func):
         def wrapper(*args,**kwargs):   #包装
             if auth_type == "local":
                 username = input("Username:").strip()
                 password = input("Password:").strip()
                 if user == username and passwd == password:
                     print("\033[32;1mUser has passed authentication\033[0m")
                     res = func(*args,**kwargs)
                     print("---after authenticaion")
                     return res
                 else:
                     exit("\033[31;1mInvalid username or password\033[0m")
             elif auth_type == "ldap":
                 print("搞毛线ldap,不会")
         return wrapper
     return outer_wrapper    

 def index():  #
     print("welcome to index page")
 @auth(auth_type = "local")   #本地的文件认证  home = wrapper
 def home():
     print("welcome to home page")
     return "from home"
 @auth(auth_type ="ldap")     #ldap认证
 def bbs():
     print("welcome to bbs page")

 index()

 print(home())  #wrapper()     

 bbs()

 结果
 auth func: local
 auth func: ldap
 welcome to index page
 Username:qian
 Password:zxc123
 User has passed authentication
 welcome to home page
 ---after authenticaion
 from home         #调用方式 home() 就没这个返回值了
 搞毛线ldap,不会

这就是我们的终极版本, 运行的具体顺序可以自己调试调试

这个函数的作用分别是:

  1.auth(auth_type) 传递装饰器的参数

  2.outer_wrapper(func) 把函数当做实参传递进来

  3.wrapper(*args,**kwargs) 真正执行装饰的函数

##home()
上一篇:Hive操作之HQL语句


下一篇:Python成长之路_装饰器