对 SQLAlchemy 中的 UniqueKeyViolation 做出反应

Question

我在 SQLAlchemy 中有 2 个表（使用 declerative_base()），它们通过单独的表关系连接起来

# no comment on the wired naming....
site_word = Table('word_site', Base.metadata,
    Column('site_id', Integer, ForeignKey('sites.site_id'), nullable = False, primary_key = True),
    Column('word_id', Integer, ForeignKey('site_words.word_id'), nullable = False, primary_key = True))

单词映射

class Word(Base):
   # snip
   word =  Column('word', Text, nullable = False, unique = True)
   sites = relationship('Site', secondary = site_word, back_populates = 'words')

站点映射

class Site(Base):
   # snip
   words = relationship('Word', secondary = site_word, back_populates = 'sites')

插入工作如预期，插入已经存在的单词当然会失败，因为 唯一 在 Word.word 中。

我尝试使用 before_insert 事件来检查该项目是否已存在。

event.listen(Word, 'before_insert', some_event)

我可以获得信息来按预期唯一识别单词，但我不知道如何向连接表 (site_word) 添加新值。

我可以为数据库编写触发器或过程，但不想将太多逻辑移入数据库（不知道这是否可行，因为此时不知道 Site时间）。我可以删除 word 列的约束，但我仍然无法弄清楚如何访问信息（连接的另一端）以在连接表中创建条目，但不能在Word 表。

我正在寻找一种方法仅在 site_word 和 Site 表中创建条目。

更新 1：
我可以将事件附加到 Site，但我看不到获取要插入的站点信息的可能性。是否可以保存站点并随后创建连接关系？

Answer 1

如果没有如何使用对象的示例，就很难看出问题。我认为您正在尝试执行以下操作：

site = Site()
site.words.append(Word('Someword'))
session.add(site)
session.commit()

如果“someword”已经存在，这当然会引发 IntegrityError 。为了解决这个问题，您需要先在数据库中查询该单词，然后再将其添加到网站：

def get_unique_word(session, word):
    try:
        return session.query(Word).filter(Word.word==word).one()
    except sqlalchemy.orm.exc.NoResultFound:
        return Word(word)

site = Site()
site.words.append(get_unique_word(session, 'someword'))
session.add(site)
session.commit()

如果这完全不合理，那么您必须更详细地描述您的问题。

Answer 2

我设法做到了。 Mark Gemmill 的答案让我朝着正确的方向思考。

我需要听 AttributeEvent 而不是 MapperEvent。

event.listen(Site.words, 'append', test_event, retval = True)

然后我可以做类似的事情：

def test_event(target, value, initiator):
    try:
        return session.query(Word).filter(Word.word == value.word.lower()).one()
    except sqlalchemy.orm.exc.NoResultFound:
        return value 

这要么从表中返回现有的单词，要么只是返回需要持久化的映射对象。