import time
from calendar import monthrange

from datetime import datetime, timedelta
#from calendar import monthrange
from netforce.model import Model, fields, get_model
from netforce.access import get_active_company, get_active_user

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"
}

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"),
        '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 _get_branch(self,context={}):
        user_id=get_active_user()
        sts=get_model("clinic.staff").search_browse([['user_id','=',user_id]])
        branch_id=None
        if sts:
            branch_id=sts[0].branch_id.id
        return branch_id

    def _get_deparment(self,context={}):
        user_id=get_active_user()
        sts=get_model("clinic.staff").search_browse([['user_id','=',user_id]])
        dpt_id=None
        if sts:
            dpt_id=sts[0].department_id.id
        return dpt_id

    _defaults={
        'date': lambda *a: time.strftime("%Y-%m-%d"),
        'date_from': lambda *a: time.strftime("%Y-%m-%d"),        
        'date_to': lambda *a: (datetime.now()+timedelta(days=DRT)).strftime("%Y-%m-%d"),        
        'department_id': _get_deparment,
        'branch_id': _get_branch,
    }
    
    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
        department_id=None
        branch_id=None

        user_id=get_active_user()
        sts=get_model("clinic.staff").search_browse([['user_id','=',user_id]])
        if sts:
            department_id=sts[0].department_id.id
            branch_id=sts[0].branch_id.id
        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

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

        lines=[]
        empty_line={
            'no': '',
            'number': '',
            'visit_id': None,
            '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'
                    hd_case_number+=" (เสร็จสิ้น)"
                elif hd_case_state=='in_progress':
                    hd_case_number+=' (กำลังรักษา)'

            number=visit.number
            if number=='/':
                number='*Waiting'
            cycle=visit.cycle_id
            patient=visit.patient_id
            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_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,
                'hd_case_state':HD_STATE.get(hd_case_state,''),
            }
            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()