有参装饰器与装饰器补充

有参装饰器(需要掌握

由于语法糖@的限制,outter函数只能有一个参数,并且该函数只用来接收被装饰对象的内存地址(也就是两层不够用,需要再来一层进行传参)
山炮玩法:不使用语法糖,第二层传参也能实现功能 例如装饰index函数 index=auth(index,'file')

有参装饰器模板:
def 有参装饰器(x,y,z):
def outter(func):
def wrapper(*args, **kwargs):
res = func(*args, **kwargs)
return res
return wrapper
return outter

装饰器补充:
偷梁换柱,即将原函数名指向的内存地址偷梁换柱成wrapper函数
所以应该将wrapper做的跟原函数一样才行

from functools import wraps

__name__, __doc__等装饰的一毛一样

@wraps(func) 去装饰我们偷梁换柱后的对象

案例
# 一:知识储备
# 由于语法糖@的限制,outter函数只能有一个参数,并且该函数只用来接收被装饰对象的内存地址
# def outter(func):
# # func = 函数的内存地址
# def wrapper(*args,**kwargs):
# res=func(*args,**kwargs)
# return res
# return wrapper
#
# # @outter # index=outter(index) # index=>wrapper
# @outter # outter(index)
# def index(x,y):
# print(x,y)

# 偷梁换柱之后
# index的参数什么样子,wrapper的参数就应该什么样子
# index的返回值什么样子,wrapper的返回值就应该什么样子
# index的属性什么样子,wrapper的属性就应该什么样子==》from functools import wraps


# 山炮玩法:
# def auth(func,db_type):
# 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)
# 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
#
# # @auth # 账号密码的来源是文件
# def index(x,y):
# print('index->>%s:%s' %(x,y))
#
# # @auth # 账号密码的来源是数据库
# def home(name):
# print('home->>%s' %name)
#
# # @auth # 账号密码的来源是ldap
# def transfer():
# print('transfer')
#
#
# index=auth(index,'file')
# home=auth(home,'mysql')
# transfer=auth(transfer,'ldap')
#
# # index(1,2)
# # home('egon')
# # transfer()


# 山炮二
# 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)
# 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
#
# deco=auth(db_type='file')
# @deco # 账号密码的来源是文件
# def index(x,y):
# print('index->>%s:%s' %(x,y))
#
# deco=auth(db_type='mysql')
# @deco # 账号密码的来源是数据库
# def home(name):
# print('home->>%s' %name)
#
# deco=auth(db_type='ldap')
# @deco # 账号密码的来源是ldap
# def transfer():
# print('transfer')
#
#
# index(1,2)
# home('egon')
# transfer()


# 语法糖
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()




# 有参装饰器模板
# def 有参装饰器(x,y,z):
# def outter(func):
# def wrapper(*args, **kwargs):
# res = func(*args, **kwargs)
# return res
# return wrapper
# return outter

# @有参装饰器(1,y=2,z=3)
# def 被装饰对象():
# pass
装饰器补充:
#偷梁换柱,即将原函数名指向的内存地址偷梁换柱成wrapper函数
# 所以应该将wrapper做的跟原函数一样才行
from functools import wraps

def outter(func):
@wraps(func)
def wrapper(*args, **kwargs):
"""这个是主页功能"""
res = func(*args, **kwargs) # res=index(1,2)
return res

# 手动将原函数的属性赋值给wrapper函数
# 1、函数wrapper.__name__ = 原函数.__name__
# 2、函数wrapper.__doc__ = 原函数.__doc__
# wrapper.__name__ = func.__name__
# wrapper.__doc__ = func.__doc__

return wrapper



@outter # index=outter(index)
def index(x,y):
"""这个是主页功能"""
print(x,y)


print(index.__name__)
print(index.__doc__) #help(index)


index(1,2) # wrapper(1,2)
 
原文地址:https://www.cnblogs.com/h1227/p/12559609.html