1 files changed, 21 insertions, 1 deletions
diff --git a/ishtar_common/lookups.py b/ishtar_common/lookups.py
index a031af9b3..ef2ff81f1 100644
--- a/ishtar_common/lookups.py
+++ b/ishtar_common/lookups.py
@@ -1,7 +1,27 @@
 from ajax_select import register, LookupChannel
 
+from django.conf import settings
 from django.db.models import Q
-from ishtar_common.models import Person, Organization, IshtarUser
+from ishtar_common.models import Person, Organization, IshtarUser, Town
+
+
+@register('town')
+class TownLookup(LookupChannel):
+    model = Town
+
+    def get_query(self, q, request):
+        if settings.COUNTRY == 'fr':
+            query = Q()
+            for term in q.strip().split(' '):
+                query &= (
+                    Q(name__icontains=term) | Q(numero_insee__icontains=term)
+                )
+        else:
+            query = Q(name__icontains=q)
+        return self.model.objects.filter(query).order_by('name')[:20]
+
+    def format_item_display(self, item):
+        return u"<span class='ajax-label'>%s</span>" % unicode(item)
 
 
 @register('organization')