在动态创建条目时,Sqlalchemy重复条目

时间:2017-08-25 20:22:21

标签: flask sqlalchemy

在我的应用程序中,我使用Flask的SQLAlchemy(通过MySQL),在我目前的情况下,我需要动态创建条目。例如,在从表A创建实例时,我的脚本将看到我们需要创建一个相关的条目B,因此在创建B之后,它将尝试将B分配给A的实例的一个关系属性。

以下是一些示例模型:

a_b = db.Table('a_b',
               db.Column('a_id', db.Integer,
                         db.ForeignKey('a.id',
                                       onupdate="CASCADE",
                                       ondelete="CASCADE"),
                         primary_key=True),
               db.Column('b_id', db.Integer,
                         db.ForeignKey('b.id',
                                       onupdate="CASCADE",
                                       ondelete="CASCADE"),
                         primary_key=True))

class ModelA(db.Model):
    __tablename__ = 'a'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), nullable=False)

    model_bs = db.relationship('ModelB', secondary=a_b,
                               back_populates='model_as')

class ModelB(db.Model):
    __tablename__ = 'b'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), nullable=False)

    model_as = db.relationship('ModelA', secondary=a_b,
                               back_populates='model_bs')

我遇到的问题是SQL重复条目异常,在运行如下内容之后:

from base import db
from base.database.models.models import ModelA, ModelB


def creation_test():
    with db.session.no_autoflush:
        # Start creating A
        a = ModelA()
        a.name = 'A Example'

        # See that we need B, so create B
        b = ModelB()
        b.name = 'B Example'
        b.model_as = []
        db.session.add(b)
        db.session.flush((b,))

        # Use B in the relationship from A
        a.model_bs = [b]
        db.session.add(a)
        db.session.flush((a,))

    db.session.commit()

creation_test()

它会吐出一个例外:

sqlalchemy.exc.IntegrityError: (_mysql_exceptions.IntegrityError) (1062, "Duplicate entry '1-1' for key 'PRIMARY'") [SQL: u'INSERT INTO a_b (a_id, b_id) VALUES (%s, %s)'] [parameters: (1L, 1L)]

如何将条目添加到会话中和/或如何刷新它们是否有问题?甚至可以实例化这样的条目吗?

我感谢任何反馈。

1 个答案:

答案 0 :(得分:0)

在更新了我的SQLAlchemy版本并测试了一些东西后,似乎我能够通过删除创建每个实例之间的刷新来解决这个问题,而只是在提交会话之前执行单个刷新:

from base import db
from base.database.models.models import ModelA, ModelB


def creation_test():
    with db.session.no_autoflush:
        # Start creating A
        a = ModelA()
        a.name = 'A Example'

        # See that we need B, so create B
        b = ModelB()
        b.name = 'B Example'
        b.model_as = []
        db.session.add(b)

        # Use B in the relationship from A
        a.model_bs = [b]
        db.session.add(a)

    db.session.flush()
    db.session.commit()

creation_test()