Надо ли ставить commit в конструкции with?

Question

[AlmazKayum]

Python 3.7.3
PostgreSQL 10

import psycopg2

class User:

    def __init__(self, database):
        self.conn = psycopg2.connect(database)
        self.cursor = self.conn.cursor()

    def set_address(self, address, user_id):
        with self.conn:
            self.cursor.execute('UPDATE users SET address=%s WHERE user_id=%s', (user_id, address))
            self.conn.commit()

Конструкция with кажется предполагает закрытие соединения с БД, надо ли ставить self.conn.commit() ?

Comments

[Akina]

COMMIT; не приводит к ошибке - поэтому даже от ненужного коммита никаких проблем не будет. Ну будет предупреждение - его можно смело игнорировать.

Answer 1

[galaxy]

Конструкция with кажется предполагает закрытие соединения с БД

не предполагает:

Cursors can be used as context managers: leaving the context will close the cursor.

Cоединение закрывает только метод close() объекта connection (или del на connection).

Делать commit() в вашем случае необходимо, т.к. по умолчанию psycopg открывает транзакцию перед выполнением первого запроса. Другой вариант - работать в режиме autocommit.

Вообще, на будущее, - конструкции с with делают в разных драйверах СУБД разные вещи и не регламентируются DB-API. Так что надо либо внимательно курить документацию, либо их не использовать.