summaryrefslogtreecommitdiff
path: root/archaeological_files/lookups.py
blob: 4de71341940df662b6e1268e095732670db2a2da (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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