init commit

workout/admin.py

@@ -0,0 +1,45 @@
+from django.contrib import admin
+from .models import *
+from exercise.models import Exercise
+from import_export.admin import ImportExportModelAdmin
+from superset.models import Superset
+
+# Register your models here.
+class WorkoutExerciseInline(admin.StackedInline):
+    model = WorkoutExercise
+    ordering = ("pk",)
+    extra = 0
+
+class SupersetInline(admin.StackedInline):
+    model = Superset
+    ordering = ("pk",)
+    extra = 0
+
+@admin.register(Workout)
+class WorkoutAdmin(ImportExportModelAdmin):
+    search_fields = ['name', 'description', ]
+    list_display = ("name", "description", "estimated_time", "registered_user", "created_at", "updated_at")
+    inlines = [
+        SupersetInline,
+    ]
+
+@admin.register(WorkoutExercise)
+class WorkoutExerciseAdmin(admin.ModelAdmin):
+    list_display = ("workout", "exercise", "weight", "reps", "created_at", "updated_at")
+
+@admin.register(CompletedWorkout)
+class CompletedWorkoutAdmin(admin.ModelAdmin):
+    list_display = ("registered_user", "workout", "workout_start_time", "difficulty", "notes", "total_time", "created_at",)
+    
+@admin.register(PlannedWorkout)
+class PlannedWorkoutAdmin(admin.ModelAdmin):
+    def formfield_for_foreignkey(self, db_field, request, **kwargs):
+        formfield = super().formfield_for_foreignkey(db_field, request, **kwargs)
+        if db_field.name == "workout":            
+            formfield.queryset = Workout.objects.all().order_by('name')        
+        
+        return formfield
+    
+    list_display = ("registered_user", "workout", "on_date",)
+
+    ordering = ['-on_date']