Ошибка __enter__. Как исправить?

Question

[Ornitier]

Пытаюсь закинуть данные в таблицу. Создать её получилось, но при закидывании данных выходит Attributeerror: __enter__

try:
    con = psycopg2.connect(
    host = host,
    user = user,
    password = password,
    database = db_name
    )

    con.autocommit = True
    cur = con.cursor()
    with state.proxy() as data:
        ncity = cur.execute(f"""DELETE FROM native_city WHERE user_id = {user_id}""")
        ncity = cur.execute("""INSERT INTO native_city (user_id, city, area) VALUES ({}, {}, {})""", (data['user_id'], data['citynative'], data['areanative']))
         return ncity
except Exception as _ex:
    print('[INFO] Error while working with PostgreSQL', _ex)
finally:
    if con:
        cur.close()
        con.close()
        print('[INFO] PostgreSQL connection closed')

Comments

[Dimonchik]

ошибку полный трейс давай, да?

[Ornitier]

Dimonchik, Traceback (most recent call last):
File "C:\Users\assoc\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\dispatcher.py", line 415, in _process_polling_updates
File "C:\Users\assoc\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\dispatcher.py", line 235, in process_updates
return await asyncio.gather(*tasks)
File "C:\Users\assoc\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\handler.py", line 116, in notify
response = await handler_obj.handler(*args, **partial_data)
File "C:\Users\assoc\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\dispatcher.py", line 256, in process_update
return await self.message_handlers.notify(update.message)
File "C:\Users\assoc\AppData\Local\Programs\Python\Python310\lib\site-packages\aiogram\dispatcher\handler.py", line 116, in notify
response = await handler_obj.handler(*args, **partial_data)
File "d:\Documents\Python\bots\weather_bot\bot.py", line 104, in select_city
await database.native_city_create(state, user_id)
File "d:\Documents\Python\bots\weather_bot\database.py", line 37, in native_city_create
with state.proxy() as data:
AttributeError: __enter__

[Dimonchik]

во первых native_city_create

во вторых - что возвращает -то state.proxy()

в третьх, конечно, хз когда такие плейсхолдеры были

[Сергей Горностаев]

Ornitier, похоже, что возвращённое вызовом state.proxy() не является контекстным менеджером.

[Vindicar]

Я подозреваю, что state.proxy() - это реализация машины состояний (FSM) для бота. Значит, на момент вызова обработчика состояние не проинициализировано как должно.

Кроме того,

f"""DELETE FROM native_city WHERE user_id = {user_id}"""

не используй f-строки в запросах.

[Ornitier]

Сергей Горностаев, В общем, решил проблему вставив async перед with и переместив все внутренности data в VALUES {}

[Ornitier]

Vindicar, Почему? Какие могут из-за этого быть проблемы в будущем?

[Сергей Горностаев]

Ornitier, во-первых, sql injection, во-вторых, интерполяция строк не проверяет корректность запроса, в-третьих, сформированные таким образом запросы не оптимизируются.

[Ornitier]

Сергей Горностаев, То есть, лучше использовать format? Я правильно понял? Или есть другие способы? Format быстрее f строки?

[Сергей Горностаев]

Ornitier, нет. Использовать надо специально предназначенные для этого средства, как вы это делаете во втором запросе.

[Ornitier]

Сергей Горностаев, Спасибо большое за совет!

[Сергей Горностаев]

Ornitier, кстати если проблему решил асинхронный with, то предполагается асинхронное выполнение, а вы делаете блокирующие запросы в БД. Стоит сменить библиотеку доступа к базе на асинхронную.