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

class LaborCost(Model):
    _name="clinic.labor.cost"
    _string="Labor Cost"
    _multi_company=True
    _name_field="cycle_item_id"
    _key=['cycle_item_id']

    def _get_all(self,ids,context={}):
        res={}
        for obj in self.browse(ids):
            hd_cases=obj.cycle_item_id.hd_cases or []
            total_case=len(hd_cases)
            total_a,total_b=0,0
            total=0.0
            for fline in obj.formulars:
                total+=fline.amount or 0
                total_a+=fline.var_a or 0
                total_b+=fline.var_b or 0
            var_ptx=total_case*(obj.var_k or 0)
            var_fml1='PTxK=%sX%s'%(total_a,total_b)
            total_bstr=total_b < 0 and "+%s"%(abs(total_b)) or "-%s"%total_b
            var_x=0.0
            if total_a:
                var_x=eval('(%s%s)/%s'%(var_ptx,total_bstr,total_a))

            total_ncost=0.0
            for line in obj.nurse_lines:
                total_ncost+=line.amount or 0.0

            total_dcost=0.0
            for line in obj.doctor_lines:
                total_dcost+=line.amount or 0.0

            total_stcost=0.0
            for line in obj.staff_lines:
                total_stcost+=line.amount or 0.0

            res[obj.id]={
                'var_pt': total_case,
                'var_ptx': total_case*(obj.var_k or 0),
                'var_fml1': '%s'%(var_fml1),
                'var_fml2': '%s=%sX%s'%(var_ptx,total_a,total_b),
                'var_fml3': '(%s%s)/%s'%(var_ptx,total_bstr,total_a),
                'var_x': round(var_x,2),
                'total': total,
                'total_ncost': total_ncost,
                'total_dcost': total_dcost,
                'total_stcost': total_stcost,
            }

        return res

    _fields={
        "cycle_item_id": fields.Many2One("clinic.cycle.item","Cycle Item",required=True),
        'var_k': fields.Float("K"),
        'var_pt': fields.Integer("PT", function="_get_all",function_multi=True),
        'var_ptx': fields.Char("PT x K:", function="_get_all",function_multi=True),
        'var_fml1': fields.Char("Step 1:",function="_get_all",function_multi=True),
        'var_fml2': fields.Char("Step 2:",function="_get_all",function_multi=True),
        'var_fml3': fields.Char("X:",function="_get_all",function_multi=True),
        'var_x': fields.Char("X:",function="_get_all",function_multi=True),
        'total': fields.Float("Total",function="_get_all",function_multi=True),
        'total_ncost': fields.Float("Total",function="_get_all",function_multi=True),
        'total_dcost': fields.Float("Total",function="_get_all",function_multi=True),
        'total_stcost': fields.Float("Total",function="_get_all",function_multi=True),
        'manual': fields.Boolean("Manual"),
        "formulars": fields.One2Many("clinic.labor.cost.formular", "labor_cost_id", "Formulars"),
        "nurse_lines": fields.One2Many("clinic.labor.cost.line", "labor_cost_id", "Nurse Lines",domain=[['type','=','nurse']]),
        "doctor_lines": fields.One2Many("clinic.labor.cost.line", "labor_cost_id", "Doctor Lines",domain=[['type','=','doctor']]),
        "staff_lines": fields.One2Many("clinic.labor.cost.line", "labor_cost_id", "staff Lines",domain=[['type','=','staff']]),
        'company_id': fields.Many2One("company","Company"),
    }

    _defaults={
        'company_id': lambda *a: get_active_company(),
    }


    def compute(self,ids,context={}):
        obj=self.browse(ids)[0]
        if not obj.manual:
            for fline in obj.formulars:
                fline.delete()

        levels={}
        for level_id in get_model('clinic.staff.level').search([['type','=','nurse']]):
            vals={
            level_id: {
                    'total': 0,
                }
            }
            levels.update(vals)
        item=obj.cycle_item_id

        for line in item.lines:
            level=line.level_id
            nurse=line.nurse_id
            if not level:
                raise Exception("Please specify level for %s"%nurse.name)
            levels[level.id]['total']+=1

        st_levels={}
        st=get_model("clinic.setting").browse(1)
        for line in st.levels:
            level=line.level_id
            st_levels[level.id]={
                'var_a': line.var_a or 0,
                'var_b': line.var_b or 0,
                'op': line.op or "",
                'formular': line.formular or "",
            }
        formulars=[]
        total_a,total_b=0,0
        for level_id, value in levels.items():
            qty=value['total']
            st_level=st_levels.get(level_id,None)
            a,b=0,0
            state="fail"
            if st_level:
                state="success"
                var_a=float(st_level['var_a'] or "0")
                var_b=float(st_level['var_b'] or "0")
                fml_org=st_level['formular']
                a=qty*var_a
                a_str='%sX'%(a)
                total_a+=a

                op=st_level['op']
                sign=op=='-' and -1 or 1
                op_str=sign==1 and op or ""

                b=(qty*var_b)*sign
                b_str=(str(b) if b else "")
                total_b+=b
            else:
                # XXX prevent to show all level if not configuration from setting
                continue

            fml=''.join([a_str, op_str, b_str])
            formulars.append(('create',{
                'level_id': level_id,
                'var_a': a,
                'var_b': b,
                'qty': qty,
                'formular_org': fml_org,
                'formular': fml if qty else "0",
                'state': state,
            }))

        if not obj.manual:
            var_pt=len(item.hd_cases)
            var_ptx=var_pt*(obj.var_k or 0)
            total_bstr=total_b < 0 and "+%s"%(abs(total_b)) or "-%s"%total_b
            var_x=0
            if total_a:
                var_x=eval('(%s%s)/%s'%(var_ptx,total_bstr,total_a))
            for fline in formulars:
                vals=fline[1]
                fml=vals['formular'].replace("X","*%s")
                vals['amount']=0
                if vals['var_a'] or vals['var_b']:
                    vals['amount']=eval(fml%var_x)
        else:
            # update qty (a)
            formulars=[]
            for fline in obj.formulars:
                level=fline.level_id
                qty=fline.qty or 0
                st_level=st_levels.get(level.id,None)
                a,b=0,0
                state="fail"
                if st_level:
                    state="success"
                    var_a=float(st_level['var_a'] or "0")
                    var_b=float(st_level['var_b'] or "0")
                    fml_org=st_level['formular']
                    a=qty*var_a
                    a_str='%sX'%(a)

                    op=st_level['op']
                    sign=op=='-' and -1 or 1
                    op_str=sign==1 and op or ""

                    b=(qty*var_b)*sign
                    b_str=(str(b) if b else "")
                else:
                    print("not found in setting ", level.id)

                fml=''.join([a_str, op_str, b_str])
                formulars.append(('write',[fline.id],{
                    'level_id': level.id,
                    'var_a': a,
                    'var_b': b,
                    'qty': qty,
                    'formular': fml if qty else "0",
                    'formular_org': fml_org,
                    'state': state,
                }))

        obj.write({
            'formulars': formulars,
        })

        levels={}
        if obj.manual:
            for fline in obj.formulars:
                fml=(fline.formular or "").replace("X","*%s")
                if fline.var_a or fline.var_b:
                    amt=eval(fml%obj.var_x)
                else:
                    amt=0
                fline.write({
                    'amount': amt,
                })

                level=fline.level_id
                levels[level.id]={
                    'amount': amt or 0.0,
                    'qty': fline.qty or 0,
                    'level_id': level.id,
                }
        else:
            for fline in formulars:
                vals=fline[1]
                level_id=vals['level_id']
                levels[level_id]={
                    'amount': vals['amount'] or 0.0,
                    'qty': vals['qty'] or 0,
                    'level_id': level_id,
                }

        #TODO UPDATE COST :  like cycle daily
        item=obj.cycle_item_id
        lines=[]
        # cost's nurses
        for line in item.lines:
            nurse=line.nurse_id
            level=line.level_id
            vals=levels.get(level.id)
            rate,amt,qty=0.0,0.0,0
            level_id=level.id
            if vals:
                level_id=vals['level_id']
                amt=vals['amount']
                qty=vals['qty']
                if qty:
                    rate=amt/qty
            lines.append(('create',{
                'cycle_id': item.cycle_id.id,
                'staff_id': nurse.id,
                'level_id': level_id,
                'rate': rate,
                'type': 'nurse',
                'qty': 1,
            }))

        # cost's doctor
        st=get_model('clinic.setting').browse(1)
        cost_per_case=st.cost_per_case or 0
        staff_total={}
        for hd_case in item.hd_cases:
            staffs=hd_case.staffs
            for ps in staffs:
                staff=ps.staff_id
                if not staff:
                    continue
                base=staff.base
                if not base:
                    base=cost_per_case
                if not staff_total.get(staff.id):
                    staff_total[staff.id]={
                        'base': 0,
                        'type': staff.type,
                        'level_id': staff.level_id.id,
                        'qty': 0,
                     }
                if staff.type!='doctor':
                    base=0
                    staff_total[staff.id]['description']=hd_case.number
                staff_total[staff.id]['base']=base
                staff_total[staff.id]['qty']+=1

        for doctor_id, value in staff_total.items():
            base=value['base']
            type=value['type']
            qty=value['qty']
            level_id=value['level_id']
            lines.append(('create',{
                'cycle_id': item.cycle_id.id,
                'staff_id': doctor_id,
                'level_id': level_id,
                'rate': base,
                'qty': qty,
                'type': type,
                'description': value.get('description',"")
            }))
        
        # clear cost line
        for line in obj.nurse_lines:
            line.delete()
        for line in obj.doctor_lines:
            line.delete()

        # group staff and cycle date
        glines={}
        for line in lines:
            mode,vals=line
            cycle_id=vals['cycle_id']
            staff_id=vals['staff_id']
            rate=vals['rate'] or 0
            qty=vals['qty'] or 0
            amount=qty*rate
            key=(cycle_id,staff_id)
            if not key in glines.keys():
                glines[key]={
                    'amount': amount,
                    'level_id': vals['level_id'],
                    'rate': rate,
                    'qty': qty,
                    'type': vals['type'],
                    'description': vals.get("description",""),
                }
                continue
            glines[key]['amount']+=amount
            glines[key]['qty']+=qty

        nurse_lines=[]
        doctor_lines=[]
        for key,vals in glines.items():
            cycle_id,staff_id=key 
            line={
                'cycle_id': cycle_id,
                'staff_id': staff_id,
                'date': item.date,
            }
            line.update(vals)
            st_type=vals['type']
            if st_type=='doctor':
                doctor_lines.append(('create',line))
            else:
                nurse_lines.append(('create',line))

        obj.write({
            'doctor_lines': doctor_lines,
            'nurse_lines': nurse_lines,
        })

        return {
            'next': {
                'name': 'clinic_labor_cost',
                'mode': 'form',
                'active_id': obj.id,
            },
            'flash':'Compute successfully',
        }

    def onchange_cost_line(self,context={}):
        data=context['data']
        path=context['path']
        line=get_data_path(data,path,parent=True)
        qty=line.get("qty",0)
        rate=line.get("rate",0.0)
        amt=qty*rate
        line['amount']=amt
        data=self.update_amt(context)
        return data

    def update_amt(self,context):
        total=0.0
        data=context['data']
        for line in data['lines']:
            qty=line['qty'] or 0
            rate=line['rate'] or 0.0
            amt=qty*rate
            total+=amt
        data['total_cost']=total
        return data

LaborCost.register()