|
# -*- coding: utf-8 -*-
|
|
|
|
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'):
|
|
service = request.COOKIES['home-service'].upper().replace('-', ' ')
|
|
qs = qs.filter(service__name=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
|
|
|