From 5664726fcac8170631b21edaa87359a3dd39d496 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Valentin=20Mogu=C3=A9rou?= <valentin@mp2i-vms.fr>
Date: Fri, 17 May 2024 00:46:48 +0200
Subject: [PATCH] admin panel

---
 colloscope/admin.py | 38 +++++++++++++++++++++++++++++++++++---
 1 file changed, 35 insertions(+), 3 deletions(-)

diff --git a/colloscope/admin.py b/colloscope/admin.py
index 2376c8c..9f908a1 100644
--- a/colloscope/admin.py
+++ b/colloscope/admin.py
@@ -1,4 +1,5 @@
 from django.contrib import admin
+from django.utils.translation import gettext_lazy as _
 from colloscope.models import *
 
 
@@ -39,10 +40,41 @@ class SlotAdmin(admin.ModelAdmin):
         jours = ["dimanche", "lundi", "mardi", "mercredi", "jeudi", "vendredi", "samedi"]
         return jours[obj.day]
 
-    view_day.short_description = 'Day'
+    view_day.short_description = _('Day')
+
+
+class SwapInline(admin.StackedInline):
+    model = Swap
+    raw_id_fields = ("colle",)
+
+
+@admin.register(Colle)
+class ColleAdmin(admin.ModelAdmin):
+    list_display = ('view_subject', 'view_colleur', 'view_room', 'datetime',)
+    list_filter = ('slot',)
+    inlines = [SwapInline]
+
+    def view_subject(self, obj):
+        return obj.slot.subject
+
+    view_subject.short_description = _('Subject')
+
+    def view_colleur(self, obj):
+        return obj.slot.colleur
+
+    def view_room(self, obj):
+        return obj.slot.room
+
+    view_subject.short_description = _('Subject')
+    view_colleur.short_description = _('Colleur')
+    view_room.short_description = _('Room')
+
+
+@admin.register(Swap)
+class SwapAdmin(admin.ModelAdmin):
+    list_display = ('slot', 'enroll', 'student')
+    list_filter = ('slot', 'enroll', 'student')
 
 
-admin.site.register(Colle)
-admin.site.register(Swap)
 admin.site.register(Profile)
 admin.site.register(CalendarLink)