SQLAlchemy会话与事务控制

1、简介:

SQLAlchemy 的 session 是用于管理数据库操作的一个像容器一样的东西. 模型实例对象本身独立存在, 而要让其修改(创建)生效, 则需要把它们加入某个 session 。同时你也可以把模型实例对象从 session 中去除。

被 session 管理的实例对象:

session.commit() 直接将数据库内存中的数据提交到数据库,此操作会内部调用session.flush(),其余的事务可以访问最新的数据;

session.rollback() 是回滚当前事务的变更数据操作;

session.flush() 的作用是在事务管理内与数据库发生交互, 对应的实例状态被反映到数据库,比如自增 ID 被填充上值,但是数据库中当前事务的数据值并未更新上;相当于预提交,等于提交到数据库内存,还未写入数据库文件;deletions and modifications to the database as INSERTs, DELETEs, UPDATE;

session.merge(obj) 查询更新操作;就是更新之前先查询,如果没有自动插入;

2、操作事务的几种模式

from sqlalchemy import create_engine, Column, String, Integer
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.exc import SQLAlchemyError


DB_CONNECT_STRING = 'mysql+pymysql://root:@localhost/my_test?charset=utf8'
# engine = create_engine(DB_CONNECT_STRING, echo=True)   # echo=True会打印出来orm执行的sql语句
engine = create_engine(DB_CONNECT_STRING)
DB_Session = sessionmaker(bind=engine)
session = DB_Session()
  •  简易更新操作

try:
    user = session.Query(User).first()
    user.name = u'改名字
    session.commit()
except:
    session.rollback()
  •  复杂事务嵌套上下文

注意:begin_nested创建一个嵌套事务,事务的提交最后由外层的commit执行,with执行完毕,内层session自动托管到外层事务上

t1 = Admin(username='test', password='123456')       # 生成admin表要插入的一条数据
t2 = Admin(username='test1', password='abcdef')      # 生成admin表要插入的一条数据

session.add(t2)

try:
    with session.begin_nested():
        session.add(t1)  # 或使用session.merge(t1),表示查询更新操作
except Exception as e:
    print(e)
    session.rollback()

session.commit()
  • 简易嵌套
t1 = Admin(username='test1', password='123456')       # 生成admin表要插入的一条数据
t2 = Admin(username='test2', password='abcdef')      # 生成admin表要插入的一条数据

session.add(t2)

# 创建一个子嵌套事务,第一个commit只是将子事务的数据托管到父事务,并未提交到数据库
session.begin_nested()
session.add(t1)
session.commit()

# 父事务执行提交,才真正将t1,t2提交到数据库
session.commit()

 

