api

opus_submitter/submissions/admin.py

@@ -1,6 +1,10 @@
 from django.contrib import admin
 from django.utils.html import format_html
-from .models import SteamAPIKey, SteamCollection, SteamCollectionItem
+from django.utils import timezone
+from .models import (
+    SteamAPIKey, SteamCollection, SteamCollectionItem,
+    Submission, PuzzleResponse, SubmissionFile
+)
 
 
 @admin.register(SteamAPIKey)
@@ -141,3 +145,167 @@ class SteamCollectionItemAdmin(admin.ModelAdmin):
         ("Metadata", {"fields": ("tags",)}),
         ("Timestamps", {"fields": ("created_at", "updated_at")}),
     )
+
+
+class SubmissionFileInline(admin.TabularInline):
+    model = SubmissionFile
+    extra = 0
+    readonly_fields = ["file_size", "content_type", "ocr_processed", "created_at"]
+    fields = ["file", "original_filename", "file_size", "content_type", "ocr_processed", "ocr_error"]
+
+
+class PuzzleResponseInline(admin.TabularInline):
+    model = PuzzleResponse
+    extra = 0
+    readonly_fields = ["created_at", "updated_at"]
+    fields = [
+        "puzzle", "puzzle_name", "cost", "cycles", "area",
+        "needs_manual_validation", "ocr_confidence_score"
+    ]
+
+
+@admin.register(Submission)
+class SubmissionAdmin(admin.ModelAdmin):
+    list_display = [
+        "id", "user", "total_responses", "needs_validation", 
+        "is_validated", "created_at"
+    ]
+    list_filter = [
+        "is_validated", "created_at", "updated_at"
+    ]
+    search_fields = ["id", "user__username", "notes"]
+    readonly_fields = ["id", "created_at", "updated_at", "total_responses", "needs_validation"]
+    inlines = [PuzzleResponseInline]
+    
+    fieldsets = (
+        ("Basic Information", {
+            "fields": ("id", "user", "notes")
+        }),
+        ("Validation", {
+            "fields": ("is_validated", "validated_by", "validated_at")
+        }),
+        ("Statistics", {
+            "fields": ("total_responses", "needs_validation"),
+            "classes": ("collapse",)
+        }),
+        ("Timestamps", {
+            "fields": ("created_at", "updated_at"),
+            "classes": ("collapse",)
+        }),
+    )
+    
+    actions = ["mark_as_validated"]
+    
+    def mark_as_validated(self, request, queryset):
+        """Mark selected submissions as validated"""
+        updated = 0
+        for submission in queryset:
+            if not submission.is_validated:
+                submission.is_validated = True
+                submission.validated_by = request.user
+                submission.validated_at = timezone.now()
+                submission.save()
+                # Also mark all responses as not needing validation
+                submission.responses.update(needs_manual_validation=False)
+                updated += 1
+        
+        self.message_user(request, f"{updated} submissions marked as validated.")
+    
+    mark_as_validated.short_description = "Mark selected submissions as validated"
+
+
+@admin.register(PuzzleResponse)
+class PuzzleResponseAdmin(admin.ModelAdmin):
+    list_display = [
+        "puzzle_name", "submission", "puzzle", "cost", "cycles", "area",
+        "needs_manual_validation", "created_at"
+    ]
+    list_filter = [
+        "needs_manual_validation", "puzzle__collection", "created_at"
+    ]
+    search_fields = [
+        "puzzle_name", "submission__id", "puzzle__title", 
+        "cost", "cycles", "area"
+    ]
+    readonly_fields = ["created_at", "updated_at"]
+    inlines = [SubmissionFileInline]
+    
+    fieldsets = (
+        ("Basic Information", {
+            "fields": ("submission", "puzzle", "puzzle_name")
+        }),
+        ("OCR Data", {
+            "fields": ("cost", "cycles", "area", "ocr_confidence_score")
+        }),
+        ("Validation", {
+            "fields": (
+                "needs_manual_validation", 
+                "validated_cost", "validated_cycles", "validated_area"
+            )
+        }),
+        ("Timestamps", {
+            "fields": ("created_at", "updated_at"),
+            "classes": ("collapse",)
+        }),
+    )
+    
+    actions = ["mark_for_validation", "clear_validation_flag"]
+    
+    def mark_for_validation(self, request, queryset):
+        """Mark selected responses as needing validation"""
+        updated = queryset.update(needs_manual_validation=True)
+        self.message_user(request, f"{updated} responses marked for validation.")
+    
+    def clear_validation_flag(self, request, queryset):
+        """Clear validation flag for selected responses"""
+        updated = queryset.update(needs_manual_validation=False)
+        self.message_user(request, f"{updated} responses cleared from validation queue.")
+    
+    mark_for_validation.short_description = "Mark as needing validation"
+    clear_validation_flag.short_description = "Clear validation flag"
+
+
+@admin.register(SubmissionFile)
+class SubmissionFileAdmin(admin.ModelAdmin):
+    list_display = [
+        "original_filename", "response", "file_size_display", 
+        "content_type", "ocr_processed", "created_at"
+    ]
+    list_filter = [
+        "content_type", "ocr_processed", "created_at"
+    ]
+    search_fields = [
+        "original_filename", "response__puzzle_name", 
+        "response__submission__id"
+    ]
+    readonly_fields = [
+        "file_size", "content_type", "ocr_processed", 
+        "created_at", "updated_at", "file_url"
+    ]
+    
+    fieldsets = (
+        ("File Information", {
+            "fields": ("file", "original_filename", "file_size", "content_type", "file_url")
+        }),
+        ("OCR Processing", {
+            "fields": ("ocr_processed", "ocr_raw_data", "ocr_error")
+        }),
+        ("Relationships", {
+            "fields": ("response",)
+        }),
+        ("Timestamps", {
+            "fields": ("created_at", "updated_at"),
+            "classes": ("collapse",)
+        }),
+    )
+    
+    def file_size_display(self, obj):
+        """Display file size in human readable format"""
+        if obj.file_size < 1024:
+            return f"{obj.file_size} B"
+        elif obj.file_size < 1024 * 1024:
+            return f"{obj.file_size / 1024:.1f} KB"
+        else:
+            return f"{obj.file_size / (1024 * 1024):.1f} MB"
+    
+    file_size_display.short_description = "File Size"