user,passwd='coke','123456' #服务器验证,本地验证 def auth(auth_type): #如果decorator本身需要传入参数,以便解决不同函数需求,用一个装饰器的情况 def out_wrapper(func): def wrapper(*args, **kwargs): #可以解决被加装饰器的函数有多个参数的情况 if auth_type=='local': username=input('input your username').strip() password=input('input your password').strip() if user == username and passwd ==password: print('\033[32;1mUser has passed authentication\033[0m') res=func(*args,**kwargs) #可以解决被加装饰器的函数存在return的情况 print('------after authentication------') return res #考虑到func函数有返回值 else: exit('\033[31;1mInvalid username or password\033[0m') elif auth_type=='ldap': print('this is in the bbs') return wrapper return out_wrapper def index(): print('welcome to index page') @auth(auth_type="local") #装饰器即为函数 def home():#-------->旧版本函数==func print('welcome to home page') return 'from home' @auth(auth_type="ldap") def bbs():#-------->旧版本函数==func print('welcome to bbs page') home() #-------->新版本函数==wrapper 也是auth('local')(home) bbs() #-------->新版本函数==wrapper
Python装饰器3
最新推荐文章于 2024-07-17 09:15:39 发布