clinic/netforce_clinic/models/visit_board.py

import time
from calendar import monthrange
from datetime import datetime, timedelta

from netforce.model import Model, fields, get_model
from netforce.access import get_active_company

from . import utils

DRT=0
HD_STATE={
    "draft":"Draft",
    'waiting_treatment':'Waiting Treatment',
    "in_progress":"In Progress",
    "completed":"Finish Treatment",
    'paid':'Paid',
    "waiting_payment":"Waiting Payment",
    "discountinued":"Discountinued",
    "cancelled":"Cancelled"
}

DAYS={
    'mon': 0,
    'tue': 1,
    'wed': 2,
    'thu': 3,
    'fri': 4,
    'sat': 5,
    'sun': 6,
}

class VisitBoard(Model):
    _name="clinic.visit.board"
    _string="Visit Board"
    _transient=True
    _name_field="date" 

    _fields={
        "date": fields.Date("Month", required=False),
        "date_from": fields.Date("From", required=True),
        "date_to": fields.Date("To", required=True),
        'patient_id': fields.Many2One("clinic.patient","Patient",domain=[['state','=','admit']]),
        'cycle_id': fields.Many2One("clinic.cycle","Cycle"),
        'doctor_id': fields.Many2One("clinic.staff","Doctor",domain=[["type","=","doctor"]]),
        'department_id': fields.Many2One("clinic.department","Department"),
        'branch_id': fields.Many2One("clinic.branch","Branch"),
    }
    
    def default_get(self,field_names=None,context={},**kw):
        defaults=context.get("defaults",{})
        date=defaults.get('date',time.strftime("%Y-%m-%d"))
        date_from=defaults.get("date_from",time.strftime("%Y-%m-%d"))
        date_to=defaults.get("date_to")
        if not date_to:
            date_to=(datetime.now()+timedelta(days=DRT)).strftime("%Y-%m-%d")
        branch_id=defaults.get("branch_id")
        department_id=defaults.get("department_id")
        res=get_model('select.company').get_select()
        if res:
            if not branch_id:
                branch_id=res['branch_id']
            if not department_id:
                if res.get('department_ids'):
                    department_id=res['department_ids'][0]
                else:
                    department_id=res['department_id']
        res={
            'date': date,
            'date_from': date_from,
            'date_to': date_to,
            'branch_id': branch_id,
            'department_id': department_id,
        }
        return res
    
    def get_report_data(self,ids,context={}):
        company_id=get_active_company()
        company=get_model("company").browse(company_id)
        date_from=datetime.now().strftime("%Y-%m-%d")
        date_to=(datetime.now()+timedelta(days=DRT)).strftime("%Y-%m-%d")
        patient_id=None
        cycle_id=None
        doctor_id=None
        defaults=self.default_get(context=context)
        department_id=defaults.get("department_id",None)
        branch_id=defaults.get("branch_id",None)
        print('defaults ', defaults)
        if ids:
            obj=self.browse(ids)[0]
            date_from=obj.date_from
            date_to=obj.date_to
            patient_id=obj.patient_id.id
            cycle_id=obj.cycle_id.id
            doctor_id=obj.doctor_id.id
            department_id=obj.department_id.id
            branch_id=obj.branch_id.id
        def auto_gen_visit(dom=[]):
            dom.append(['dispose','=',False])
            dom.append(['walkin','=','no'])
            def daterange(start_date, end_date):
                for n in range(int ((end_date - start_date).days)):
                    yield start_date + timedelta(n)
            def convert_date(date_txt):
                try:
                    if not date_txt:
                        raise Exception("Date Empty")
                    return datetime.strptime(date_txt,"%Y-%m-%d")
                except:
                    raise Exception("Wrong Format Date")
            # IMPROVE
            # check day of week of this day
            for date in daterange(convert_date(date_from),convert_date(date_to)+timedelta(days=1)):
                weekday=date.weekday()
                date_txt=date.strftime("%Y-%m-%d")
                datenow=time.strftime("%Y-%m-%d")
                if date_txt < datenow:
                    print("continue ", date_txt, datenow)
                    continue
                for pt in get_model("clinic.patient").search_browse(dom):
                    for pc in pt.cycles:
                        w=DAYS.get(pc.day,0) #default monday
                        cycle=pc.cycle_id
                        department=pc.department_id
                        if weekday==w:
                            dom2=[
                                ['visit_date','>=',date_txt],
                                ['visit_date','<=',date_txt],
                                ['patient_id','=',pt.id],
                                ['cycle_id','=',cycle.id],
                                ['department_id','=',department.id],
                            ]
                            res=get_model("clinic.visit").search(dom2)
                            # create visit auto
                            if not res:
                                vals={
                                    'patient_id': pt.id,
                                    'department_id': department.id,
                                    'cycle_id': cycle.id,
                                    'doctor_id': pt.doctor_id.id,
                                    'branch_id': department.branch_id.id,
                                    'time_start': '%s %s:00'%(date_txt,cycle.time_start),
                                    'time_stop': '%s %s:00'%(date_txt,cycle.time_stop),
                                    'visit_date': date_txt,
                                    'state': 'pending',
                                }
                                visit_id=get_model("clinic.visit").create(vals)
                                print('create new visit %s for %s'%(visit_id,pt.name))
        time_start='%s 00:00:00'%(date_from)
        time_stop='%s 23:59:59'%(date_to)
        dom=[]
        dom.append(['time_start','>=','%s'%time_start])
        dom.append(['time_stop','<=','%s'%time_stop])
        if patient_id:
            dom.append(['patient_id','=',patient_id])
        if cycle_id:
            dom.append(['cycle_id','=',cycle_id])
        if doctor_id:
            dom.append(['doctor_id','=',doctor_id])
        if department_id:
            dom.append(['department_id','=',department_id])
        if branch_id:
            dom.append(['branch_id','=',branch_id])
        #gen visit
        if department_id:
            auto_gen_visit(dom=[['department_id','=',department_id]])
        elif branch_id:
            auto_gen_visit(dom=[['branch_id','=',branch_id]])
        lines=[]
        empty_line={
            'no': '',
            'number': '',
            'visit_id': None,
            'visit_state': '',
            'cycle_name': '',
            'cycle_color': '',
            'department_id': None,
            'department_name': None,
            'branch_id': None,
            'branch_name': None,
            'patient_id': None,
            'patient_name': '',
            'patient_type': '',
            'patient_type_id': None,
            'doctor_name': '',
            'doctor_id': None,
            'hd_case_number': '',
            'hd_case_id': None,
            'success_color': '',
            'date':'',
            'title': True,
            'footer': False,
            'details':'',
            'hd_case_state': '',
            'hn': '',
        }
        
        patient_types={t['id']:t['name'] for t in get_model("clinic.patient.type").search_read([[]],['name'])}
        cycle_names={t['id']:t['name'] for t in get_model("clinic.cycle").search_read([[]],['name'])}
        brch_names={t['id']:t['name'] for t in get_model("clinic.branch").search_read([[]],['name'])}
        dpt_names={t['id']:t['name'] for t in get_model("clinic.department").search_read([[]],['name'])}
        total_wait=0
        total_done=0
        total_cancel=0
        types={}
        cycles={}
        no=1
        dpt={}
        brch={}
        for visit in get_model("clinic.visit").search_browse(dom):
            if visit.state in ('draft','pending'):
                total_wait+=1
            elif visit.state in ('confirmed'):
                total_done+=1
            else:   
                total_cancel+=1

            hd_case_id=None
            hd_case_number=''
            hd_case_state=''
            visit_color=''
            sickbed_name='N/A'
            sickbed_id=None
            if visit.hd_cases:
                hd_case=visit.hd_cases[0]
                sickbed_name=hd_case.sickbed_id.name or "N/A"
                sickbed_id=hd_case.sickbed_id.id
                hd_case_id=hd_case.id,
                if hd_case.number=='/':
                    hd_case_number='Waiting'
                else:
                    hd_case_number=hd_case.number
                hd_case_state=hd_case.state
                if hd_case_state=='completed':
                    #visit_color='#99ff99'
                    visit_color=''
                elif hd_case_state=='in_progress':
                    #visit_color='#f9e37d'
                    visit_color=''
                elif hd_case_state=='cancelled':
                    visit_color='#dbdbdb',
                    hd_case_number="Cancelled"
            number=visit.number
            if number=='/':
                number='Waiting'
            cycle=visit.cycle_id
            patient=visit.patient_id
            if not patient.active:
                print('skip not active ', patient.name)
                continue
            branch=visit.branch_id
            department=visit.department_id
            if not branch:
                branch=department.branch_id 
           
            hn_name=patient.hn_no or '-'

            visit_date=visit.visit_date
            if visit.state=='cancelled':
                visit_color='#dbdbdb',
                if visit.number=='/':
                    number='Cancelled'
                else:
                    number+='Cancelled'
            line={
                'number': number,
                'hn_name': hn_name,
                'visit_id': visit.id,
                'sickbed_name': sickbed_name,
                'sickbed_id': sickbed_id,
                'cycle_name': cycle.name,
                'cycle_color': cycle.color,
                'department_id': department.id,
                'department_name': department.name or '',
                'branch_id': branch and branch.id or None,
                'branch_name': branch and branch.name or '',
                'patient_name': patient.name,
                'patient_id': patient.id,
                'patient_type': patient.type_id.name or "",
                'patient_type_id': patient.type_id.id or None,
                'doctor_name': visit.doctor_id.name,
                'doctor_id': visit.doctor_id.id,
                'hd_case_number': hd_case_number,
                'hd_case_state': hd_case_state,
                'hd_case_state_txt':HD_STATE.get(hd_case_state,''),
                'hd_case_id': hd_case_id,
                'visit_color':  visit_color,
                'date': visit_date,
                'title': False,
                'footer': False,
                'details':'',
                'details1':'',
                'details2':'',
                'details3':'',
                'details4':'',
                'details5':'',
                'no': no,
                'note': visit.note,
            }
            lines.append(line)
            no+=1
            if not types.get(visit_date):
                ptype={}
                [ptype.setdefault(t,0) for t in patient_types.keys()]
                types[visit_date]=ptype
            types[visit_date][patient.type_id.id]+=1

            if not cycles.get(visit_date):
                cycle_name={}
                [cycle_name.setdefault(cid,0) for cid in cycle_names.keys()]
                cycles[visit_date]=cycle_name
            cycles[visit_date][cycle.id]+=1

            if not brch.get(visit_date):
                brch_name={}
                [brch_name.setdefault(cid,0) for cid in brch_names.keys()]
                brch[visit_date]=brch_name
            brch[visit_date][branch.id]+=1

            if not dpt.get(visit_date):
                dpt_name={}
                [dpt_name.setdefault(cid,0) for cid in dpt_names.keys()]
                dpt[visit_date]=dpt_name
            dpt[visit_date][department.id]+=1
        
        dates=[]
        index=0
        total_qty=0
        count=0
        for line in lines:
            date=line['date']
            if not date:
                continue
            if date not in dates:
                count=0
                total_qty=0
                line=empty_line.copy()
                for qty in types[date].values():
                    total_qty+=qty
                line['cycle_name']=utils.date2thai(date,format='%(Td)s %(d)s %(Tm)s',lang="th_TH2"),
                lines.insert(index,line)
                dates.append(date)
            if count==total_qty and not patient_id:
                index+=1
                # footer
                line=empty_line.copy()
                patient_str='%s'%', '.join('%s: %s'%(patient_types[k],v) for k,v in types[date].items())
                cycle_str='%s'%', '.join('%s: %s'%(cycle_names[k],v) for k,v in cycles[date].items())
                brch_str='%s'%', '.join('%s: %s'%(brch_names[k],v) for k,v in brch[date].items())
                dpt_str='%s'%', '.join('%s: %s'%(dpt_names[k],v) for k,v in dpt[date].items())
                summary_str='%s'%', '.join(['รับไว้: %s'%total_wait,'จำหน่ายแล้ว: %s'%total_done, 'ยกเลิก: %s'%total_cancel])
                line['details']='ทั้งหมด %s:  %s'%(total_qty,', '.join([summary_str,cycle_str,patient_str]))
                line['details1']=brch_str
                line['details2']=summary_str
                line['details3']=cycle_str
                line['details4']=patient_str
                line['details5']=dpt_str
                line['footer']=True
                line['title']=False
                lines.insert(index,line)
            count+=1
            index+=1

        has_duration=False
        if date_from != date_to:
            has_duration=True
        data={
            'lines': lines,
            'date': utils.date2thai(date_from,format='ประจำวัน%(Td)s ที่ %(d)s %(Tm)s %(BY)s'),
            'company_name': company.name,
            'company_parent_name': company.parent_id.name,
            'has_duration': has_duration,
            'date_from': utils.date2thai(date_from,format='%(d)s %(Tm)s %(By)s',lang="th_TH2"),
            'date_to': utils.date2thai(date_to,format='%(d)s %(Tm)s %(By)s',lang="th_TH2"),
        }
        return data

    def onchange_date(self,context={}):
        data=context['data']
        date=data['date']
        year,month,day=date.split("-")
        weekday, total_day=monthrange(int(year), int(month))
        data['date_from']="%s-%s-01"%(year,month)
        data['date_to']="%s-%s-%s"%(year,month,total_day)
        return data

    def onchange_datefrom(self,context={}):
        data=context['data']
        data['date_to']=data['date_from']
        return data

    def onchange_branch(self,context={}):
        data=context['data']
        data['department_id']=None
        return data

    def confirm(self,ids,context={}):
        obj=self.browse(ids)[0]
        return {
            'next': {
                'refer_id': obj.id,
                'name': 'clinic_popup_visit_confirm',
            }
        }

VisitBoard.register()