ensuring that training and training participant are part of admin

Signed-off-by: vsoch <vsochat@stanford.edu>
vsoch · Sep 4, 2020 · b7c2afc · b7c2afc
1 parent 20901ea
commit b7c2afc
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/mhttc/apps/main/admin.py b/mhttc/apps/main/admin.py
@@ -9,7 +9,7 @@
 """
 
 from django.contrib import admin
-from mhttc.apps.main.models import Project, FormTemplate
+from mhttc.apps.main.models import Project, FormTemplate, Training, TrainingParticipant
 
 
 class ProjectAdmin(admin.ModelAdmin):
@@ -22,6 +22,23 @@ class ProjectAdmin(admin.ModelAdmin):
     )
 
 
+class TrainingAdmin(admin.ModelAdmin):
+    list_display = (
+        "name",
+        "description",
+        "center",
+        "contact",
+    )
+
+
+class TrainingParticipantAdmin(admin.ModelAdmin):
+    list_display = (
+        "name",
+        "email",
+        "training",
+    )
+
+
 class FormTemplateAdmin(admin.ModelAdmin):
     list_display = (
         "name",
@@ -57,5 +74,7 @@ class FormTemplateAdmin(admin.ModelAdmin):
     )
 
 
+admin.site.register(Training, TrainingAdmin)
+admin.site.register(TrainingParticipant, TrainingParticipantAdmin)
 admin.site.register(FormTemplate, FormTemplateAdmin)
 admin.site.register(Project, ProjectAdmin)