diff options
-rw-r--r-- | archaeological_context_records/lookups.py | 4 | ||||
-rw-r--r-- | archaeological_files/lookups.py | 4 | ||||
-rw-r--r-- | archaeological_finds/lookups.py | 14 | ||||
-rw-r--r-- | archaeological_operations/lookups.py | 4 | ||||
-rw-r--r-- | archaeological_warehouse/lookups.py | 4 | ||||
-rw-r--r-- | ishtar_common/lookups.py | 14 |
6 files changed, 28 insertions, 16 deletions
diff --git a/archaeological_context_records/lookups.py b/archaeological_context_records/lookups.py index 92115d98b..3d7b16d4a 100644 --- a/archaeological_context_records/lookups.py +++ b/archaeological_context_records/lookups.py @@ -1,4 +1,6 @@ -from ajax_select import register, LookupChannel +from ajax_select import register + +from ishtar_common.lookups import LookupChannel from django.db.models import Q diff --git a/archaeological_files/lookups.py b/archaeological_files/lookups.py index 90b904945..dab3fb38c 100644 --- a/archaeological_files/lookups.py +++ b/archaeological_files/lookups.py @@ -1,4 +1,6 @@ -from ajax_select import register, LookupChannel +from ajax_select import register + +from ishtar_common.lookups import LookupChannel from django.db.models import Q from archaeological_files.models import File diff --git a/archaeological_finds/lookups.py b/archaeological_finds/lookups.py index 5b30035ad..c1298fb83 100644 --- a/archaeological_finds/lookups.py +++ b/archaeological_finds/lookups.py @@ -1,4 +1,6 @@ -from ajax_select import register, LookupChannel +from ajax_select import register + +from ishtar_common.lookups import LookupChannel from django.db.models import Q from django.utils.encoding import force_text @@ -11,16 +13,6 @@ from archaeological_finds import models class BaseFindLookup(LookupChannel): model = models.BaseFind - def get_objects(self, items): - # TODO: why IDs are not given here? M2M issue - ids = [] - for item in items: - if hasattr(item, 'pk'): - ids.append(item.pk) - else: - ids.append(item) - return super(BaseFindLookup, self).get_objects(ids) - def get_query(self, q, request): query = Q() for term in q.strip().split(' '): diff --git a/archaeological_operations/lookups.py b/archaeological_operations/lookups.py index 83cf6fa3a..c4027c2d2 100644 --- a/archaeological_operations/lookups.py +++ b/archaeological_operations/lookups.py @@ -1,4 +1,6 @@ -from ajax_select import register, LookupChannel +from ajax_select import register + +from ishtar_common.lookups import LookupChannel from django.db.models import Q from django.utils.encoding import force_text diff --git a/archaeological_warehouse/lookups.py b/archaeological_warehouse/lookups.py index 57d3a892d..f4df88a7d 100644 --- a/archaeological_warehouse/lookups.py +++ b/archaeological_warehouse/lookups.py @@ -1,4 +1,6 @@ -from ajax_select import register, LookupChannel +from ajax_select import register + +from ishtar_common.lookups import LookupChannel from django.db.models import Q from django.utils.encoding import force_text diff --git a/ishtar_common/lookups.py b/ishtar_common/lookups.py index 9570f8bc6..f409536da 100644 --- a/ishtar_common/lookups.py +++ b/ishtar_common/lookups.py @@ -1,10 +1,22 @@ -from ajax_select import register, LookupChannel +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 +class LookupChannel(BaseLookupChannel): + def get_objects(self, items): + # TODO: why IDs are not given here? M2M issue + ids = [] + for item in items: + if hasattr(item, 'pk'): + ids.append(item.pk) + else: + ids.append(item) + return super(LookupChannel, self).get_objects(ids) + + @register('town') class TownLookup(LookupChannel): model = Town |