Зарегистрируйтесь, чтобы продолжить обучение

Контекстные менеджеры Python: SQL

Посмотрите на код ниже и попробуйте ответить на вопрос, что с ним не так?

sql = "INSERT INTO users (username, phone) VALUES ('tommy', '123456789');"
cursor = conn.cursor()
cursor.execute(sql)
conn.commit()
cursor.close()

Этот код написан из предположения, что вызов cursor.execute(sql) никогда не приведет к ошибке. А если приведет? Тогда возникнет исключение и метод cursor.close() никогда не будет вызван. Для обработки таких случаев можно использовать стандартный синтаксис try-except

sql = "INSERT INTO users (username, phone) VALUES ('tommy', '123456789');"
cursor = conn.cursor()
try:
    cursor.execute(sql)
except:
    print("Cannot execute SQL-query")
finally:
    cursor.close()

Теперь, если возникнет любая ошибка, то курсор правильно закроется. В теории это выглядит нормально, но на практике, об этом сложно постоянно помнить и программисты регулярно забывают добавлять блок finally. Да и весь синтаксис выглядит громоздким. Потому создатели библиотеки добавили синтаксис контекстных менеджеров для объектов соединения и курсора:

conn = psycopg2.connect("postgresql://user:password@host:port/hexlet_test")
sql = "INSERT INTO users (username, phone) VALUES ('tommy', '123456789');"
with conn.cursor() as curs:
    curs.execute(sql)

conn.close()  # закрываем соединение

Важное замечание, что несмотря на то, что привычные вам контекстные менеджеры освобождают ресурсы после выхода из блока with, здесь соединение не закрывается. И потому важно закрыть его принудительно самим.


Дополнительные материалы

  1. Официальная документация

Для полного доступа к курсу нужен базовый план

Базовый план откроет полный доступ ко всем курсам, упражнениям и урокам Хекслета, проектам и пожизненный доступ к теории пройденных уроков. Подписку можно отменить в любой момент.

Получить доступ
1000
упражнений
2000+
часов теории
3200
тестов

Открыть доступ

Курсы программирования для новичков и опытных разработчиков. Начните обучение бесплатно

  • 130 курсов, 2000+ часов теории
  • 1000 практических заданий в браузере
  • 360 000 студентов
Отправляя форму, вы принимаете «Соглашение об обработке персональных данных» и условия «Оферты», а также соглашаетесь с «Условиями использования»

Наши выпускники работают в компаниях:

Логотип компании Альфа Банк
Логотип компании Aviasales
Логотип компании Yandex
Логотип компании Tinkoff