python有参装饰器详解_24、Python之有参装饰器

一、前置知识

1.1 无参装饰器模板

def outter(func):

def wrapper(*args,**kwargs):

# 1、调用原函数

# 2、给原函数增加新的功能

res=func(*args,**kwargs)

# 3、给原函数增加新的功能

return res

return wrapper

@outter # index=outter(index)

def index(x,y):

"""这个是主要功能"""

print(x,y)

print(index) # .wrapper at 0x0000027FF77B20D0>

print(index.__name__) # wrapper，实际上还是wrapper函数

print(help(index)) # wrapper的文档注释如下:

'''

Help on function wrapper in module __main__:

wrapper(*args, **kwargs)

这个是主要功能

None

'''

# 经装饰之后的函数，还是没有完全伪装的跟原函数一样，需要继续优化

1.2 装饰器wraps

#偷梁换柱，即将原函数名指向的内存地址偷梁换柱成wrapper函数，所以应该将wrapper做的跟原函数一样才行

from functools import wraps # 引入functools模块下的装饰器wraps

def outter(func):

@wraps(func) # functools模块下提供的装饰器wraps，可以实现保留原函数属性的操作

def wrapper(*args,**kwargs):

# 1、调用原函数

# 2、给原函数增加新的功能

res=func(*args,**kwargs)

# 3、给原函数增加新的功能

return res

return wrapper

二、有参装饰器

2.1 有参装饰器的实现

若在无参装饰器的基础上，再实现一个用来为被装饰对象添加认证功能的装饰器，需要在deco外部再包一层函数auth，用来专门接受额外的参数，这样便保证了在auth函数内无论多少层都可以引用到。

def auth(db_type):

def deco(func):

def wrapper(*args, **kwargs):

name = input('your name>>>: ').strip()

pwd = input('your password>>>: ').strip()

if db_type == 'file':

print('基于文件的验证')

if name == 'egon' and pwd == '123':

res = func(*args, **kwargs) # index(1,2)

return res

else:

print('user or password error')

elif db_type == 'mysql':

print('基于mysql的验证')

elif db_type == 'ldap':

print('基于ldap的验证')

else:

print('不支持该db_type')

return wrapper

return deco

@auth(db_type='file') # @deco # index=deco(index) # index=wrapper

def index(x, y):

print('index->>%s:%s' % (x, y))

@auth(db_type='mysql') # @deco # home=deco(home) # home=wrapper

def home(name):

print('home->>%s' % name)

@auth(db_type='ldap') # 账号密码的来源是ldap

def transfer():

print('transfer')

index(1, 2)

home('egon')

transfer()

2.2 有参装饰器模板

def 有参装饰器(x,y,z):

def outter(func):

def wrapper(*args, **kwargs):

# 1、为其增加新功能

res = func(*args, **kwargs)

return res

# 2、调用原函数后为其增加其他新功能

return wrapper

return outter

@有参装饰器(1,y=2,z=3)

def 被装饰对象():

pass