Patch asyncio handles to measure task runtime

tasktiming.py

import time
import asyncio


def task_timing_patch():

    class TaskTimingHandle(asyncio.events.Handle):

        def _add_runtime(self, value):
            try:
                instance = self._callback.__self__
            except AttributeError:
                return
            if not isinstance(instance, asyncio.Task):
                return
            try:
                runtime = instance.runtime
            except AttributeError:
                runtime = 0.
            instance.runtime = runtime + value

        def _run(self, *args, **kwargs):
            start = time.time()
            try:
                return super()._run(*args, **kwargs)
            finally:
                self._add_runtime(time.time() - start)

    asyncio.events.Handle = TaskTimingHandle


async def test():
    time.sleep(0.2)
    await asyncio.sleep(0.8)


async def main():
    task = asyncio.create_task(test())
    await task
    print(task.runtime)


if __name__ == "__main__":
    task_timing_patch()
    asyncio.run(main(), debug=True)

That's so hackish. Hahaha~