python之带有参数的装饰器

徘徊边缘 提交于 2019-12-03 01:42:11

一个小demo

def set_level(level_num):
    def set_func(func):
        def call_func(*args, **kwargs):
            if level_num == 1:
                print("----权限级别1,验证----")
            elif level_num == 2:
                print("----权限级别2,验证----")
            return func()
        return call_func
    return set_func

# 带有参数的装饰器装饰过程分为2步:
# 1. 调用set_level函数,把1当做实参
# 2. set_level返回一个装饰器的引用,即set_func
# 3. 用返回的set_func对test1函数进行装饰(装饰过程与之前一样)
@set_level(1)
def test1():
    print("-----test1---")
    return "ok"

@set_level(2)
def test2():
    print("-----test2---")
    return "ok"


test1()
test2()

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!