migrating comments and adding authors

migration/tables/comments.py

@@ -1,10 +1,33 @@
-from html2text import html2text
 import datetime
+import json
+from os.path import abspath
+from orm import Shout
+from orm.base import local_session
+from migration.html2text import html2text
 
-# markdown = Converter()
+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!
@@ -21,10 +44,16 @@ def migrate(entry):
         old_id: String
     }
     '''
-    # TODO: implement comments migration
-    return {
-        'slug': entry['slug'],
-        'createdAt': entry['createdAt'],
-        'body': html2text(entry['body']),
-        'replyTo': entry['']
-    }
+    with local_session() as session:
+        shout_id = session.query(Shout).filter(Shout.old_id == entry['_id']).first()
+        return {
+            'old_id': entry['_id'],
+            'old_thread': entry['thread'],
+            'createdBy': users_dict[entry['createdBy']],
+            'createdAt': entry['createdAt'],
+            'body': html2text(entry['body']),
+            'shout': shout_id,
+            'rating': entry['rating'],
+            'ratings': [] # TODO: ratings in comments
+        }
+    return None