带参数的装饰器理解无非记住两点:
1.本质不过在基本的装饰器外面再封装一层带参数的函数
2.在使用装饰器语法糖的时候与普通装饰器不同,必须要加()调用,且()内的内容可以省略(当省略时,admin默认为函数is_admin声明时变量"admin")
import functools def is_admin(admin="admin"): def decorated(func): @functools.wraps(func) def wrapper(*args,**kwargs): if kwargs.get("username")!=admin: raise Exception("this user is not allowed to get food") result = func(*args,**kwargs) return result return wrapper return decorated @is_admin(admin="root")#这里也可以@is_admin() 此时admin默认为函数声明时定义的"admin" def barfoo(username="someone"): """do crazy stuff""" print("%s get food" % (username)) if __name__=="__main__": barfoo(username="root")#这里调用的时候必须加上形参的名字
[注意] 如果main中调用 barfoo时不想带形参,可以使用inspect模块的getcallargs方法,该方法返回一个将参数名字和值作为键值对的字典(按照形参和实际传入参数的位置形成键值对)
代码如下:
import functools import inspect def is_admin(admin="admin"): def decorated(func): @functools.wraps(func) def wrapper(*args,**kwargs): func_args = inspect.getcallargs(func,*args,**kwargs) if func_args.get("username")!=admin : raise Exception("this user is not allowed to get food") result = func(*args,**kwargs) return result return wrapper return decorated @is_admin("admin") def barfoo(username="someone"): """do crazy stuff""" print("%s get food" % (username)) if __name__=="__main__": barfoo("admin")
上述两块代码结果都为:
admin get food