装饰器通用模型
|
1
2
3
4
5
|
def wrapper(fn): def inner(*args, **kwargs): ret = fn(*args, **kwargs) return ret return inner |
装饰器几个关键点
|
1
2
3
4
5
6
7
|
"""1.函数可以当参数传递2.函数可以作为返回值进行返回3.函数名称可以当成变量一样进行赋值操作装饰器本质上是个闭包在不改变原有函数调用的情况下,给函数增加新的功能""" |
举个例子
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#!/usr/bin/pythondef admin(game): def inner(*args, **kwargs): # inner添加了参数,args 一定是个元组 kwargs 一定是字典 print('打开Wg') result = game(*args, **kwargs) # * ** 表示把args元组和kwargs打散成位置参数,关键字参数传递进去 print('关闭Wg') return result return inner@admindef play_dnf(username, password): print(f'开始玩DNF,账号:{username},密码:{password}') print('刀斩肉身,心斩灵魂') return '掉落:戮蛊的哀鸣炮'@admindef play_wow(race, occupation, server_name, camp): print(f'开始玩魔兽世界,种族:{race},职业:{occupation},服务器:{server_name},阵营:{camp}') print('为了辛多雷的荣耀') return '掉落:灰烬使者'if __name__ == '__main__': ret1 = play_dnf('大马猴', '888888') print(ret1) ret2 = play_wow('血精灵', '圣骑士', '回音山', '部落') print(ret2) |
这代码还是很好懂的，我就不解释了，然后是执行结果
|
1
2
3
4
5
6
7
8
9
10
11
12
|
python demo.py打开Wg开始玩DNF,账号:大马猴,密码:888888刀斩肉身,心斩灵魂关闭Wg掉落:戮蛊的哀鸣炮打开Wg开始玩魔兽世界,种族:血精灵,职业:圣骑士,服务器:回音山,阵营:部落为了辛多雷的荣耀关闭Wg掉落:灰烬使者Process finished with exit code 0 |
一个函数被多个装饰器装饰，又将如何执行呢？
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
#!/usr/bin/pythondef wrapper1(fn): def inner(*args, **kwargs): print('这是w1进入') ret = fn(*args, **kwargs) print('这是w1出去') return ret return innerdef wrapper2(fn): def inner(*args, **kwargs): print('这是w2进入') ret = fn(*args, **kwargs) print('这是w2出去') return ret return inner@wrapper1@wrapper2def target(): print('我是目标')if __name__ == '__main__': target() |
直接给出执行顺序
一个函数被多个装饰器装饰的执行顺序
|
1
|
# w1 w2 target w2 w1 |
带参数的装饰器
装饰器的语法允许我们在调用时，提供其它参数，比如@decorator(a)。这样，就为装饰器的编写和使用提供了更大的灵活性。
(在上面又套了一层函数)
比如，我们可以在装饰器中指定日志的等级，因为不同业务函数可能需要的日志级别是不一样的。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
def use_logging(level): def decorator(func): def wrapper(*args, **kwargs): if level == "warn": logging.warn("%s is running" % func.__name__) elif level == "info": logging.info("%s is running" % func.__name__) return func(*args) return wrapper return decorator@use_logging(level="warn")def foo(name='foo'): print("i am %s" % name)foo() |
类装饰器
没错，装饰器不仅可以是函数，还可以是类，相比函数装饰器，类装饰器具有灵活度大、高内聚、封装性等优点。使用类装饰器主要依靠类的__call__方法，当使用 @ 形式将装饰器附加到函数上时，就会调用此方法。
|
1
2
3
4
5
6
7
8
9
10
11
|
class Foo(object): def __init__(self, func): self._func = func def __call__(self): print ('class decorator runing') self._func() print ('class decorator ending')@Foodef bar(): print ('bar')bar() |
总结
本篇文章就到这里了，希望能够给你带来帮助，也希望您能够多多关注服务器之家的更多内容!
原文链接：https://blog.csdn.net/hans99812345/article/details/123673788
