想了解装饰器原理的可以查看该文章Python中的函数装饰器和闭包原理
想了解协程原理的可以查看该文章Python中的协程
坑
使用异步装饰器装饰协程比较写法比较简单,但是其中有一个坑,那就是装饰器中必须await method()调用被装饰协程,否则相当于没有装饰。
代码
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @File : test_async_decorators.py
# @Author: itnoobzzy
# @Date : 2021/3/4
# @Desc : 异步装饰器装饰异步方法
import functools
import asyncio
def async_decorators(method):
"""
异步装饰器装饰异步方法
:param method: 被装饰协程(异步方法)
:return:
"""
@functools.wraps(method)
async def wrapper(*args, **kwargs):
print(f'装饰器装饰{__name__}方法')
# 此处必须await 切换调用被装饰协程,否则不会运行该协程
await method(*args, **kwargs)
return wrapper
@async_decorators
async def test_async_method(*args, **kwargs):
print('该协程被运行')
pass
if __name__ == '__main__':
coroutine = test_async_method()
loop = asyncio.get_event_loop()
task = loop.create_task(coroutine)
loop.run_until_complete(task)
装饰器中去掉await method,可以发现结果只允许了装饰器(装饰器会被预加载执行),没运行被装饰协程: