clinic/netforce_clinic/models/cycle_item.py

import time

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

class CycleItem(Model):
    _name="clinic.cycle.item"
    _string="Cycle Item"
    _name_field="name"
    
    def _get_all(self,ids,context={}):
        res={}
        for obj in self.browse(ids):
            cycle=obj.cycle_id
            cycle_name=''
            if cycle:
                cycle_name=cycle.name
            name="%s-%s"%(cycle_name,obj.date)
            res[obj.id]={
                'name': name,
            }
        return res
    
    _fields={
        'name': fields.Char("Name", function="_get_all",function_multi=True),
        'date': fields.Date("Date",search=True),
        'cycle_id': fields.Many2One("clinic.cycle", "Cycle",search=True),
        'cycle_daily_id': fields.Many2One("clinic.cycle.daily", "Cycle Daily",search=True),
        'lines': fields.One2Many("clinic.cycle.item.line","item_id", "Lines"),
        'visits': fields.One2Many("clinic.visit","cycle_item_id", "Visits"),
        'hd_cases': fields.One2Many("clinic.hd.case","cycle_item_id", "HD Cases"),
        'nurses': fields.One2Many("clinic.cycle.item.nurse",'cycle_item_id','Nurses'),
        'nurse_id': fields.Many2One("clinic.staff","Nurse",domain=[['type','=','nurse']]), #XXX
        'sequence': fields.Char("Sequence"), # for sort item
        'company_id': fields.Many2One("company", "Company"),
        "state": fields.Selection([("draft","Draft"),("validated","Validated")],"Status",required=True),
        'user_id': fields.Many2One("base.user","Validator"),
    }
    
    def _get_vark(self,context={}):
        st=get_model("clinic.setting").browse(1)
        return st.var_k or 0

    _defaults={
        'state': 'draft',
        'company_id': lambda *a: get_active_company(),
        'date': lambda *a: time.strftime("%Y-%m-%d"),
        'user_id': lambda *a: get_active_user(),
        'var_k': _get_vark,
    }

    _order="date desc"
    _sql_constraints=[
        ("cycle_item_uniq","unique (cycle_id,date,company_id)","Cycle item should be unique"),
    ]
    
    def get_cycle_daily(self,date):
        dom=[]
        print("get ", date)
        dom.append(['date','=',date])
        cd_ids=get_model('clinic.cycle.daily').search(dom)
        cd_id=None
        if cd_ids:
            cd_id=cd_ids[0]
        else:
            cd_id=get_model('clinic.cycle.daily').create({
                'date':date,
                'name':date,
                })
        return cd_id
        

    def create(self, vals,**kw):
        date=vals['date']
        cycle_id=vals['cycle_id']
        cycle=get_model("clinic.cycle").browse(cycle_id)
        vals['sequence']='%s-%s'%(date,cycle.sequence) #date-sequence
        obj_id=super().create(vals,**kw)
        return obj_id

    def write(self,ids,vals,**kw):
        obj=self.browse(ids)[0]
        cycle=obj.cycle_id
        vals['sequence']='%s-%s'%(obj.date,cycle.sequence) #date-sequence
        super().write(ids,vals,**kw)
    
    def validate(self,ids,context={}):
        obj=self.browse(ids)[0]
        for obj_id in ids:
            lcost_ids=get_model("clinic.labor.cost").search([['cycle_item_id','=',obj_id]])
            labor_cost=get_model("clinic.labor.cost")
            if not lcost_ids:
                st=get_model("clinic.setting").browse(1)
                lc_id=labor_cost.create({
                    'cycle_item_id': obj_id,
                    'var_k': st.var_k,
                })
                lc=labor_cost.browse(lc_id)
                lc.compute()
            else:
                lc=labor_cost.search_browse([['cycle_item_id','=',obj.id]])
                if lc:
                    lc.compute()
        obj.write({
            'state': 'validate',
        })

        return

        levels={}
        for line in obj.lines:
            level=line.level_id
            levels[level.id]={
                'amount': line.amount or 0,
                'qty': line.qty,
                'level_id': level.id,
            }

        lines=[]
        # cost's nurses
        for nr in obj.nurses:
            nurse=nr.nurse_id
            level=nr.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': obj.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 obj.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,
                        'level_id': staff.level_id.id,
                        'type': ps.type,
                        'qty': 0,
                     }
                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': obj.cycle_id.id,
                'staff_id': doctor_id,
                'level_id': level_id,
                'rate': base,
                'qty': qty,
                'type': type,
            }))
        
        cycle_daily=obj.cycle_daily_id
        # clear cost line
        for line in cycle_daily.lines:
            cycle=line.cycle_id
            if obj.cycle_id.id==cycle.id: # only own cycle
                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']
            #amount=vals['amount'] or 0
            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,
                    'type': vals['type'],
                    'level_id': vals['level_id'],
                    'rate': rate,
                    'qty': qty,
                }
                continue
            glines[key]['amount']+=amount
            glines[key]['qty']+=qty

        lines=[]
        for key,vals in glines.items():
            cycle_id,staff_id=key 
            line={
                'cycle_id': cycle_id,
                'staff_id': staff_id,
                'date': obj.date,
            }
            line.update(vals)
            lines.append(('create',line))

        # call from outside 
        if context.get('called'):
            return lines

        cycle_daily.write({
            'lines': lines,
        })

        obj.write({
            'state': 'done',
        })

        return {
            'next': {
                'name': 'clinic_cycle_daily',
                'mode': 'form',
                'active_id': cycle_daily.id,
            },
            'flash': 'Cycle Item has been validated, please see the detail of cost below.',
        }

    def to_draft(self,ids,context={}):
        obj=self.browse(ids)[0]
        obj.write({
            'state': 'draft',
        })
    
    def onchange_nurse(self,context={}):
        data=context["data"]
        path=context["path"]
        line=get_data_path(data,path,parent=True)
        nurse_id=line['nurse_id']
        nurse=get_model('clinic.staff').browse(nurse_id)
        line['level_id']=nurse.level_id.id
        return data
    
    def view_schedule(self,ids,context={}):
        obj=self.browse(ids)[0]
        date=obj.date
        schd_ids=get_model('clinic.schedule').search([['date','=',date]])
        schedule_id=None
        if schd_ids:
            schedule_id=schd_ids[0]
        return {
            'next': {
                'name': 'clinic_schedule',
                'mode': 'form',
                'active_id': schedule_id,
            }
        }

    def view_cycle_daily(self,ids,context={}):
        obj=self.browse(ids)[0]
        if not obj.cycle_daily_id:
            raise Exception("Please validate cycle item")
        return {
            'next': {
                'name': 'clinic_cycle_daily',
                'mode': 'form',
                'active_id': obj.cycle_daily_id.id,
            }
        }
    
    def load_nurse_from_schedule(self,ids,context={}):
        obj=self.browse(ids)[0]
        #TODO check cycle number & date from schedule
        schedules=get_model("clinic.schedule").search_browse([['date','=',obj.date]])
        nurses=[]
        for schedule in schedules:
            for line in schedule.lines:
                cycle=line.cycle_id
                if obj.cycle_id.id==cycle.id:
                    nurse=line.nurse_id
                    level=line.level_id
                    nurses.append(('create',{
                        'nurse_id': nurse.id,
                        'level_id': level.id,
                    }))
        for nurse in obj.nurses:
            nurse.delete()
        obj.write({
            'nurses': nurses,
        })

        return {
            'next': {
                'name': 'clinic_cycle_item',
                'mode': 'form',
                'active_id': obj.id,
            },
            'flash': 'Load nurse from schedule to cycle item successfully',
        }

CycleItem.register()