migration-email-confirm-fix

migration/__init__.py

@@ -7,7 +7,7 @@ import sys
 from datetime import datetime
 
 import bs4
-
+from base.redis import redis
 from migration.tables.comments import migrate as migrateComment
 from migration.tables.comments import migrate_2stage as migrateComment_2stage
 from migration.tables.content_items import get_shout_slug
@@ -181,7 +181,11 @@ async def all_handle(storage, args):
     print("[migration] handle everything")
     await users_handle(storage)
     await topics_handle(storage)
+    print("[migration] users and topics are migrated")
+    await redis.connect()
+    print("[migration] redis connected")
     await shouts_handle(storage, args)
+    print("[migration] migrating comments")
     await comments_handle(storage)
     # export_email_subscriptions()
     print("[migration] done!")
@@ -295,9 +299,9 @@ def create_pgdump():
 
 
 async def handle_auto():
-    print("[migration] no option given, auto mode")
     url = os.getenv("MONGODB_URL")
     if url:
+        print("[migration] connecting mongo")
         mongo_download(url)
     bson_handle()
     await all_handle(data_load(), sys.argv)

migration/tables/content_items.py

@@ -149,6 +149,12 @@ async def migrate(entry, storage):
     if entry.get("published"):
         r["publishedAt"] = date_parse(entry.get("publishedAt", OLD_DATE))
         r["visibility"] = "public"
+        with local_session() as session:
+            # update user.emailConfirmed if published
+            author = session.query(User).where(User.slug == userslug).first()
+            author.emailConfirmed = True
+            session.add(author)
+            session.commit()
     else:
         r["visibility"] = "authors"
     if "deletedAt" in entry: