Replace explicit Prof/Course model with ManyToMany relationship.

[oweals/karmaworld.git] / karmaworld / apps / courses / admin.py

diff --git a/karmaworld/apps/courses/admin.py b/karmaworld/apps/courses/admin.py
index e484058cbc92b9d9ca999ece48c131a92e399dee..bfafca140c59b2be68d715c44872b5cdc7a6760e 100644 (file)
--- a/karmaworld/apps/courses/admin.py
+++ b/karmaworld/apps/courses/admin.py
@@ -3,9 +3,30 @@
 # Copyright (C) 2012  FinalsClub Foundation
 """ Administration configuration for courses """
 
+import reversion
+
 from django.contrib import admin
 
-import models
+from karmaworld.apps.courses.models import School
+from karmaworld.apps.courses.models import Course
+from karmaworld.apps.courses.models import Professor
+from karmaworld.apps.courses.models import Department
+
+class CourseAdmin(reversion.VersionAdmin, admin.ModelAdmin):
+    """ an Admin handler for the Course model that handles fk search """
+    raw_id_fields = ('department',)
+    autocomplete_lookup_fields = {
+        'fk': ['department']
+    }
+
+class DepartmentAdmin(reversion.VersionAdmin, admin.ModelAdmin):
+    """ an Admin handler for the Department model that handles fk search """
+    raw_id_fields = ('school',)
+    autocomplete_lookup_fields = {
+        'fk': ['school']
+    }
 
-admin.site.register(models.School)
-admin.site.register(models.Course)
+admin.site.register(School)
+admin.site.register(Professor)
+admin.site.register(Course, CourseAdmin)
+admin.site.register(Department, DepartmentAdmin)