Control set of unhandled delayed calls

gistfile1.txt

import asyncio
import functools


handles = set()


def call_later(delay, cb, *args, loop=None):

    if loop is None:
        loop = asyncio.get_event_loop()

    @functools.wraps(cb)
    def wrapper(*args):
        try:
            return cb(*args)
        finally:
            handles.remove(h)

    h = loop.call_later(delay, wrapper, *args)
    handles.add(h)
    return h


def cb(arg):
    print('Call', arg)


loop = asyncio.get_event_loop()

call_later(0.01, cb, 'a')
call_later(0.02, cb, 'b')
call_later(0.1, cb, 'c')

print(handles)

loop.run_until_complete(asyncio.sleep(0.05))

print(handles)