26 lines
773 B
Python
26 lines
773 B
Python
from selectable.base import ModelLookup
|
|
from selectable.registry import registry
|
|
|
|
from django.contrib.auth.models import User
|
|
from regluit.core.models import Work
|
|
|
|
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
|
|
|
|
|
|
registry.register(OwnerLookup)
|
|
registry.register(WorkLookup) |