52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
from selectable.base import ModelLookup
|
|
from selectable.registry import registry
|
|
|
|
from django.contrib.auth.models import User
|
|
from regluit.core.models import Work, PublisherName, Edition
|
|
|
|
class OwnerLookup(ModelLookup):
|
|
model = User
|
|
search_fields = ('username__icontains',)
|
|
|
|
class WorkLookup(ModelLookup):
|
|
model = Work
|
|
search_fields = ('title__istartswith',)
|
|
filters = {'language': 'en', }
|
|
def get_item_label(self,item):
|
|
return "%s (%s)"%(item.title,item.id)
|
|
|
|
def get_query(self, request, term):
|
|
results = super(WorkLookup, self).get_query(request, term)
|
|
language = request.GET.get('language', 'en')
|
|
results = results.filter(language=language)
|
|
return results
|
|
|
|
class PublisherNameLookup(ModelLookup):
|
|
model = PublisherName
|
|
search_fields = ('name__icontains',)
|
|
def create_item(self, value):
|
|
publisher_name, created = PublisherName.objects.get_or_create(name=value)
|
|
publisher_name.save()
|
|
return publisher_name
|
|
|
|
class EditionLookup(ModelLookup):
|
|
model = Edition
|
|
search_fields = ('title__icontains',)
|
|
filters = {'ebooks__isnull': False, }
|
|
|
|
def get_query(self, request, term):
|
|
return super(EditionLookup, self).get_query(request, term).distinct()
|
|
|
|
def get_item(self, value):
|
|
item = None
|
|
if value:
|
|
try:
|
|
item = Edition.objects.get(pk=value)
|
|
except (ValueError, Edition.DoesNotExist):
|
|
item = None
|
|
return item
|
|
|
|
registry.register(OwnerLookup)
|
|
registry.register(WorkLookup)
|
|
registry.register(PublisherNameLookup)
|
|
registry.register(EditionLookup) |