SQL Model 本身并不直接支持事务处理,但是你可以在 SQL Model 中使用数据库的事务处理功能。以下是一个使用 SQLAlchemy(SQL Model 的底层 ORM)进行事务处理的示例:
首先,确保你已经安装了 SQLAlchemy:
pip install sqlalchemy
然后,创建一个简单的模型:
from sqlalchemy import create_engine, Column, Integer, String from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import sessionmaker Base = declarative_base() class User(Base): __tablename__ = 'users' id = Column(Integer, primary_key=True) name = Column(String) age = Column(Integer) # 替换为你的数据库连接字符串 DATABASE_URL = "sqlite:///test.db" engine = create_engine(DATABASE_URL) Base.metadata.create_all(engine) Session = sessionmaker(bind=engine)
现在,你可以使用上下文管理器和 session.begin()
来实现事务处理:
def add_user(name, age): session = Session() try: with session.begin(): user = User(name=name, age=age) session.add(user) # 如果需要,你可以在这里添加更多的数据库操作 except Exception as e: print(f"Error: {e}") session.rollback() else: print("User added successfully") finally: session.close() # 添加一个用户 add_user("Alice", 30)
在这个示例中,我们使用 with session.begin()
上下文管理器来确保在事务中执行所有操作。如果在事务中发生任何错误,session.rollback()
将回滚事务。如果一切顺利,事务将自动提交。最后,我们使用 finally
子句确保会话在操作完成后关闭。