<think>我们正在讨论SQLAlchemy中开启事务时`begin`参数的使用。根据引用[1]和引用[2],事务管理在SQLAlchemy中是通过会话Session)来处理的。在创建会话时,可以通过参数控制事务的行为。 在SQLAlchemy中,当我们创建一个会话Session)时,可以使用参数`begin`来控制事务的开启时机。具体来说: - 如果`begin=True`(默认值),则在会话创建后不会立即开启一个新的事务,而是在会话首次需要数据库连接时自动开启事务(例如执行第一个查询或操作时)。 - 如果`begin=False`,则会话在创建时不会自动开启事务,需要显式调用`session.begin()`来开启事务。 这种设计允许更灵活地控制事务边界。例如,在需要将多个操作组合在一个事务中,但又不想在创建会话后立即开启事务的情况下,可以使用`begin=False`。 ### 详细说明 1. **默认行为(`begin=True`)**: ```python from sqlalchemy.orm import Session engine = create_engine("sqlite://") session = Session(engine) # 等同于 Session(engine, begin=True) # 此时事务并未立即开启,但在第一次操作(如query)时会自动开启 ``` 2. **设置`begin=False`**: ```python session = Session(engine, begin=False) # 此时会话没有事务,需要显式开启 session.begin() ``` 3. **使用上下文管理器(推荐)**: 无论`begin`参数如何,都可以使用上下文管理器来管理事务: ```python with Session(engine) as session: # 在进入上下文时,如果begin=True(默认),则自动开始事务 # 执行操作 session.add(some_object) # 退出上下文时,如果没有异常则提交,有异常则回滚 ``` 如果设置`begin=False`,则需要在上下文管理器中显式开启事务: ```python with Session(engine, begin=False) as session: with session.begin(): # 在这个内层上下文中,事务开始 session.add(some_object) # 退出内层上下文时,事务提交(或异常时回滚) ``` 或者显式调用`begin()`和`commit()`/`rollback()`,但上下文管理器更安全。 ### 示例代码 以下示例展示了`begin`参数的不同设置和使用方式: #### 示例1:默认行为(`begin=True`) ```python from sqlalchemy import create_engine from sqlalchemy.orm import Session engine = create_engine("sqlite:///:memory:") # 默认begin=True,在需要时自动开启事务 with Session(engine) as session: # 进入上下文时,并没有立即开启事务 # 第一次操作(如add)时开启事务 user = User(name='Alice') session.add(user) # 在退出上下文时,如果没有异常,则提交事务;有异常则回滚 ``` #### 示例2:设置`begin=False`并显式开启事务 ```python with Session(engine, begin=False) as session: # 此时没有事务,需要显式开启 session.begin() try: user = User(name='Bob') session.add(user) session.commit() # 显式提交 except: session.rollback() # 显式回滚 raise ``` #### 示例3:使用嵌套上下文管理器(`begin=False`时) ```python # 使用嵌套上下文管理器,更简洁 with Session(engine, begin=False) as session: with session.begin(): # 在此块中,事务已开启 user = User(name='Charlie') session.add(user) # 退出内层上下文时,事务已提交(或回滚) ``` ### 注意事项 - 在大多数情况下,使用默认的`begin=True`并结合`with`上下文管理器是最简单且推荐的方式,因为它自动处理了事务的提交和回滚。 - 当需要更细粒度地控制事务(例如,在一个会话中执行多个独立的事务)时,可以使用`begin=False`,然后通过多次调用`session.begin()`来开启多个事务(但注意,SQLAlchemySession通常设计为一次只进行一个事务,多个事务需要以非重叠的方式使用,或者使用“分离”模式)。 - 在设置`begin=False`后,如果不显式开启事务就执行数据库操作,会引发错误(因为不在事务中)。 ### 引用说明 根据引用[1]和引用[2]中关于事务嵌套和管理的描述,SQLAlchemy事务管理是分层的,而`begin`参数正是控制这种层次结构的一种方式[^1][^2]。 ### 相关问题 1. SQLAlchemy中如何实现嵌套事务? 2.SQLAlchemy中,事务的提交和回滚有哪些最佳实践? 3. 如何设置SQLAlchemy会话事务隔离级别? 4. 使用SQLAlchemy时,如何处理长事务? 5. SQLAlchemy中的事务数据库连接(Connection)的事务有何关系?</think>### SQLAlchemy开启事务时`begin`参数的详细说明 在SQLAlchemy中,`begin`参数用于显式控制事务的开启时机和行为,特别是在使用`Session`对象时。以下是关于`begin`参数的详细说明和使用方法: #### 1. `begin`参数的核心作用 - **控制事务启动时机**:决定会话(`Session`)创建时是否**立即开启事务** - **默认行为**:当`begin=True`(默认值)时,会话创建后**不会立即开启事务**,而是在首次数据库操作时自动开启 - **显式控制**:当`begin=False`时,需要手动调用`session.begin()`来启动事务 #### 2. 参数取值及效果 | 取值 | 行为说明 | 适用场景 | |-----------|--------------------------------------------------------------------------|-----------------------------------| | `True` | 会话创建后不立即开启事务,在首次数据库操作时自动开启(默认行为) | 大多数常规操作 | | `False` | 会话创建后不开启事务,必须显式调用`session.begin()`启动事务 | 需要精确控制事务边界的情况 | | `None` | 使用SQLAlchemy的默认行为(等同于`True`) | | | 其他值 | 抛出`ArgumentError`异常 | | #### 3. 使用示例 ```python from sqlalchemy.orm import Session # 场景1:默认行为(begin=True) with Session(engine) as session: # 事务未开启 user = session.get(User, 1) # 首次操作时自动开启事务 user.name = "Updated" session.commit() # 提交事务 # 场景2:显式控制事务begin=False) with Session(engine, begin=False) as session: session.begin() # 手动开启事务 try: user = User(name="New User") session.add(user) session.commit() # 提交事务 except: session.rollback() # 回滚事务 raise # 场景3:使用上下文管理器自动提交 with Session(engine) as session, session.begin(): # 进入块时自动开启事务 user = User(name="Auto Commit") session.add(user) # 退出块时自动提交(异常时自动回滚) ``` #### 4. 事务嵌套处理 当使用嵌套事务时(如引用[2]所述),`begin`参数的行为: ```python with Session(engine) as outer_session: # 外层事务自动开启 with outer_session.begin_nested(): # 创建保存点 # 内层事务操作 user = User(name="Nested") outer_session.add(user) # 外层事务继续 ``` #### 5. 最佳实践 1. **常规操作**:使用默认`begin=True` + 上下文管理器 ```python with Session(engine) as session, session.begin(): # 自动事务管理 ``` 2. **复杂事务**:使用`begin=False` + 显式控制 ```python session = Session(engine, begin=False) session.begin() try: # 多步骤操作 session.commit() except: session.rollback() finally: session.close() ``` 3. **连接重用**:当需要复用现有连接时 ```python with engine.connect() as conn: with Session(bind=conn, begin=False) as session: session.begin() # 使用现有连接开启事务 ``` #### 6. 注意事项 1. 提交/回滚后事务自动结束,需重新开启 2. 使用`begin=False`时忘记调用`session.begin()`会引发`InvalidRequestError` 3. 在异步环境中(如`AsyncSession`),需使用`await session.begin()` 4. 事务隔离级别可通过`session.begin(isolation_level='REPEATABLE_READ')`设置 > 关键点:`begin`参数不决定事务是否存在,而是控制**事务开启的时机和方式**。SQLAlchemy事务管理遵循"会话事务"模型,每个会话默认关联一个事务[^1][^2]。 ### 相关问题 1. SQLAlchemy中如何设置事务的隔离级别? 2.SQLAlchemy中如何处理嵌套事务(如保存点)? 3. 使用`AsyncSession`时事务管理有哪些不同? 4. SQLAlchemy事务回滚机制是如何工作的? 5. 如何在Flask应用中集成SQLAlchemy事务管理?
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值