Python 在多个列中实现 sqlalchemy 唯一约束

在本文中，我们将介绍如何使用 Python 的 SQLAlchemy 库在多个列中实现唯一约束。唯一约束是数据库中一种重要的约束条件，它用于确保表中的某些列的组合值是唯一的。

什么是 SQLAlchemy？

SQLAlchemy 是 Python 中一个广泛使用的数据库工具库，提供了一种高层面的 SQL 表达式语言和对象关系映射（ORM）模式的接口。它支持多种数据库后端，并可以方便地进行数据库查询和数据操作。

在单个列中实现唯一约束

在 SQLAlchemy 中，我们可以通过在模型类的列属性上添加 unique=True 参数来实现单列的唯一约束。例如：

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

# 创建数据库引擎和会话工厂
engine = create_engine('sqlite:///test.db')
Session = sessionmaker(bind=engine)
session = Session()

# 创建基类
Base = declarative_base()

# 创建模型类
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    username = Column(String, unique=True)
    email = Column(String, unique=True)

# 创建表格
Base.metadata.create_all(bind=engine)

# 添加数据
user1 = User(username='alice', email='alice@example.com')
session.add(user1)
session.commit()

在上面的示例中，我们在 username 和 email 两个列上分别添加了唯一约束。在创建表格时，SQLAlchemy 会自动生成唯一索引。如果我们尝试插入重复的数据，会抛出 IntegrityError 异常。

在多个列中实现唯一约束

要在多个列中实现唯一约束，我们可以使用 UniqueConstraint 类。这个类接受一个或多个列作为参数，可以在模型类的 __table_args__ 属性中定义。

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

# 创建数据库引擎和会话工厂
engine = create_engine('sqlite:///test.db')
Session = sessionmaker(bind=engine)
session = Session()

# 创建基类
Base = declarative_base()

# 创建模型类
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    username = Column(String)
    email = Column(String)
    __table_args__ = (UniqueConstraint('username', 'email'),)

# 创建表格
Base.metadata.create_all(bind=engine)

# 添加数据
user1 = User(username='alice', email='alice@example.com')
session.add(user1)

user2 = User(username='bob', email='alice@example.com')  # 插入重复数据
session.add(user2)

session.commit()  # 抛出 IntegrityError 异常

在上面的示例中，我们通过 UniqueConstraint 类在 username 和 email 列上定义了唯一约束。正如我们预期的那样，当插入重复的数据时，会抛出 IntegrityError 异常。