Admin: add lookups for finds, base finds, warehouses and containers

2 files changed, 133 insertions, 0 deletions

diff --git a/archaeological_finds/lookups.py b/archaeological_finds/lookups.py
new file mode 100644
index 000000000..1449ff52b
--- /dev/null
+++ b/archaeological_finds/lookups.py
@@ -0,0 +1,84 @@
+from ajax_select import register, LookupChannel
+
+from django.db.models import Q
+from django.utils.encoding import force_text
+from django.utils.html import escape
+
+from archaeological_finds import models
+
+
+@register('base_find')
+class BaseFindLookup(LookupChannel):
+    model = models.BaseFind
+
+    def get_query(self, q, request):
+        query = Q()
+        for term in q.strip().split(' '):
+            subquery = (
+                Q(cache_complete_id__icontains=term) |
+                Q(label__icontains=term)
+            )
+            query &= subquery
+        return self.model.objects.filter(query).order_by(
+            'cache_complete_id')[:20]
+
+    def format_item_display(self, item):
+        return u"<span class='ajax-label'>%s</span>" % item.cache_complete_id
+
+
+@register('find')
+class FindLookup(LookupChannel):
+    model = models.Find
+
+    def get_query(self, q, request):
+        query = Q()
+        for term in q.strip().split(' '):
+            subquery = (
+                Q(cached_label__icontains=term) |
+                Q(base_finds__cache_complete_id__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.full_label
+
+    def format_match(self, obj):
+        return escape(force_text(obj.full_label))
+
+
+@register('treatment')
+class TreatmentLookup(LookupChannel):
+    model = models.Treatment
+
+    def get_query(self, q, request):
+        query = Q()
+        for term in q.strip().split(' '):
+            subquery = (
+                Q(cached_label__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
+
+
+@register('treatment_file')
+class TreatmentFileLookup(LookupChannel):
+    model = models.TreatmentFile
+
+    def get_query(self, q, request):
+        query = Q()
+        for term in q.strip().split(' '):
+            subquery = (
+                Q(cached_label__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
diff --git a/archaeological_warehouse/lookups.py b/archaeological_warehouse/lookups.py
new file mode 100644
index 000000000..57d3a892d
--- /dev/null
+++ b/archaeological_warehouse/lookups.py
@@ -0,0 +1,49 @@
+from ajax_select import register, LookupChannel
+
+from django.db.models import Q
+from django.utils.encoding import force_text
+from django.utils.html import escape
+
+from archaeological_warehouse import models
+
+
+@register('container')
+class ContainerLookup(LookupChannel):
+    model = models.Container
+
+    def get_query(self, q, request):
+        query = Q()
+        for term in q.strip().split(' '):
+            subquery = (
+                Q(reference__icontains=term) |
+                Q(container_type__label__icontains=term) |
+                Q(cached_label__icontains=term) |
+                Q(responsible__name__icontains=term)
+            )
+            query &= subquery
+        return self.model.objects.filter(query).order_by(
+            'cached_label')[:20]
+
+    def format_match(self, obj):
+        return escape(force_text(obj.cached_label))
+
+    def format_item_display(self, item):
+        return u"<span class='ajax-label'>%s</span>" % item.cached_label
+
+
+@register('warehouse')
+class WarehouseLookup(LookupChannel):
+    model = models.Warehouse
+
+    def get_query(self, q, request):
+        query = Q()
+        for term in q.strip().split(' '):
+            subquery = (
+                Q(warehouse_type__label=term) |
+                Q(name__icontains=term)
+            )
+            query &= subquery
+        return self.model.objects.filter(query).order_by('name')[:20]
+
+    def format_item_display(self, item):
+        return u"<span class='ajax-label'>%s</span>" % item.name