Prod Server

api/management/commands/generate_acheivements.py

@@ -1,6 +1,7 @@
 from django.core.management.base import BaseCommand
 from typing import Any, Optional
-from db.achievement import Achievement, UserAchievements
+from db.achievement import Achievement, UserAchievementsLog
+from django.conf import settings
 
 from db.user import User
 
@@ -17,9 +18,14 @@ def handle(self, *args: Any, **options: Any) -> Optional[str]:
         levels = {}
 
         for i in range(1, 7):
-            levels[i] = Achievement.objects.get(level__level_order=i)
+            levels[i] = Achievement.objects.filter(level_id__level_order=i).first()
+        i = 0
+        print("started generating achievements")
 
         for user in users:
+            i += 1
+            if i % 100 == 0:
+                print("Processed users: ", i)
             try:
                 level_link = user.user_lvl_link_user
             except:
@@ -28,11 +34,24 @@ def handle(self, *args: Any, **options: Any) -> Optional[str]:
                 continue
             level = level_link.level
             level_order = level.level_order
+
+            batch = []
+
             for i in range(1, level_order + 1):
-                achievement = levels[i]
-                UserAchievements.objects.create(
+                achievement = levels.get(i)
+                if not achievement:
+                    continue
+                if UserAchievementsLog.objects.filter(
                     user_id=user, achievement_id=achievement
+                ).exists():
+                    continue
+                batch.append(
+                    UserAchievementsLog(
+                        user_id=user,
+                        achievement_id=achievement,
+                        created_by=settings.SYSTEM_ADMIN_ID,
+                        updated_by=settings.SYSTEM_ADMIN_ID,
+                    )
                 )
-            print(
-                "Achievements generated for user: ", user.muid, "Level: ", level_order
-            )
+            UserAchievementsLog.objects.bulk_create(batch)
+            print("added user acheivements for users")