discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

rewrite replyto stage

Browse Source
This commit is contained in:
tonyrewin 2022-07-04 09:19:32 +03:00
parent b2552f43b0
commit bd4221e9af
1 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
migration/tables/comments.py
View File

@ -84,16 +84,12 @@ def migrate(entry, shouts_by_oid):
raise e
return comment_dict
def migrate_2stage(entry, id_map):
old_reply_to = entry.get('replyTo')
if not old_reply_to:
return
old_id = entry['_id']
if not old_id in id_map:
return
id = id_map[old_id]
def migrate_2stage(cmt, old_new_id):
reply_oid = cmt.get('replyTo')
if not reply_oid: return
new_id = old_new_id.get(cmt['_id'])
if not new_id: return
with local_session() as session:
comment = session.query(Comment).filter(Comment.id == id).first()
reply_to = id_map.get(old_reply_to)
comment.replyTo = reply_to
comment = session.query(Comment).filter(Comment.id == new_id).first()
comment.replyTo = old_new_id.get(reply_oid)
session.commit()