Метод begin класса sessionmaker
Метод begin класса sessionmaker создает новый объект
сессии и сразу начинает транзакцию. Он не принимает обязательных
параметров, но может принимать именованные аргументы, которые
передаются в конструктор сессии. Возвращаемый объект работает
как контекстный менеджер: при выходе из блока with
транзакция автоматически коммитится, а при исключении -
откатывается.
Синтаксис
sessionmaker_instance.begin()
Пример
Давайте создадим таблицу и добавим в нее одну статью,
используя метод begin:
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import DeclarativeBase, sessionmaker
class Base(DeclarativeBase):
pass
class Article(Base):
__tablename__ = 'articles'
id = Column(Integer, primary_key=True)
title = Column(String)
engine = create_engine('sqlite:///:memory:')
Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
with Session.begin() as session:
session.add(Article(title='article 1'))
with Session.begin() as session:
res = session.query(Article).all()
for article in res:
print(article.title)
Результат выполнения кода:
"article 1"
Пример
Давайте посмотрим, как метод begin откатывает транзакцию
при возникновении исключения:
<+python+>
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.orm import DeclarativeBase, sessionmaker
class Base(DeclarativeBase):
pass
class Article(Base):
__tablename__ = 'articles'
id = Column(Integer, primary_key=True)
title = Column(String)
engine = create_engine('sqlite:///:memory:')
Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
try:
with Session.begin() as session:
session.add(Article(title='article 2'))
raise ValueError('something went wrong')
except ValueError as error:
print(error)
with Session.begin() as session:
res = session.query(Article).count()
print(res)
<-python+>
Результат выполнения кода:
<+python+>
"something went wrong"
"0"
<-python+>
Смотрите также
-
класс
sessionmaker,
который создает фабрику сессий -
метод
configure,
который настраивает параметры фабрики сессий -
метод
__call__,
который создает объект сессии без автоматического начала транзакции -
метод
begin,
который открывает новую транзакцию сессии
