From 5063bf07c5128131f5eecbce994cfeb670ef8a08 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Fri, 4 Aug 2017 17:11:52 +0200
Subject: Admin context records: add context record lookup, improve admin forms

---
 archaeological_context_records/lookups.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 archaeological_context_records/lookups.py

(limited to 'archaeological_context_records/lookups.py')

diff --git a/archaeological_context_records/lookups.py b/archaeological_context_records/lookups.py
new file mode 100644
index 000000000..92115d98b
--- /dev/null
+++ b/archaeological_context_records/lookups.py
@@ -0,0 +1,23 @@
+from ajax_select import register, LookupChannel
+
+from django.db.models import Q
+
+from archaeological_context_records import models
+
+
+@register('context_record')
+class ContextRecordLookup(LookupChannel):
+    model = models.ContextRecord
+
+    def get_query(self, q, request):
+        query = Q()
+        for term in q.strip().split(' '):
+            subquery = (
+                Q(cached_label__icontains=term) |
+                Q(parcel__town__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>" % item.cached_label
-- 
cgit v1.2.3