From 6044b3522b2511eae088a40338115c902a9b0f82 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Thu, 3 Aug 2017 20:56:56 +0200
Subject: Admin: improve operation form

---
 archaeological_operations/lookups.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

(limited to 'archaeological_operations/lookups.py')

diff --git a/archaeological_operations/lookups.py b/archaeological_operations/lookups.py
index b6fdaad23..690b3c566 100644
--- a/archaeological_operations/lookups.py
+++ b/archaeological_operations/lookups.py
@@ -1,7 +1,7 @@
 from ajax_select import register, LookupChannel
 
 from django.db.models import Q
-from archaeological_operations.models import Operation
+from archaeological_operations.models import Operation, ArchaeologicalSite
 
 
 @register('operation')
@@ -19,3 +19,21 @@ class OperationLookup(LookupChannel):
 
     def format_item_display(self, item):
         return u"<span class='ajax-label'>%s</span>" % item.cached_label
+
+
+@register('archaeological_site')
+class ArchaeologicalSiteLookup(LookupChannel):
+    model = ArchaeologicalSite
+
+    def get_query(self, q, request):
+        query = Q()
+        for term in q.strip().split(' '):
+            subquery = (
+                Q(reference__icontains=term) |
+                Q(name__icontains=term)
+            )
+            query &= subquery
+        return self.model.objects.filter(query).order_by('cached_label')[:20]
+
+    def format_item_display(self, item):
+        return u"<span class='ajax-label'>%s</span>" % unicode(item)
-- 
cgit v1.2.3