from django.contrib import admin
-import models
+from karmaworld.apps.notes.models import DriveAuth
+from karmaworld.apps.notes.models import Note
-admin.site.register(models.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(DriveAuth)
+admin.site.register(Note, NoteAdmin)