1 files changed, 32 insertions, 0 deletions
diff --git a/archaeological_files/lookups.py b/archaeological_files/lookups.py
new file mode 100644
index 000000000..4de713419
--- /dev/null
+++ b/archaeological_files/lookups.py
@@ -0,0 +1,32 @@
+from ajax_select import register, LookupChannel
+
+from django.db.models import Q
+from archaeological_files.models import File
+
+
+@register('file')
+class FileLookup(LookupChannel):
+    model = File
+
+    def get_query(self, q, request):
+        query = Q()
+        for term in q.strip().split(' '):
+             subquery = (
+                Q(towns__name__icontains=term) |
+                Q(towns__numero_insee__icontains=term) |
+                Q(main_town__name__icontains=term) |
+                Q(main_town__numero_insee__icontains=term) |
+                Q(name__icontains=term) |
+                Q(internal_reference__icontains=term) |
+                Q(permit_reference__icontains=term)
+             )
+             try:
+                 num = int(term)
+                 subquery |=   Q(year=num) | Q(numeric_reference=num)
+             except ValueError:
+                 pass
+             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