使用sqlmodel实现唯一性校验2，插入之前检查是否已存在-编程知识

使用sqlmodel实现唯一性校验2，插入之前检查是否已存在

虽然之前添加唯一性校验的方法能够解决数据唯一的问题，但是如果忘了处理异常，则可能会导致程序崩溃。

在此基础上，我们可以在插入数据之前检查该数据是否已存在。

原来的代码：

from sqlmodel import Field, Session, SQLModel, create_engine# 声明模型
class User(SQLModel, table=True):id: int | None = Field(default=None, primary_key=True)# 不能为空，必须唯一name: str = Field(nullable=False, unique=True)age: int | None = None# 创建引擎
engine = create_engine("sqlite:///database.db", echo=True)# 创建表
SQLModel.metadata.create_all(engine)# 创建两个名字重复的用户
zs_1 = User(name="张三", age=22)
zs_2 = User(name="张三", age=23)with Session(engine) as session:session.add(zs_1)session.add(zs_2)session.commit()

我们改写如下：

from sqlalchemy import select
from sqlmodel import Field, Session, SQLModel, create_engine# 声明模型
class User(SQLModel, table=True):id: int | None = Field(default=None, primary_key=True)# 不能为空，必须唯一name: str = Field(nullable=False, unique=True)age: int | None = None# 创建引擎
engine = create_engine("sqlite:///database.db", echo=True)# 初始化表
SQLModel.metadata.drop_all(engine)
SQLModel.metadata.create_all(engine)# 添加用户的方法
def add_user(name: str, age: int):user = User(name=name, age=age)with Session(engine) as session:# 检查是否已存在statement = select(User).where(User.name == name)results = session.exec(statement)db_user = results.first()print("查询结果：", db_user)if db_user is not None:print(f"用户 {name} 已存在")return# 执行添加session.add(user)session.commit()# 创建两个名字重复的用户
add_user("张三", 23)
add_user("张三", 24)# 查询所有用户
with Session(engine) as session:statement = select(User)results = session.exec(statement).all()print(results)