Source code for caspia.meadow.cli.utils

import asyncio
import functools
import logging


def measure_time_async(func):
    logger = logging.getLogger('performance')

    @functools.wraps(func)
    async def wrapped(*args, **kwargs):
        loop = asyncio.get_event_loop()
        start = loop.time()
        await func(*args, **kwargs)
        end = loop.time()
        logger.info('%r took %s ms', func, int((end - start) * 1000))

    return wrapped