Produits Entr'ouvert » Calebasse

# -*- coding: utf-8 -*-

from calebasse.cbv import HOME_SERVICE_COOKIE

from calebasse.lookups import CalebasseLookup

from calebasse.dossiers.models import PatientRecord, PatientAddress

from itertools import chain

class PatientRecordLookup(CalebasseLookup):

    model = PatientRecord

    search_field = 'last_name'

    homonym = False

    def get_query(self,q,request):

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

        not_closed_filter_field = 'last_state__status__name'

        qs = self.model.objects.filter(**kwargs).order_by(self.search_field)

        if request.COOKIES.has_key(HOME_SERVICE_COOKIE):

            service = request.COOKIES[HOME_SERVICE_COOKIE]

            qs = qs.filter(service__slug=service)

        #nb = qs.count()

        #nb_distinct = qs.distinct('display_name').count()

        #if nb != nb_distinct:

        #    self.homonym = True

        qs.prefetch_related('last_state__status')

        qs.query.order_by = [not_closed_filter_field, ]

        # filtering all closed records to put them at the end

        separation_criteria = {'last_state__status__name': 'Clos'}

        closed = qs.filter(**separation_criteria)

        qs = qs.exclude(**separation_criteria)

        qs = qs.order_by('-%s' % not_closed_filter_field)

        return chain(qs, closed)

    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):

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

        if obj.paper_id or obj.last_state:

            text += u' ('

        if obj.paper_id:

            text += obj.paper_id

            if obj.last_state:

                text += u' - '

        if obj.last_state:

            text += obj.last_state.status.name

        if obj.paper_id or obj.last_state:

            text += u')'

        return unicode(text)

class PatientAddressLookup(CalebasseLookup):

    model = PatientAddress

    search_field = 'display_name'

    def get_query(self, q, request):

        qs = super(PatientAddressLookup, self).get_query(q, request)

        if request.session.has_key('patientrecord_id'):

            qs = qs.filter(patientcontact__id=request.session['patientrecord_id'])

        return qs