python3.x 基础三:装饰器

时间:2022-04-18 22:10:53

装饰器:本质是函数,用于装饰其他函数,在不改变其他函数的调用和代码的前提下,增加新功能

原则:

1.不能修改被装饰函数的源代码

2.不能修改被装饰函数的调用方式

3.装饰函数对于被装饰函数透明

参考如下链接:

http://egon09.blog.51cto.com/9161406/1836763

实现装饰器的需求:

  1. 函数即“变量”,函数名指向内存中的函数体,加()表示调用
  2. 高价函数,将函数当做参数传递给其他函数
  3. 嵌套函数,在函数体内再定义函数

1.调用简单的函数:

import time
def test1():
time.sleep(
1)
print('in the test1')
test1()

output:
in the test1

2.在不改变原来函数的定义和调用方式,增加计时功能

import time
def timmer(func):
def warpper(*args,**kwargs):
start_time
=time.time()
func()
stop_time
=time.time()
print('the func run time is %s' %(stop_time-start_time))
return warpper

@timmer
def test1():
time.sleep(
1)
print('in the test1')
test1()

output:
in the test1
the function run time is -1.0007600784301758:
python3.x 基础三:装饰器python3.x 基础三:装饰器
# /usr/bin/env python
#
-*- coding: utf-8 -*-
#
Author:Jenvid.yang
#
def foo():
#
print('in the foo')
#
bar()
#
foo()
#
bar() 未定义,肯定会报错

# def bar():
#
print('in the bar')
#
def foo():
#
print('in the foo')
#
bar()
#
foo()

# def foo():
#
print('in the foo')
#
bar()
#
def bar():
#
print('in the bar')
#
foo()
#
定义函数的时候,只是用函数名指向内存地址,只要函数执行步骤在函数定义之后即可,各函数间前后关系不影响

# def foo():
#
print('in the foo')
#
bar()
#
foo()
#
def bar():
#
print('in the bar')
函数名及调用关系

总结1:定义函数的时候,只是用函数名指向内存地址,只要函数执行步骤在函数定义步骤之后即可,各函数间定义先后不影响,与变量定义先后一个道理

 

高阶函数

1.把一个函数名当做实参传给另外一个函数--实现装饰器:不修改被装饰函数下,给源代码,也就是被装饰函数增加新功能

2.返回值中包含函数名--实现装饰器:不修改函数的调用方式

  • 函数返回实参函数内存地址
def bar():
print('in the bar')
def test1(func):
print(func)
test1(bar)
print('分割线'.center(30,'-'))
def test2(func):
return func
fff=test2(bar) # 将内存地址,也就是函数体,也就是test2返回值,赋值给一个变量
fff() # 变量加上(),表示调用
output:
<function bar at 0x7f5fa0a7d730>
-------------分割线--------------
in the bar
  • 高阶函数传参方式实现源代码bar没有被更改,调用方式被更改了
import time
def bar():
time.sleep(
1)
print('in the bar')
def test1(func):
start_time
=time.time()
func()
stop_time
=time.time()
print('the function run time is %s' % (stop_time-start_time))
test1(bar)
output:
in the bar
the function run time is 1.0011107921600342
#被调用函数原来调用方式是:bar(),函数体里面的内容也没有更改
#新的函数是test1,调用的是test1(),改变了调用函数名,如果有生产线上已经在调用bar(),这种情况下意味着所有的调用bar函数的其他函数,都需要将bar改成test1
  • 高阶函数返回值方式实现函数调用方式不更改
import time
def bar():
time.sleep(1)
print('in the bar')
def test1(func):
start_time=time.time()
return func
stop_time=time.time()
print('the function run time is %s' % (stop_time-start_time))
bar=test1(bar)
bar()
output:
in the bar
# 高阶函数返回内存地址实现调用方式不更改,但函数的新功能又没加上
  • 嵌套函数,在高阶函数下,给新功能函数增加多一层函数嵌套
import time
def bar():
time.sleep(
1)
print('in the bar')
def timmer(func):
def test1():
start_time
=time.time()
func()
stop_time
=time.time()
print('the function run time is %s' % (stop_time-start_time))
return test1 #返回test1的内存地址
bar
=timmer(bar) #将bar函数地址作为实参传递给timmer函数的形参func,func将在嵌套函数test1里面进行调用
bar() #此时的bar其实是timmer函数地址,并不是原来的bar
output:
in the bar
the function run time is 1.0010881423950195
  • 语法糖
