Skip to content

Commit

Permalink
Remove draft revisions when review is published
Browse files Browse the repository at this point in the history
  • Loading branch information
anshg1214 committed Sep 26, 2022
1 parent 9620264 commit e1b59c4
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 0 deletions.
19 changes: 19 additions & 0 deletions critiquebrainz/db/revision.py
Original file line number Diff line number Diff line change
Expand Up @@ -230,3 +230,22 @@ def votes(revision_id):
else: # False = negative
revision_votes["negative"] += 1
return revision_votes


def delete_draft_revisons(review_id):
"""Delete all draft revisions of a review after the review has been published.
Args:
review_id: ID of the review.
"""
with db.engine.connect() as connection:
connection.execute(sqlalchemy.text("""
DELETE
FROM revision
USING review
WHERE review.id = revision.review_id
AND revision.review_id = :review_id
AND revision.timestamp < review.published_on
"""), {
"review_id": review_id,
})
3 changes: 3 additions & 0 deletions critiquebrainz/frontend/views/review.py
Original file line number Diff line number Diff line change
Expand Up @@ -417,6 +417,9 @@ def edit(id):
raise BadRequest(lazy_gettext("Changing license of a published review\
or converting a published review back to drafts is not allowed."))

if form.state.data == "publish" and review["is_draft"] == True:
db_revision.delete_draft_revisons(review["id"])

flash.success(gettext("Review has been updated."))
return redirect(url_for('.entity', id=review["id"]))
else:
Expand Down

0 comments on commit e1b59c4

Please sign in to comment.