core/migration/tables/remarks.py

from base.orm import local_session
from migration.extract import extract_md
from migration.html2text import html2text
from orm.reaction import Reaction, ReactionKind


def migrate(entry, storage):
    post_oid = entry['contentItem']
    print(post_oid)
    shout_dict = storage['shouts']['by_oid'].get(post_oid)
    if shout_dict:
        print(shout_dict['body'])
        remark = {
            "shout": shout_dict['id'],
            "body": extract_md(
                html2text(entry['body']),
                shout_dict
            ),
            "kind": ReactionKind.REMARK
        }

        if entry.get('textBefore'):
            remark['range'] = str(
                    shout_dict['body']
                        .index(
                            entry['textBefore'] or ''
                        )
                ) + ':' + str(
                    shout_dict['body']
                        .index(
                            entry['textAfter'] or ''
                        ) + len(
                            entry['textAfter'] or ''
                        )
                )

        with local_session() as session:
            rmrk = Reaction.create(**remark)
            session.commit()
            del rmrk["_sa_instance_state"]
            return rmrk
    return