admin panel

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)