From 68f0b07848478b525898b73e0788f92bb6670210 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Wed, 18 Apr 2018 18:12:11 +0200
Subject: Associate area to userprofiles (refs #4060)

---
 ishtar_common/lookups.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

(limited to 'ishtar_common/lookups.py')

diff --git a/ishtar_common/lookups.py b/ishtar_common/lookups.py
index f409536da..181fc97a1 100644
--- a/ishtar_common/lookups.py
+++ b/ishtar_common/lookups.py
@@ -2,7 +2,8 @@ from ajax_select import register, LookupChannel as BaseLookupChannel
 
 from django.conf import settings
 from django.db.models import Q
-from ishtar_common.models import Person, Organization, IshtarUser, Town, Author
+from ishtar_common.models import Area, Person, Organization, IshtarUser, Town, \
+    Author
 
 
 class LookupChannel(BaseLookupChannel):
@@ -105,3 +106,15 @@ class UserLookup(LookupChannel):
 
     def format_item_display(self, item):
         return u"<span class='ajax-label'>%s</span>" % unicode(item.person)
+
+
+@register('area')
+class AreaLookup(LookupChannel):
+    model = Area
+
+    def get_query(self, q, request):
+        query = Q(label__icontains=q)
+        return self.model.objects.filter(query).order_by('label')[:20]
+
+    def format_item_display(self, item):
+        return u"<span class='ajax-label'>%s</span>" % unicode(item)
-- 
cgit v1.2.3