Asyncio / aiopg — как получить connection из корутины?

Question

[un1t]

В aiopg есть вот такой пример:

async with create_engine(user='aiopg',  database='aiopg', host='127.0.0.1', password='passwd') as engine:
    async with engine.acquire() as conn:
        await conn.execute(tbl.insert().values(val='abc'))

А как в своей корутине получить conn, не копипастя эту портянку в каждкую корутину? Ну и пул соединений конечно должен создаваться один раз, а не в каждой корутине свой.

Answer 1

[Абдулла Мурсалов]

async def some_coro(engine: aiopg.sa.Engine):
    async with engine.acquire() as conn:
        await conn.execute(query)

loop = asyncio.get_event_loop()
engine = loop.run_until_complete(aiopg.sa.create_engine(db_url, loop=loop))
loop.run_until_complete(some_coro(engine=engine))
engine.close()
loop.run_until_complete(engine.wait_closed())

Comments

[un1t]

Спасибо! Только я не хочу явно передавать engine в корутину как переменную.
Сделал глобальной переменной.

db.py

loop = asyncio.get_event_loop()
engine = loop.run_until_complete(create_engine(config.DATABASE_URI, echo=config.DEBUG))

coro.py

from db import engine

async def mycoro():
    async with engine.acquire() as conn:
        rows = await conn.execute('select 1')
        for row in rows:
            print(row)