Produits Entr'ouvert » Calebasse

import itertools

from calebasse.lookups import CalebasseLookup

from calebasse.personnes.models import Worker

from calebasse.ressources.models import Service

class FakeGroup:

    pk = None

    label = None

    def __init__(self, pk, label):

        self.pk = pk

        self.label = label

    def __unicode__(self):

        return u'Groupe: %s' % self.label

class WorkerOrGroupLookup(CalebasseLookup):

    model = Worker

    search_field = 'last_name'

    enabled = True

    def get_query(self, q, request):

        service = None

        if q.startswith('group:'):

            service = Service.objects.get(pk=int(q[6:]))

        else:

            try:

                service = Service.objects.get(name__iexact=q)

            except Service.DoesNotExist:

                pass

        if service:

            kwargs = { "%s__istartswith" % self.search_field : q }

            group = FakeGroup('group:%s' % service.id, service.name)

            return itertools.chain([group], self.model.objects.for_service(service.id).order_by('last_name'))

        kwargs = { "%s__istartswith" % self.search_field : q }

        if self.enabled:

            return self.model.objects.filter(enabled=True).filter(**kwargs).order_by('last_name')

        return self.model.objects.filter(**kwargs).order_by('last_name')

    def get_result(self, obj):

        return self.format_item_display(obj)

    def format_match(self, obj):

        return self.format_item_display(obj)

    def format_item_display(self,obj):

        if isinstance(obj, FakeGroup):

            text = obj.label

        else:

            text = obj.last_name.upper() + ' ' + obj.first_name

        return unicode(text)

class AllWorkerOrGroupLookup(WorkerOrGroupLookup):

    enabled = False