异步装饰器装饰协程(异步方法)坑记录

预备知识

想了解装饰器原理的可以查看该文章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，可以发现结果只允许了装饰器（装饰器会被预加载执行），没运行被装饰协程：