import time
def timmer(func):
def test1():
start_time
=time.time()
func()
stop_time
=time.time()
print('the function run time is %s' % (stop_time-start_time))
return test1
# bar=timmer(bar)
@timmer #只需要在被调用函数的前面用关键字符“@装饰器函数”,既可以实现装饰器效果
def bar():
time.sleep(
1)
print('in the bar')
bar()
output:
in the bar
the function run time is 1.0011053085327148
import time
def timmer(func):
def warpper(): #一般将装饰器嵌套函数名命名为warpper 据说懂了这个可以解决90%的问题
start_time
=time.time()
func()
stop_time
=time.time()
print("the function run time is %s:" %(start_time-stop_time))
return warpper
@timmer #这里还是相当于更改被装饰函数的’变量名‘内存地址 bar=timmer(bar)
def bar():
time.sleep(
1)
print('in the bar')
bar()
  • 装饰器参数传值-被装饰函数传1个值
import time
def timmer(func):
def warpper(name):
start_time
=time.time()
func(name)
stop_time
=time.time()
print(
"the function run time is %s:" %(start_time-stop_time))
return warpper
@timmer
def test1(name):
time.sleep(
1)
print(
'in the test1 %s',name)
test1(
'alex')
output:
in the test1 %s alex
the function  run time is -1.0006732940673828:
  • 装饰器参数传值-被装饰函数传2个值
import time
def timmer(func):
def warpper(name,age):
start_time
=time.time()
func(name,age)
stop_time
=time.time()
print(
"the function run time is %s:" %(start_time-stop_time))
return warpper
@timmer
def test1(name,age):
time.sleep(
1)
print(
'in the test1 name is %s,age is %d'%(name,age))
test1(
'alex',18)
output:
in the test1 name is alex,age is 18
the function  run time is -1.000108003616333:
  • 装饰器参数传值-被装饰函数传n个值
import time
def timmer(func):
# print(
'func timmer')
def wrapper(
*args,**kwargs):
print(
'打印实参:',args,kwargs)
start_time
=time.time()
func(
*args,**kwargs)
stop_time
=time.time()
print(
"the function run time is %s:" % (stop_time-start_time))
return wrapper
@timmer
# test1
=timmer(test1)
def test1(name,age):
time.sleep(
1)
print(
'in the test1 func, name is %s,age is %d' % (name,age))
test1(
'alex',18)

@timmer
# test1
=timmer(test1)
def test2(name,age,sex,salary):
time.sleep(
1)
print(
'in the test1 func, name is %s,age is %d,sex is %s,salary is %d' % (name,age,sex,salary))
test2(
'alex',18,'man',3000)
output:
打印实参: ('alex', 18) {}
in the test1 func, name is alex,age is 18
the function  run time is 1.0013768672943115:
打印实参: ('alex', 18, 'man', 3000) {}
in the test1 func, name is alex,age is 18,sex is man,salary is 3000
the function  run time is 1.0012915134429932:
  • 试错过程:
  • import time
    def timmer(func):
    def warpper():
    # print(
    '打印实参:',args,kwargs)
    start_time
    =time.time()
    func()
    stop_time
    =time.time()
    print(
    "the function run time is %s:" % (stop_time-start_time))
    return warpper
    @timmer #这里相当于test1=timmer(test1)=wrapper,test1加上参数相当于test1(name,age)=wrapper(name,age)
    def test1(name,age):
    time.sleep(
    1)
    print(
    'in the test1 name is %s,age is %s'%(name,age))
    test1(
    'alex',18)
    输出如下:
    /usr/bin/python3.5 /root/PycharmProjects/S14/day04/bin/decoreate03.py
    Traceback (most recent call last):
    File
    "/root/PycharmProjects/S14/day04/bin/decoreate03.py", line 17, in <module>
    test1(
    'alex',18)
    TypeError: warpper() takes
    0 positional arguments but 2 were given

    Process finished with exit code
    1
    step1:定义装饰器函数体timmer
    step2:语法糖,调用装饰器函数,准备装饰下一行函数,返回内层嵌套函数变量warpper内存地址
    step3:定义函数体test1
    step4:调用函数test1,将test1作为实参,传递给装饰器timmer
    step5:warpper函数调用,需要讲test1的实参传递给wrapper,由于命名空间关系,test1的参数无法传递到warpper,func也无法获得实参
  • 给包装函数warrper增加形参
  • import time
    def timmer(func):
    def warpper(name,age):
    # print('打印实参:',args,kwargs)
    start_time=time.time()
    func()
    stop_time
    =time.time()
    print("the function run time is %s:" % (stop_time-start_time))
    return warpper
    @timmer
    #test1=timmer(test1)
    def test1(name,age):
    time.sleep(
    1)
    print('in the test1 name is %s,age is %s'%(name,age))
    test1(
    'alex',18)
    输出如下
    /usr/bin/python3.5 /root/PycharmProjects/S14/day04/bin/decoreate03.py
    Traceback (most recent call last):
    File
    "/root/PycharmProjects/S14/day04/bin/decoreate03.py", line 17, in <module>
    test1(
    'alex',18)
    File
    "/root/PycharmProjects/S14/day04/bin/decoreate03.py", line 9, in warpper
    func()
    TypeError: test1() missing
    2 required positional arguments: 'name' and 'age'
    这个解释很明显,func就是test1,缺少了2个位置参数,把func()改成func(name,age)即正确

 

  • python3.x 基础三:装饰器python3.x 基础三:装饰器
    # /usr/bin/env python
    #
    -*- coding: utf-8 -*-
    #
    Author:Jenvid.yang
    import time
    def timmer(func): #func=test2
    def wrapper(*args,**kwargs):
    start_time
    =time.time()
    func(
    *args,**kwargs)
    stop_time
    =time.time()
    print('func run time is %s' %(stop_time-start_time))
    return wrapper

    @timmer
    def bar():
    time.sleep(
    1)
    print('in the bar')
    bar()
    @timmer
    #test2=timmer(test2)=warpper ->test2(name)=warpper(name)
    def test2(name):
    time.sleep(
    1)
    print('in the test2',name)
    test2(
    'alex')
    课堂分析源码
  • 被装饰函数如果带有返回值的情况
import time
def timmer(func):
# print('func timmer')
def wrapper(*args,**kwargs):
print('打印实参:',args,kwargs)
start_time
=time.time()
func(
*args,**kwargs)
stop_time
=time.time()
print("the function run time is %s:" % (stop_time-start_time))
return wrapper
@timmer
# test1=timmer(test1)
def test1(name,age,address):
time.sleep(
1)
print('in the test1 func, name is %s,age is %d,address is %s' % (name,age,address))
test1(
'alex',18,'nuber001')
# @timmer
#
test1=timmer(test1)
def test3(name,age,sex,salary):
time.sleep(
1)
# print('in the test1 func, name is %s,age is %d,sex is %s,salary is %d' % (name,age,sex,salary))
return (name,age,sex,salary)
print(test3('alex',18,'man',3000))
输出如下:
/usr/bin/python3.5 /root/PycharmProjects/S14/day04/decoreate02.py
打印实参: ('alex', 18, 'nuber001') {}
in the test1 func, name is alex,age is 18,address is nuber001
the function  run time is 1.0009291172027588:
('alex', 18, 'man', 3000)
Process finished with exit code 0
函数test1,没有返回值,装饰后正常输出
函数test2,带有返回值,没有装饰,打印了实参元组
如果test2装饰后呢?
import time
def timmer(func):
# print('func timmer')
def wrapper(*args,**kwargs):
print('打印实参:',args,kwargs)
start_time
=time.time()
func(
*args,**kwargs)
stop_time
=time.time()
print("the function run time is %s:" % (stop_time-start_time))
return wrapper
@timmer
# test1=timmer(test1)
def test1(name,age,address):
time.sleep(
1)
print('in the test1 func, name is %s,age is %d,address is %s' % (name,age,address))
test1(
'alex',18,'nuber001')
@timmer
# test1=timmer(test1)
def test3(name,age,sex,salary):
time.sleep(
1)
# print('in the test1 func, name is %s,age is %d,sex is %s,salary is %d' % (name,age,sex,salary))
return (name,age,sex,salary)
print(test3('alex',18,'man',3000))
输出如下:
打印实参: ('alex', 18, 'nuber001') {}
in the test1 func, name is alex,age is 18,address is nuber001
the function  run time is 1.0010955333709717:
打印实参: ('alex', 18, 'man', 3000) {}
the function  run time is 1.0010552406311035:
None
带有返回值的test2被装饰后,变成了None,改变了原函数的输出,这是不允许的

