Optionally use FKautocomplete if available

catalpainternational · Jun 28, 2018 · c3d1120 · c3d1120
1 parent 9d594ba
commit c3d1120
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/simple_locations/admin.py b/simple_locations/admin.py
@@ -9,6 +9,15 @@
 
 from simple_locations.models import Point, AreaType, Area
 
+try:
+    # optionally use django_extensions' ForeignKeyAutocompleteAdmin if available
+    from django_extensions.admin import ForeignKeyAutocompleteAdmin
+    class MPTTModelAutocompleteAdmin(MPTTModelAdmin, ForeignKeyAutocompleteAdmin):
+        pass
+except ImportError:
+    class MPTTModelAutocompleteAdmin(MPTTModelAdmin):
+        pass
+
 
 class PointAdmin(admin.ModelAdmin):
     list_display = ('id', 'latitude', 'longitude')
@@ -18,11 +27,12 @@ class AreaTypeAdmin(admin.ModelAdmin):
     list_display = ('slug', 'name')
 
 
-class AreaAdmin(MPTTModelAdmin):
+class AreaAdmin(MPTTModelAutocompleteAdmin):
     list_display = ( 'name', 'kind', 'location', 'code')
     search_fields = ['code', 'name']
     list_filter = ('kind',)
-
+    related_search_fields = {'parent': ('^name',)}
+
 
 admin.site.register(Point, PointAdmin)
 admin.site.register(AreaType, AreaTypeAdmin)