spinner_await.py

#!/usr/bin/env python3

# spinner_await.py

# credits: Example by Luciano Ramalho inspired by
# Michele Simionato's multiprocessing example in the python-list:
# https://mail.python.org/pipermail/python-list/2009-February/538048.html

import asyncio
import itertools


async def spin(msg):  # <1>
    for char in itertools.cycle('|/—\\'):
        status = char + ' ' + msg
        print('\r' + status, end='', flush=True)
        try:
            await asyncio.sleep(.1)  # <2>
        except asyncio.CancelledError:  # <3>
            break
    print('\r', end='', flush=True)


async def slow_function():  # <4>
    # pretend waiting a long time for I/O
    await asyncio.sleep(3)  # <5>
    return 42


async def supervisor(loop):  # <6>
    spinner = loop.create_task(spin('thinking!'))  # <7>
    print('spinner object:', spinner)  # <8>
    result = await slow_function()  # <9>
    spinner.cancel()  # <10>
    return result


def main():
    loop = asyncio.get_event_loop()  # <11>
    result = loop.run_until_complete(supervisor(loop))  # <12>
    loop.close()
    print('Answer:', result)


if __name__ == '__main__':
    main()