Как создать составные уникальные ограничения с помощью SQLAlchemy и Alembic? [дубликат] ⇐ Python

[Anonymous]

Я пытаюсь выяснить, как добавить составное уникальное значение в таблицу, где комбинация должна быть уникальной, но отдельные поля могут быть одинаковыми
Как выглядит моя миграция: (использует пакетные операции, поскольку база данных разработки является sqlite)

Код: Выделить всё

def upgrade() -> None:
"""Upgrade schema."""
with op.batch_alter_table("authors") as batch_op:
batch_op.create_unique_constraint(
"uq_author_fullname", ["firstname", "lastname"]
)

pass

def downgrade() -> None:
"""Downgrade schema."""
with op.batch_alter_table("authors") as batch_op:
batch_op.drop_constraint("uq_author_fullname", type_="unique")
pass

Что создает Alembic (полагаю, мне нужно избавиться от UNIQUE (фамилия) и UNIQUE (имя)

Код: Выделить всё

CREATE TABLE _alembic_tmp_authors (
id INTEGER NOT NULL,
firstname VARCHAR(100),
lastname VARCHAR(100),
PRIMARY KEY (id),
CONSTRAINT uq_author_fullname UNIQUE (firstname, lastname),
UNIQUE (lastname),
UNIQUE (firstname)
)

Модель автора:

Код: Выделить всё

class Author(Base):
__tablename__ = "authors"
id = Column(Integer, primary_key=True)
firstname = Column(String(100))
lastname = Column(String(100))

books = relationship("Book", secondary=authorBook, back_populates="authors")

UniqueConstraint("firstname", "lastname", name="fullname_unique")

Как видите, я пытаюсь использовать UniqueConstraint()
Теперь предположим, что два объекта: Питер Паркер и Питер Гриффин
И следующий код вставки:

Код: Выделить всё

# Imports ommited
def add(db: Session, first_name: str, last_name: str):
author = Author(firstname=first_name, lastname=last_name)
db.add(author)
try:
db.commit()
db.refresh(author)
except IntegrityError:
return None

return author

Если Питер Паркер находится в базе данных, добавить Питера Гриффина не удастся, IntegrityError нарушает ограничение имени. Но это нежелательное поведение, я хочу, чтобы оно не удавалось только в том случае, если комбинация имени и фамилии совпадает, поэтому я пытаюсь добавить Питера Паркера еще раз.