Merge pull request #476 from PROCOLLAB-github/feature/user_field_migrations

PRO-484: User old field migration

Author: sh1nkey

core/management/commands/migrate_education.py

@@ -1,4 +1,7 @@
+import json
+
 from django.db import transaction
+from django.conf import settings
 from django.core.management.base import BaseCommand
 from django.contrib.auth import get_user_model
 
@@ -32,13 +35,30 @@ def migrate_organization_to_education() -> int:
     """
     Migrate old field `organization` to new model `Education`.
     Returns count migrated users.
+    Stored migrated info into `BASE_DIR / "log" / "migrated_users.json"`
     """
-    users_with_irganization = CustomUser.objects.exclude(organization=None).exclude(organization="")
+    user_with_education_ids: list[int] = UserEducation.objects.values_list("user__id", flat=True)
+    users_with_organization_without_education = (
+        CustomUser.objects
+        .exclude(organization=None)
+        .exclude(organization="")
+        .exclude(id__in=user_with_education_ids)
+    )
     UserEducation.objects.bulk_create([
         UserEducation(
             user=user,
             organization_name=user.organization,
         )
-        for user in users_with_irganization
+        for user in users_with_organization_without_education
     ])
-    return users_with_irganization.count()
+
+    data = [
+        {"user_id": user.id, "user_organization_field": user.organization}
+        for user in users_with_organization_without_education
+    ]
+
+    file_dump = settings.BASE_DIR / "log" / "migrated_users.json"
+    with open(file_dump, "w", encoding="utf-8") as file:
+        json.dump(data, file, indent=4, ensure_ascii=False)
+
+    return users_with_organization_without_education.count()

core/management/commands/migrate_old_to_new_fields_trigram.py

@@ -0,0 +1,14 @@
+[
+    {
+        "user_id": 3,
+        "user_organization_field": "Учеба 1"
+    },
+    {
+        "user_id": 4,
+        "user_organization_field": "Учеба 2"
+    },
+    {
+        "user_id": 5,
+        "user_organization_field": "Учеба 3"
+    }
+]