Async que se faz - TDC 2018

async_rotina.py

# Código usado na palestra "Async que se faz - asyncio na prática"
# no TDC 2018 em Florianópolis
# Creative Commons Attribution 4.0 International license <https://creativecommons.org/licenses/by/4.0/>

import asyncio
import random
from time import sleep

from utils import UPDATE_AVG, async_request, timeit


async def buscar_pedidos_no_banco():
    print('- Buscar pedidos no banco')
    return range(18000)

async def checar_pedido_no_correios(pedido):
    await async_request(pedido, 'Correios')

async def atualizar_pacote_no_banco(pedido):
    print('- {}: Atualiza pacote'.format(pedido))

async def notificar_usuario(pedido):
    await async_request(pedido, 'Telegram')

async def verificar_pedidos(pedido):
    await checar_pedido_no_correios(pedido)

    if random.choice(UPDATE_AVG):
        await atualizar_pacote_no_banco(pedido)
        await notificar_usuario(pedido)

async def main():
    pedidos = await buscar_pedidos_no_banco()

    tarefas = []
    for pedido in pedidos:
        tarefas.append(verificar_pedidos(pedido))

    await asyncio.wait(tarefas)

loop = asyncio.get_event_loop()
loop.run_until_complete(timeit(main()))