core/migration/tables/comments.py

from dateutil.parser import parse as date_parse
import json
import datetime
from os.path import abspath
from orm import Shout, Comment, CommentRating, User
from orm.base import local_session
from migration.html2text import html2text

# users_dict = json.loads(open(abspath('migration/data/users.dict.json')).read())
# topics_dict = json.loads(open(abspath('migration/data/topics.dict.json')).read()) # old_id keyed

def migrate(entry):
    '''
    {
      "_id": "hdtwS8fSyFLxXCgSC",
      "body": "<p>",
      "contentItem": "mnK8KsJHPRi8DrybQ",
      "createdBy": "bMFPuyNg6qAD2mhXe",
      "thread": "01/",
      "createdAt": "2016-04-19 04:33:53+00:00",
      "ratings": [
        { "createdBy": "AqmRukvRiExNpAe8C", "value": 1 },
        { "createdBy": "YdE76Wth3yqymKEu5", "value": 1 }
      ],
      "rating": 2,
      "updatedAt": "2020-05-27 19:22:57.091000+00:00",
      "updatedBy": "0"
    }

    ->

    type Comment {
        id: Int!
        author: Int!
        body: String!
        replyTo: Int!
        createdAt: DateTime!
        updatedAt: DateTime
        shout: Int!
        deletedAt: DateTime
        deletedBy: Int
        rating: Int
        ratigns: [CommentRating]
        views: Int
        old_id: String
        old_thread: String
    }
    '''
    with local_session() as session:
        shout = session.query(Shout).filter(Shout.old_id == entry['_id']).first()
        if not shout: print(entry)
        assert shout, '=== NO SHOUT IN COMMENT ERROR ==='
        author = session.query(User).filter(User.old_id == entry['_id']).first()
        comment_dict = {
            'old_id': entry['_id'],
            'author': author.id if author else 0,
            'createdAt': date_parse(entry['createdAt']),
            'body': html2text(entry['body']),
            'shout': shout
        }
        if 'rating' in entry:
          comment_dict['rating'] = entry['rating']
        if entry.get('deleted'):
          comment_dict['deletedAt'] = entry['updatedAt']
          comment_dict['deletedBy'] = entry['updatedBy']
        if 'thread' in entry:
          comment_dict['old_thread'] = entry['thread']
        # print(entry.keys())
        comment = Comment.create(**comment_dict)
        for comment_rating_old in entry.get('ratings',[]):
            rater_id = session.query(User).filter(User.old_id == comment_rating_old['createdBy']).first()
            comment_rating_dict = {
                'value': comment_rating_old['value'],
                'createdBy': rater_id or 0,
                'createdAt': comment_rating_old.get('createdAt', datetime.datetime.now()),
                'comment_id': comment.id
            }
            try:
              comment_rating = CommentRating.create(**comment_rating_dict)
              # TODO: comment rating append resolver
              # comment['ratings'].append(comment_rating)
            except Exception as e:
              print(comment_rating)
              pass # raise e
        return comment