Merge branch 'master' of github.com:FinalsClub/karmaworld

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

diff --git a/karmaworld/apps/notes/admin.py b/karmaworld/apps/notes/admin.py
index 650e9cba5c511d3db0b3931763b843df27f725b7..91eab81068fd426a195718d7148ebd5dd63e1db3 100644 (file)
--- a/karmaworld/apps/notes/admin.py
+++ b/karmaworld/apps/notes/admin.py
@@ -5,8 +5,16 @@
 
 from django.contrib import admin
 
-from karmaworld.apps.notes import models
+from karmaworld.apps.notes.models import DriveAuth
+from karmaworld.apps.notes.models import Note
 
+class NoteAdmin(admin.ModelAdmin):
+    """ an Admin handler for the Note model that handles autocomplete to Course 
+    """
+    raw_id_fields = ('course',)
+    autocomplete_lookup_fields = {
+        'fk': ['course']
+    }
 
-admin.site.register(models.DriveAuth)
-admin.site.register(models.Note)
+admin.site.register(DriveAuth)
+admin.site.register(Note, NoteAdmin)