Почему await con.commit() вызывает ошибку ValueError?

Question

[Hikanosu]

Aiomysql использую в связке с Aiogram. Эксперементальным путём выяснил, что строка await con.commit() вызывает ошибку приведённую в конце. Но почему?

async def __query(self, *args, **kwargs):
    con = await aiomysql.connect(
            host=self.__db_host,
            user=self.__db_login,
            password=self.__db_pass,
            db=self.__db_name,
    )
    cur = await con.cursor()
    await cur.execute(*args, **kwargs)
    res = await cur.fetchall()
    await con.commit()
    await cur.close()

Ошибка что возникает:

Traceback (most recent call last):
  File "/home/iindin/.local/lib/python3.9/site-packages/aiogram/dispatcher/handler.py", line 117, in notify
    response = await handler_obj.handler(*args, **partial_data)
GeneratorExit

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/iindin/.local/lib/python3.9/site-packages/aiogram/dispatcher/dispatcher.py", line 273, in process_update
    return await self.callback_query_handlers.notify(update.callback_query)
  File "/home/iindin/.local/lib/python3.9/site-packages/aiogram/dispatcher/handler.py", line 127, in notify
    current_handler.reset(ctx_token)
ValueError: <Token var=<ContextVar name='current_handler' at 0x7f7ab7a7df90> at 0x7f7ab5caa800> was created in a different Context
Exception ignored in: <coroutine object Handler.notify at 0x7f7ab5c42140>
Traceback (most recent call last):
  File "/home/iindin/.local/lib/python3.9/site-packages/aiogram/dispatcher/handler.py", line 127, in notify
    current_handler.reset(ctx_token)
ValueError: <Token var=<ContextVar name='current_handler' at 0x7f7ab7a7df90> at 0x7f7ab5c7bc40> was created in a different Context
    return res

Answer 1

[Vindicar]

Посмотри базовый пример, может, лучше переделать по его образу и подобию?

pool = await aiomysql.create_pool(host='127.0.0.1', port=3306,
                                      user='root', password='',
                                      db='mysql', loop=loop)
    async with pool.acquire() as conn:
        async with conn.cursor() as cur:
            await cur.execute("SELECT 42;")
            print(cur.description)
            (r,) = await cur.fetchone()
            assert r == 42
    pool.close()
    await pool.wait_closed()

Comments

[Hikanosu]

Хорошо, но откуда взять loop? Его уже использует же aiogram.

[Vindicar]

Hikanosu, я думаю, aiogram использует loop = asyncio.get_event_loop().
Это логично - в норме в асинхронном приложении только один реактор, и все асинхронные библиотеки должны использовать его.

Так что и ты должен делать то же самое.

[Hikanosu]

Vindicar, Т.е нужно заранее создать loop. И отправлять его и в aiogram и aiomysql?

[Vindicar]

Hikanosu, нет. asyncio.get_event_loop() создаёт loop, если его ещё нет. Если он был создан, она возвращает существующий.
Так что если в нескольких местах кода независимо вызывается get_event_loop(), они всё равно будут использовать один и тот же loop. Так что либо просто передай результат вызова get_event_loop() в aiomysql, либо параметр loop можно просто опустить, и это будет поведение по умолчанию.

loop – is an optional event loop instance, asyncio.get_event_loop() is used if loop is not specified.

[Hikanosu]

Vindicar, Не помогло, хотя переделал(

[Vindicar]

Hikanosu, Покажи новый код тогда? Если возможно, минимальный полный пример.

[Hikanosu]

Vindicar,
Использую сокращения, так-как код очень большой и закавыристый. Если я правильно заметил, ошибка возникает только при UPDATE и возможно INSERT,

...
    async def __query(self, *args, **kwargs):
        pool = await aiomysql.create_pool(
            host=self.__db_host,
            port=self.__port,
            user=self.__db_login,
            password=self.__db_pass,
            db=self.__db_name
        )
        res = None
        async with pool.acquire() as conn:
            async with conn.cursor() as cur:
                await cur.execute(*args, **kwargs)
                res = await cur.fetchall()
                await conn.commit()
        pool.close()
        await pool.wait_closed()
        return res
 ...
    async def execute(self, *args, **kwargs):
        return await self.__query(*args, **kwargs)
...
    async def func(self, args...):
        ...
        await db.execute(f'UPDATE table_name SET column_name = {new_data} WHERE user_id = "{self.user_id}"')
        ...

[Vindicar]

Hikanosu, Что насчёт DELETE?
Потому что fetchall() реально имеет смысл только для SELECT.
И честно, я не очень понимаю смысл функции __query(). Она вносит целую кучу неэффективностей: тут и соединение каждый раз, тут и fetchall(), который может аукнуться если выбрано много данных...
Почему не создать соединение единожды в начале работы бота, и не использовать его?

[Hikanosu]

Vindicar, 0. Мне кажется ошибка возникает, когда какие-то данные сохраняются в базе данных, При этом не всегда. С DELETE и CREATE проблем не обнаружил. 1. __query() это базавая функция на которой основано около 7 функций, понятное дело в проектах она более большая. Но выше, это минимальный её размер, при котором смог воспроизвести эту ошибку. 2. Думал об этом, но так и не придумал как от туда забирать данные при SELECT. 3. fetchall() - забираю много данных часто плюс чтобы всегда был один результат функции без сюпризов.