526互联

async装饰器

发布时间 2023-08-03 16:14:48作者: lxd670

1.异步无参装饰器

import time
from functools import wraps

def total_time(func):
  @wraps(func)
  async def wrapper(*args, **kwargs):
    start = time.time()
    res = await func(*args, **kwargs)
    print(f'RUN TIME: {time.time() - start}')
    return res
  return wrapper

2.异步有参数装饰器

import time
from functools import wraps

def set_num(desc: str):
  def outer(func):
    @wraps(func)
    async def wrapper(*args, **kwargs):
      start = time.time()
      res = await func(*args, **kwargs)
      print(f'RUN TIME - {desc}: {time.time() - start}')
      return res
    return wrapper
  return outer