Add checkmutations command

profiles/management/commands/checkmutations.py

@@ -0,0 +1,84 @@
+from django.core.management.base import BaseCommand
+from django.db.models import F, FilteredRelation, Q, QuerySet
+from tqdm import tqdm
+
+from profiles.enums import ScoreMutation, ScoreResult
+from profiles.models import Score
+
+
+class Command(BaseCommand):
+    help = "Checks all scores have required mutations created."
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--fix",
+            action="store_true",
+            help="Whether create missing mutations after detecting them",
+        )
+
+    def handle(self, *args, **options):
+        fix = options["fix"]
+        self.check_nochoke_mutations(fix)
+
+    def check_nochoke_mutations(self, create_missing_mutations: bool):
+        choke_scores = Score.objects.filter_mutations().filter(
+            result=F("result").bitand(ScoreResult.CHOKE)
+        )
+
+        choke_scores_missing_nochoke_mutation = (
+            Score.objects.filter_mutations()
+            .filter(result=F("result").bitand(ScoreResult.CHOKE))
+            .annotate(
+                nochoke_mutation=FilteredRelation(
+                    "mutations",
+                    condition=Q(mutations__mutation=ScoreMutation.NO_CHOKE),
+                )
+            )
+            .filter(nochoke_mutation=None)
+        )
+
+        total_count = choke_scores.count()
+        missing_count = choke_scores_missing_nochoke_mutation.count()
+
+        if missing_count == 0:
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f"All {total_count} choke scores have nochoke mutations created"
+                )
+            )
+            return
+
+        self.stdout.write(
+            self.style.ERROR(
+                f"{missing_count} / {total_count} choke scores are missing nochoke mutations ({(total_count - missing_count) / total_count * 100:.2f}% complete)"
+            )
+        )
+        if create_missing_mutations:
+            self.create_missing_nochoke_mutations(choke_scores_missing_nochoke_mutation)
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f"Created {missing_count} missing nochoke mutations for choke scores"
+                )
+            )
+        else:
+            self.stdout.write(
+                self.style.WARNING(
+                    "Use --fix to create missing mutations for choke scores"
+                )
+            )
+
+    def create_missing_nochoke_mutations(self, choke_scores: QuerySet[Score]):
+        with tqdm(
+            desc="No-choke", total=choke_scores.count(), smoothing=0
+        ) as progress_bar:
+            while len(page := choke_scores.select_related("beatmap")[:2000]) > 0:
+                self.create_missing_nochoke_mutations_page(page, progress_bar)
+
+    def create_missing_nochoke_mutations_page(
+        self, choke_scores: QuerySet[Score], progress_bar: tqdm
+    ):
+        scores_to_create = []
+        for score in choke_scores:
+            scores_to_create.append(score.get_nochoke_mutation())
+            progress_bar.update(1)
+        Score.objects.bulk_create(scores_to_create)

profiles/management/commands/recalculate.py

@@ -2,7 +2,7 @@
 
 from django.core.management.base import BaseCommand
 from django.core.paginator import Paginator
-from django.db.models import Count, QuerySet
+from django.db.models import QuerySet
 from tqdm import tqdm
 
 from common.error_reporter import ErrorReporter