添加和删除具有一对多关系的行时出现问题

Question

这两个表都包含唯一的约束。我已经为此苦苦挣扎了一段时间了。添加和删除的命令应该是什么

做加法

movie = Movie(title="transformers", director="mb")
movie2 = Movie(title="transformers2", director="mb")
genre = Genre(category="action")
db.session.add(movie)
db.session.add(movie2)
movie.genres.append(genre)
movie2.genres.append(genre)
db.session.commit()

给予

(sqlite3.IntegrityError) UNIQUE constraint failed: genres.category
[SQL: INSERT INTO genres (category) VALUES (?)]
[parameters: ('action',)]
(Background on this error at: http://sqlalche.me/e/14/gkpj)

通过执行删除

db.session.delete(Movie.query.filter_by(title="t1").first())
db.session.commit()

不删除

期望是什么：

• 如果电影表中不存在标题，则添加新行；如果流派表中不存在流派，则添加新行；并在关联表中创建链接

• 从电影表中删除标题并从关联表中删除链接。不要为体裁做任何事

我需要改变我的型号吗

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

class Config:
    SQLALCHEMY_DATABASE_URI = 'sqlite:///:memory:?charset=utf8'
    SQLALCHEMY_DATABASE_ECHO = False
    SQLALCHEMY_TRACK_MODIFICATIONS = 'False'
    FLASK_ENV = 'test'
    DEBUG = True


db = SQLAlchemy()
conf = Config()


def create_app():
    app = Flask(__name__)
    app.config.from_object(conf)
    db.init_app(app)

    return app


class Rating(db.Model):
    __tablename__ = 'ratings'
    id = db.Column(db.Integer, db.ForeignKey("movies.id"), primary_key=True,)
    rating = db.Column(db.String(4), index=True, nullable=False,)
    movie = db.relationship("Movie", back_populates="rating",)

    def __repr__(self):
        return '{}'.format(self.rating)


Association = db.Table('association',
                       db.Column('movies_id', db.Integer,
                                 db.ForeignKey('movies.id'), index=True,),
                       db.Column(
                           'genres_id',
                           db.Integer, db.ForeignKey('genres.id'), index=True,),
                       )


class Movie(db.Model):
    __tablename__ = 'movies'
    id = db.Column(db.Integer, primary_key=True, index=True,)
    title = db.Column(db.String(80), index=True, unique=True, nullable=False,)
    director = db.Column(db.String(30), primary_key=False,
                         unique=False, nullable=False)
    rating = db.relationship(
        "Rating", uselist=False, back_populates="movie"
    )
    genres = db.relationship(
        "Genre", secondary='association', backref=db.backref('movies'),
    )

    def __repr__(self):
        return '{}'.format(self.title)


class Genre(db.Model):
    __tablename__ = 'genres'
    id = db.Column(
        db.Integer,
        primary_key=True,
        index=True,
    )
    category = db.Column(
        db.String(80),
        index=True,
        unique=True,
        nullable=False,
    )

    def __repr__(self):
        return '{}'.format(self.category)


app = create_app()
context = app.app_context()
context.push()
db.create_all()

movie = Movie(title="transformers", director="mb")
rating = Rating(rating="5", movie=movie)
genre = Genre(category="action")
db.session.add_all([movie, rating])
movie.genres.append(genre)

try:
    db.session.commit()
    print("commit successfull")
except Exception as e:
    print(f"{e}")
    db.session.rollback()

movie2 = Movie(title="transformers2", director="mb")
rating = Rating(rating="5", movie=movie2)
genre2 = Genre.query.filter_by(category="action").first()
db.session.add_all([movie2, rating])
movie2.genres.append(genre2)

try:
    db.session.commit()
    print("commit successfull")
except Exception as e:
    print(f"{e}")
    db.session.rollback()

to_delete = Movie.query.filter_by(title="transformers").first()

try:
    db.session.delete(to_delete)
    db.session.commit()
    print("deletion successfull")
except Exception as e:
    print(f"{e}")
    db.session.rollback()

context.pop()

这导致了

commit successfull
commit successfull
Dependency rule tried to blank-out primary key column 'ratings.id' on instance '<Rating at 0x2022c81bdc8>'

Answer 1

如果您的模型确实像您所写的那样简单，那么我将使用Association Proxy来简化这种多对多关系

您的代码将按以下方式更改：

• 添加一个_genre_find_or_create函数：

def _genre_find_or_create(category):
    obj = Genre.query.filter_by(category=category).first()
    return obj or Genre(category=category)

• 将Movie定义更改为genres

class Movie(db.Model):
    # ...

    _genres = db.relationship(
        "Genre",
        secondary="association",
        backref=db.backref("movies"),
    )
    genres = association_proxy(
        "_genres",
        "category",
        creator=_genre_find_or_create,
    )

然后您可以按如下方式使用它：

# pre-adding data to ensure the category already exists
genre = Genre(category="action")
db.session.add(genre)
db.session.commit()

# adding new data
movie = Movie(title="transformers")  # , director="mb")
movie2 = Movie(title="transformers2")  # , director="mb")
db.session.add(movie)
db.session.add(movie2)
# OLD way:
# genre = Genre(category="action")
# movie.genres.append(genre)
# movie2.genres.append(genre)
# NEW way: use just 'category'
movie.genres.append("action")
movie2.genres.append("action")

db.session.commit()

---

另请参见使用此解决方案的另一个question

---

我仍然不明白你为什么对DELETE有问题。您的示例代码运行正常，运行时生成以下SQL语句：

DELETE FROM association WHERE association.movies_id = ? AND association.genres_id = ?
DELETE FROM movies WHERE movies.id = ?

---

更新：删除 给定更新的模型，很明显，删除与Movie和Rating之间的一对一关系有关

简单的解决方案是将cascade="delete"添加到Movie.rating关系中，这将在删除Movie的情况下删除与Rating相关的实例：

rating = db.relationship(
    "Rating",
    uselist=False,
    back_populates="movie",
    cascade="delete",  # <- **NEW**
)

阅读更多关于Cascades

Answer 2

发生IntegrityError是因为类型表中已经有动作类型。在这种情况下，必须从数据库中获取现有类型

movie = Movie(title="transformers", director="mb")
movie2 = Movie(title="transformers2", director="mb")
genre = Genre.query.filter_by(category="action").first()
db.session.add(movie)
db.session.add(movie2)
movie.genres.append(genre)
movie2.genres.append(genre)
db.session.commit()

我无法在问题的原始版本中重现删除问题。然而错误

Dependency rule tried to blank-out primary key column 'ratings.id' on instance '<Rating at 0x2022c81bdc8>'

可以通过在电影模型中的分级关系上配置deletecascade来删除

    rating = db.relationship(
        "Rating", uselist=False, back_populates="movie",
        cascade='save-update, merge, delete',
    )

这样可以确保删除电影时，也会删除关联的分级。如果没有它，正如您所发现的，SQLAlchemy不会尝试删除相关的评级，从而导致错误