@async_agnostic decorator for running async functions at the REPL

async_agnostic.py

import asyncio

def async_agnostic(corofn):
    @wraps(corofn)
    def wrapper(*args, **kwargs):
        with suppress(RuntimeError):
            loop = asyncio.get_running_loop()
            if loop.is_running():
                return corofn(*args, **kwargs)
        return asyncio.run(corofn(*args, **kwargs))

    return wrapper

License: CC0-1.0