import time

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

class HDCaseExpense(Model):
    _name="clinic.hd.case.expense"
    _string="Expense"
    _name_field="date"
    _multi_company=True
    
    def _get_patient_conflict(self,ids,context={}):
        res={}
        for obj in self.browse(ids):
            hd_case=obj.hd_case_id
            patient=hd_case.patient_id
            conflict=False
            if obj.patient_id.id!=patient.id:
                conflict=True
            res[obj.id]=conflict
        return res

    _fields={
        'date': fields.Date("Date",required=True,search=True),
        'patient_id': fields.Many2One("clinic.patient","Patient",required=True,search=True),
        'hd_case_id': fields.Many2One("clinic.hd.case","HD Case",required=True,search=True),
        'payment_id': fields.Many2One("account.payment","Payment",search=True),
        "invoices": fields.One2Many("account.invoice","clinic_expense_id","Invoices"),
        'fee_amt': fields.Float("Fee"),
        'mdc_amt': fields.Float("Medicine"), # EPOadm
        'srv_amt': fields.Float("Service"), # Allow
        'fee': fields.Boolean("Fee"),
        'medicine': fields.Boolean("Medicine"),
        'service': fields.Boolean("Service"),
        'state': fields.Selection([['draft','Draft'],['waiting_matching','Waiting Matching'],['match','Match'],['unmatch','Unmatch'],['approved','Approved']],'State'),
        'note': fields.Text("Note"),
        'pt_conflict': fields.Boolean("Patient Conclict",function="_get_patient_conflict"),
        'company_id': fields.Many2One("company","Company"),
        'match_id': fields.Many2One("clinic.report.payment.matching","Match"),
    }

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

    def create(self,vals,**kw):
        new_id=super().create(vals,**kw)
        hd_case_id=vals['hd_case_id']
        hd_case=get_model("clinic.hd.case").browse(hd_case_id)
        for inv in hd_case.invoices:
            inv.write({
                'clinic_expense_id': new_id,
            })
        return new_id

    def approve(self,ids,context={}):
        obj=self.browse(ids)[0]
        obj.write({
            'state': 'approved',
        })

    def to_draft(self,ids,context={}):
        obj=self.browse(ids)[0]
        obj.write({
            'state': 'draft',
        })

    def delete(self,ids,context={}):
        for obj in self.browse(ids):
            if obj.state!='draft':
                continue #XXX migration
                raise Exception("Status is not draft!")
        super().delete(ids,context)
    
    def do_match(self,ids,context={}):
        obj=self.browse(ids)[0]
        obj.write({
            'state': 'match',
        })

    def do_unmatch(self,ids,context={}):
        obj=self.browse(ids)[0]
        obj.write({
            'state': 'unmatch',
        })

    def cancel(self,ids,context={}):
        obj=self.browse(ids)[0]
        obj.write({
            'state': 'cancelled',
        })

HDCaseExpense.register()