虽然之前添加唯一性校验的方法能够解决数据唯一的问题,但是如果忘了处理异常,则可能会导致程序崩溃。
在此基础上,我们可以在插入数据之前检查该数据是否已存在。
原来的代码:
python
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()
我们改写如下:
python
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)
执行结果如下:
第一次的时候,查询出来的是个None,第二个则提示已存在。
用这种方案,能够更安全的解决数据唯一性的问题。
当然,在模型中声明唯一性约束也是必要的,可以方案用户跳过接口,直接以操作数据库的方式对表进行重复数据写入。