SQLAlchemy 是否可以添加新记录并自动处理表之间的关系，而无需我手动检查 PK 唯一性？

Question

我是 SQLAlchemy 的新手，并且已经阅读了基本文档。我目前正在关注 Mike Driscoll 的 MediaLocker教程并为了我自己的目的修改/扩展它。

我有三张表（loans、people、card）。卡到贷款和人到贷款都是一对多关系，并且建模如下：

from sqlalchemy import Table, Column, DateTime, Integer, ForeignKey, Unicode
from sqlalchemy.orm import backref, relation
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base

engine = create_engine("sqlite:///cardsys.db", echo=True)
DeclarativeBase = declarative_base(engine)
metadata = DeclarativeBase.metadata

class Loan(DeclarativeBase):
    """
    Loan model
    """

    __tablename__ = "loans"

    id = Column(Integer, primary_key=True)
    card_id = Column(Unicode, ForeignKey("cards.id"))
    person_id = Column(Unicode, ForeignKey("people.id"))
    date_issued = Column(DateTime)
    date_due = Column(DateTime)
    date_returned = Column(DateTime)
    issue_reason = Column(Unicode(50))
    person = relation("Person", backref="loans", cascade_backrefs=False)
    card = relation("Card", backref="loans", cascade_backrefs=False)

class Card(DeclarativeBase):
    """
    Card model
    """

    __tablename__ = "cards"

    id = Column(Unicode(50), primary_key=True)
    active = Column(Boolean)

class Person(DeclarativeBase):
    """
    Person model
    """

    __tablename__ = "people"

    id = Column(Unicode(50), primary_key=True)
    fname = Column(Unicode(50))
    sname = Column(Unicode(50))

当我尝试创建新贷款（在我的控制器中使用以下方法）时，它对于独特的卡和人来说效果很好，但是一旦我尝试要为特定的人或卡添加第二笔贷款，它会给我一个“非唯一”错误。显然它不是唯一的，这就是重点，但我认为 SQLAlchemy 会为我处理幕后的事情，并添加正确的现有人员或卡 ID 作为新贷款中的 FK，而不是尝试创建新的人员和卡片记录。我是否需要查询数据库来检查 PK 唯一性并手动处理这个问题？我的印象是这应该是 SQLAlchemy 能够自动处理的事情？

def addLoan(session, data):

    loan = Loan()
    loan.date_due = data["loan"]["date_due"]
    loan.date_issued = data["loan"]["date_issued"]
    loan.issue_reason = data["loan"]["issue_reason"]

    person = Person()
    person.id = data["person"]["id"]
    person.fname = data["person"]["fname"]
    person.sname = data["person"]["sname"]
    loan.person = person

    card = Card()
    card.id = data["card"]["id"]
    loan.card = card

    session.add(loan)
    session.commit()

在 MediaLocker 示例中，新行是使用自动增量 PK 创建的（即使对于重复项，也不符合规范化规则）。我想要一个规范化的数据库（即使在一个小项目中，只是为了学习的最佳实践），但在网上找不到任何示例来学习。

如何实现上述目标？

Answer 1

在尝试添加具有重复主键的新对象之前，您需要检索现有的 Person 或 Card 对象并将其分配给关系。您可以通过对代码进行一些小的更改来实现此目的。

def addLoan(session, data):

    loan = Loan()
    loan.date_due = data["loan"]["date_due"]
    loan.date_issued = data["loan"]["date_issued"]
    loan.issue_reason = data["loan"]["issue_reason"]

    person = session.query(Person).get(data["person"]["id"])
    if not person:
        person = Person()
        person.id = data["person"]["id"]
        person.fname = data["person"]["fname"]
        person.sname = data["person"]["sname"]
    loan.person = person

    card = session(Card).query.get(data["card"]["id"])
    if not card:
        card = Card()
        card.id = data["card"]["id"]
    loan.card = card

    session.add(loan)
    session.commit()

还有一些针对 get_or_create 函数的解决方案，如果您想将其合并为一个步骤。

如果您从头开始将大量记录加载到新数据库中，并且您的查询比 get 更复杂（会话对象应该缓存 get 查询） ），您可以通过 ID 将每个新的 Person 和 Card 对象添加到临时字典中，并检索那里的现有对象而不是访问数据库，从而完全避免以内存为代价的查询。