noita leaderboard + view + management commands

2026-05-10 02:11:30 +02:00
parent 52a6a4adb2
commit fa53d74295
6 changed files with 147 additions and 7 deletions
@@ -0,0 +1,62 @@
+from django.core.management.base import BaseCommand
+from noita.models import ObjectivPoint
+from noita.services.decode import POINTS
+
+
+class Command(BaseCommand):
+    help = "Load ObjectivPoints from the POINTS dictionary in services.decode"
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--clear",
+            action="store_true",
+            help="Clear all existing ObjectivPoints before loading",
+        )
+
+    def handle(self, *args, **options):
+        if options["clear"]:
+            ObjectivPoint.objects.all().delete()
+            self.stdout.write(self.style.SUCCESS("Cleared existing ObjectivPoints"))
+
+        created_count = 0
+        updated_count = 0
+
+        for objectiv_id, point_value in POINTS.items():
+            # Skip special entries
+            if objectiv_id in {"-", "DEBUG"}:
+                continue
+
+            # Get display string from objectiv_id (convert to title case)
+            display_string = objectiv_id.replace("_", " ").title()
+
+            # Create or update ObjectivPoint
+            obj, created = ObjectivPoint.objects.get_or_create(
+                objectiv_id=objectiv_id,
+                defaults={
+                    "display_string": display_string,
+                    "point": point_value,
+                    "max_count": 1,  # Default max_count is 1
+                },
+            )
+
+            if created:
+                created_count += 1
+                self.stdout.write(f"✓ Created: {objectiv_id} - {point_value} points")
+            else:
+                # Update if points changed
+                if obj.point != point_value or obj.display_string != display_string:
+                    obj.point = point_value
+                    obj.display_string = display_string
+                    obj.save()
+                    updated_count += 1
+                    self.stdout.write(
+                        self.style.WARNING(
+                            f"↻ Updated: {objectiv_id} - {point_value} points"
+                        )
+                    )
+
+        self.stdout.write(self.style.SUCCESS(f"\nCreated: {created_count}"))
+        self.stdout.write(self.style.SUCCESS(f"Updated: {updated_count}"))
+        self.stdout.write(
+            self.style.SUCCESS(f"Total ObjectivPoints: {ObjectivPoint.objects.count()}")
+        )