给包装函数中的调用原函数增加赋值和返回值:
import time
def timmer(func):
# print('func timmer')
def wrapper(*args,**kwargs):
print('打印实参:',args,kwargs)
start_time
=time.time()
res
=func(*args,**kwargs) #用一个变量保存原函数输出值
stop_time
=time.time()
print("the function run time is %s:" % (stop_time-start_time))
return res #返回原函数结果给wrapper
return wrapper
@timmer
# test1=timmer(test1)
def test1(name,age,address):
time.sleep(
1)
print('in the test1 func, name is %s,age is %d,address is %s' % (name,age,address))
test1(
'alex',18,'nuber001')
@timmer
# test1=timmer(test1)
def test3(name,age,sex,salary):
time.sleep(
1)
# print('in the test1 func, name is %s,age is %d,sex is %s,salary is %d' % (name,age,sex,salary))
return (name,age,sex,salary)
print(test3('alex',18,'man',3000))
输出如下:
打印实参: ('alex', 18, 'nuber001') {}
in the test1 func, name is alex,age is 18,address is nuber001
the function  run time is 1.0010664463043213:
打印实参: ('alex', 18, 'man', 3000) {}
the function  run time is 1.0010428428649902:
('alex', 18, 'man', 3000)
  • 语法糖带参数情况,index使用本地验证,home使用ldap验证,同一个装饰器,如何实现不同的验证方式?
  • def deco(func):
    def wrapper(username,password):
    print('I am 包装1')
    func(username,password)
    print('I am 包装2')
    return wrapper
    @deco(auth_type
    ='local')
    def index(username,password):
    print('welcome to index page',username)
    index(
    'yzw','secret')
    @deco(auth_type
    ='ldap')
    def home(username,password):
    print('welcome to home page',username)
    home(
    'alex','secret')

     

  • 再嵌套一层函数
  • def auth(auth_type):
    def out_wrapper(func):
    def wrapper(username,password):
    if auth_type == 'local':
    print('via local certification')
    print('I am 包装前')
    func(username,password)
    print('I am 包装后')
    elif auth_type == 'ldap':
    print('嘻嘻嘻嘻')
    func(username, password)
    print('via ldap certfication')
    return wrapper
    return out_wrapper
    @auth(auth_type
    ='local')
    #1.全部不带参数,相当于index=deco(index)
    #
    2.被装饰函数带参数,相当于index=deco(index)=wrapper ->index(args)=wrapper(args)
    #
    3.语法糖带参数,就需要多嵌套一层外层函数,将语法糖的参数传递给这一外层函数
    def index(username,password):
    print('welcome to index page <local>',username)
    index(
    'yzw','secret')
    @auth(auth_type
    ='ldap')
    def home(username,password):
    print('welcome to home page <ldap>',username)
    home(
    'alex','secret')
    输出如下:
    /usr/bin/python3.5 /root/PycharmProjects/S14/day04/bin/decorate04.py
    via local certification
    I am 包装前
    welcome to index page <local> yzw
    I am 包装后
    嘻嘻嘻嘻
    welcome to home page <ldap> alex
    via ldap certfication

    Process finished with exit code 0

  • python3.x 基础三:装饰器python3.x 基础三:装饰器
    #/usr/bin/env python
    #
    -*- coding: utf-8 -*-
    #
    Author:jenvid.yang
    def auth(auth_type):
    def out_wrapper(func):
    def wrapper(*args,**kwargs):
    if auth_type == 'local':
    print('via local certification')
    func(
    *args,**kwargs)
    elif auth_type == 'ldap':
    print('via ldap certfication')
    func(
    *args, **kwargs)
    return wrapper
    return out_wrapper
    @auth(auth_type
    ='local')
    #1.全部不带参数,相当于index=deco(index)
    #
    2.被装饰函数带参数,相当于index=deco(index)=wrapper ->index(args)=wrapper(args)
    #
    3.语法糖带参数,就需要多嵌套一层外层函数,将语法糖的参数传递给这一外层函数
    def index(username,password):
    print('welcome to index page <local>',username)
    index(
    'yzw','secret')
    @auth(auth_type
    ='ldap')
    def home(username,password):
    print('welcome to home page <ldap>',username)
    home(
    'alex','secret')
    改成万能参数
  • python3.x 基础三:装饰器python3.x 基础三:装饰器
    # /usr/bin/env python
    #
    -*- coding: utf-8 -*-
    #
    Author:Jenvid.yang
    import time
    user,passwd
    ='alex','abc123'
    def auth(auth_type):
    print('auth func:',auth_type)
    def outer_wrapper(func):
    def wrapper(*args,**kwargs):
    print('wrapper func args:',*args,**kwargs)
    if auth_type == 'local':
    username
    =input('username:').strip()
    password
    =input('password:').strip()
    if user == username and passwd == password:
    print('\033[32;1m user has passed authentication \033[0m')
    res
    = func(*args,**kwargs)
    print('xxxxx')
    return res
    else:
    exit(
    '\033[31;1m invalid username or password \033[0m')
    elif auth_type == 'ldap':
    print('xxxx')
    return wrapper
    return outer_wrapper
    # @auth
    def index():
    print('welcome to index page')
    @auth(auth_type
    ='local') #home=wrapper()
    def home():
    print('welcom to home page')
    return ('from home')
    # 函数结果被改变
    @auth(auth_type='ldap')
    def bbs():
    print('welcome to bbs page')

    index()
    print(home()) # wrapper()
    bbs()
    课堂分析源码