diff options
| -rw-r--r-- | mediagoblin/db/migrations/versions/4066b9f8b84a_use_comment_link_ids_notifications.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/mediagoblin/db/migrations/versions/4066b9f8b84a_use_comment_link_ids_notifications.py b/mediagoblin/db/migrations/versions/4066b9f8b84a_use_comment_link_ids_notifications.py index dd0fb883..2e9c7513 100644 --- a/mediagoblin/db/migrations/versions/4066b9f8b84a_use_comment_link_ids_notifications.py +++ b/mediagoblin/db/migrations/versions/4066b9f8b84a_use_comment_link_ids_notifications.py @@ -40,7 +40,8 @@ def upgrade(): # rather than the ID of TextComment object. db.execute(notification_table.update().values( object_id=comment_link.id - + )) + # Find the GMR for this comment or make one if one doesn't exist. gmr = db.execute(gmr_table.select().where(and_( gmr_table.c.obj_pk == comment_link.id, @@ -88,6 +89,7 @@ def downgrade(): # Update the notification with the TextComment (i.e. the comment object) db.execute(notification_table.update().values( object_id=comment_link.comment_id + )) # Find the GMR for the TextComment gmr = db.execute(gmr_table.select().where(and_( gmr_table.c.obj_pk == comment_link.id, |