import time from datetime import datetime, timedelta from netforce.model import Model, fields, get_model from netforce.utils import get_data_path from netforce.access import get_active_user,set_active_user from netforce.access import get_active_company class HDcase(Model): _name="clinic.hd.case" _string="Treatment" _audit_log=True _name_field="number" _multi_company=True def get_hrs(self,ids,context={}): res={} fmt="%Y-%m-%d %H:%M:%S" for obj in self.browse(ids): diff=datetime.strptime(obj.time_stop,fmt)-datetime.strptime(obj.time_start,fmt) total_time=round(diff.seconds/3600,2) res[obj.id]=total_time return res _fields={ "number": fields.Char("Number",required=True,search=True), "patient_id": fields.Many2One("clinic.patient","Patient",required=True,search=True), "doctor_id": fields.Many2One("clinic.doctor","Doctor", required=True,search=True), "nurse_id": fields.Many2One("clinic.nurse","Nurse", required=True,search=True), "date_start": fields.DateTime("Time start",required=True,search=True), "date_stop": fields.DateTime("Time stop",required=True,search=True), "department_id": fields.Many2One("clinic.department", "Department",search=True), "wh_start": fields.Float("Wt.Kg start"), "wh_stop": fields.Float("Wt.Kg stop"), "bp_start": fields.Integer("BP mmHG start"), "per_bp_start": fields.Integer("/Per start"), "bp_stop": fields.Integer("BP mmHG stop"), "per_bp_stop": fields.Integer("/Per stop"), "epo_tn" : fields.Char("EpoTn (Drug name)"), "epo_unit" : fields.Integer("EpoUnit (Unit of Used drug)"), "hct": fields.Integer("HCT %", required=True), "check_goverment_pay" : fields.Boolean("The Government Pay"), "check_personal_pay" : fields.Boolean("Pay yourself"), "state": fields.Selection([("draft","Draft"),("confirmed","Confirmed"),("approved","Approved"),("cancelled","Cancelled"),("paid","Paid")],"Status",required=True), "hct": fields.Integer("HCT %"), "state": fields.Selection([("draft","Draft"),("in_progress","In Progress"),("completed","Completed"),("discountinued","Discountinued"),("uncompleted","Uncompleted")],"Status",required=True), "dialyzers": fields.One2Many("clinic.dialyzer.line","hd_case_id","Dializers"), "lines": fields.One2Many("clinic.hd.case.line","hd_case_id","Lines"), "comments": fields.One2Many("message","related_id","Comments"), "company_id": fields.Many2One("company","Company"), "fee": fields.Float("HD Fee"), "amount": fields.Float("Due Amount",function="get_total",readonly=True,function_multi=True), "total": fields.Float("Total",function="get_total",readonly=True,function_multi=True), "reconcile_id": fields.Many2One("account.reconcile","Reconcile Id",readonly=True), "invoices": fields.One2Many("account.invoice","related_id","Invoices"), "pickings": fields.One2Many("stock.picking","related_id","Pickings"), "payments": fields.One2Many("account.payment","related_id","Payments"), 'visit_id': fields.Many2One("clinic.visit", "Visit"), 'total_time': fields.Integer("Total Time(Hrs)",function="get_hrs"), "fee_type": fields.Selection([("mg","Medical Government"),("sc","Social Security"),("nhso","NHSO (30฿)"),("personal","Personal"),("others","Others")],"Fee Type"), 'fee_partner_id': fields.Many2One("partner","Contact Fee"), 'paid': fields.Boolean("Paid"), "time_start": fields.DateTime("Time start",required=True,search=True), "time_stop": fields.DateTime("Time stop",required=True,search=True), "date": fields.Date("Date Treatment",required=True), 'planes': fields.One2Many("clinic.visit.plane","hd_case_id","Planning"), 'note': fields.Text("Note"), } def _get_number(self,context={}): while 1: seq_id=get_model("sequence").find_sequence(name="Clinic HD Case") num=get_model("sequence").get_next_number(seq_id,context=context) if not num: return None user_id=get_active_user() set_active_user(1) res=self.search([["number","=",num]]) set_active_user(user_id) if not res: return num get_model("sequence").increment_number(seq_id,context=context) def _get_nurse(self,context={}): user_id=get_active_user() print("user_id ",user_id) nurse_ids=get_model("clinic.nurse").search([['user_id','=',user_id]]) if nurse_ids: return nurse_ids[0] return None _defaults={ "state": "draft", "date": lambda *a: time.strftime("%Y-%m-%d"), "time_start": lambda *a: time.strftime("%Y-%m-%d %H:%M:%S"), "time_stop": lambda *a: time.strftime("%Y-%m-%d %H:%M:%S"), 'nurse_id': _get_nurse, "number": _get_number, "company_id": lambda *a: get_active_company(), "fee": 1500, } _order="date desc,number desc" def onchange_dialyzer(self,context={}): data=context["data"] path=context["path"] line=get_data_path(data,path,parent=True) dialyzer_id=line.get("dialyzer_id") if not dialyzer_id: return {} dialyzer=get_model("clinic.dialyzer").browse(dialyzer_id) line["description"]=dialyzer.description or "" line["use_time"]=dialyzer.use_time line["max_use_time"]=dialyzer.max_use_time line["member_type"]=dialyzer.member_type line["dialyzer_type"]=dialyzer.dialyzer_type line["bid_flow_rate"]=dialyzer.bid_flow_rate line["ultrafittration"]=dialyzer.ultrafittration line["state"]=dialyzer.state return data def onchange_patient(self,context={}): data=context['data'] patient_id=data['patient_id'] hd_cases=self.search_browse([['patient_id','=',patient_id]]) # TODO reset dialyzer if hd_cases: hd_case=hd_cases[-1] data['doctor_id']=hd_case.doctor_id.id data['department_id']=hd_case.department_id.id else: data['doctor_id']=None data['department_id']=None return data def onchange_product(self,context={}): data=context['data'] path=context["path"] line=get_data_path(data,path,parent=True) product_id=line.get('product_id') if not product_id: return prod=get_model("product").browse(product_id) line['uom_id']=prod.uom_id.id line['description']=prod.name line['qty']=1 line['price']=prod.sale_price or 0.0 line['amount']=prod.sale_price or 0.0 self.onchange_line(context) return data def onchange_line(self,context={}): data=context['data'] total=0.0 for line in data['lines']: price=line.get('price') or 0 qty=line.get('qty') or 0 amt=qty * price product_id=line.get("product_id") if product_id: line['amount']=amt total+=amt data['total']=total return data def onchange_htc(self,context={}): data=context['htc'] total=0.0 if (htc > 39): data['htc']=total return data def cancelled(self,ids,context={}): obj=self.browse(ids)[0] obj.write({"state":"cancelled"}) def make_invoices(self,ids,context={}): setting=get_model("settings").browse(1) currency_id=setting.currency_id.id # account_receivable_id=setting.account_receivable_id.id company_id=get_active_company() uom=get_model("uom").search_browse([['name','ilike','%Unit%']]) if not uom: raise Exception("Unit not found in uom") obj=self.browse(ids[0]) due_date=obj.date[0:10] # XXX context['type']='out' context['inv_type']='invoice' if obj.lines and obj.total: # XXX need to split function prepare invoice vals={ "type": "out", "inv_type": "invoice", "tax_type": "tax_in", 'due_date': due_date, "ref": obj.number, "related_id": "clinic.hd.case,%s"%obj.id, "currency_id": currency_id, "company_id": company_id, "lines": [], "company_id": company_id, } partner=obj.patient_id.partner_id if not partner: raise Exception("No contact for patient %s"%obj.patient_id.name) vals["partner_id"]=partner.id for line in obj.lines: account_id=line.product_id.sale_account_id.id or account_receivable_id if not account_id: raise Exception("%s not found account recievable %s"%partner.name) line={ "product_id": line.product_id.id, "description": line.description, "qty": line.qty, "uom_id": line.uom_id.id, "unit_price": line.price, "amount": line.amount, 'account_id': account_id, } vals['lines'].append(('create',line)) inv_id=get_model("account.invoice").create(vals,context) # create picking obj.make_pickings() if obj.fee: vals={ "type": "out", "inv_type": "invoice", "tax_type": "tax_in", 'due_date': due_date, "ref": obj.number, "related_id": "clinic.hd.case,%s"%obj.id, "currency_id": currency_id, "company_id": company_id, "lines": [], "company_id": company_id, } vals["partner_id"]=obj.fee_partner_id.id # XXX #if obj.fee_partner_id.account_payable_id: #account_receivable_id=obj.fee_partner_id.account_payable_id.id if obj.fee_partner_id.account_receivable_id: account_id=obj.fee_partner_id.account_receivable_id.id or account_receivable_id if not account_receivable_id: raise Exception("%s not found account recievable %s"%obj.fee_partner_id.name) line={ "product_id": None, "description": "Fee", "qty": 1, "uom_id": uom[0].id, "unit_price": obj.fee, "amount": obj.fee, 'account_id': account_id, } vals['lines'].append(('create',line)) inv_id=get_model("account.invoice").create(vals,context) def make_pickings(self,ids,context={}): obj=self.browse(ids[0]) # no picking if not obj.lines: return partner=obj.patient_id.partner_id if not partner: raise Exception("Contact not for this patient") ship_address_id=None for address in partner.addresses: if address.type=="shipping": ship_address_id=address.id break if not ship_address_id: raise Exception("contact %s dont'have address with type shipping"%partner.name) pick_vals={ "type": "out", "ref": obj.number, "related_id": "clinic.hd.case,%s"%obj.id, "partner_id": obj.patient_id.partner_id.id, "ship_address_id": ship_address_id, "state": "draft", "lines": [], } res=get_model("stock.location").search([["type","=","customer"]]) if not res: raise Exception("Customer location not found") cust_loc_id=res[0] for line in obj.lines: prod=line.product_id if prod.type != 'stock': continue wh_loc_id=prod.location_id.id if not wh_loc_id: res=get_model("stock.location").search([["type","=","internal"]]) if not res: raise Exception("Warehouse not found") wh_loc_id=res[0] line_vals={ "product_id": prod.id, "qty": 1, "uom_id": prod.uom_id.id, "location_from_id": wh_loc_id, "location_to_id": cust_loc_id, } pick_vals["lines"].append(("create",line_vals)) if not pick_vals["lines"]: return { "flash": "Nothing left to deliver", } picking_obj=get_model("stock.picking") pick_id=picking_obj.create(pick_vals,context={"pick_type": "out"}) pick=picking_obj.browse(pick_id) pick.set_done([pick_id]) def post_invoices(self,ids,context={}): obj=self.browse(ids[0]) for inv in obj.invoices: inv.post() print("Post!") def confirm(self,ids,context={}): obj=self.browse(ids)[0] obj.write({"state":"in_progress"}) def discontinue(self,ids,context={}): obj=self.browse(ids)[0] # TODO pop to note obj.write({"state":"uncompleted"}) def complete(self,ids,context={}): obj=self.browse(ids)[0] obj.make_invoices() obj.post_invoices() obj.write({ "state":"completed", #'paid': True, }) return { 'next': { 'name': 'clinic_hd_case', 'mode': 'form', 'active_id': obj.id, }, 'flash': '%s is completed'%obj.number, } def journal_report(self,ids,context={}): obj=self.browse(ids[0]) move_id = get_model("account.move").search([["narration","=",obj.number]]) if not move_id: raise Exception("Order is not post or nove have jounal entry.") return { "next": { "name": "journal_entry", "mode":"form", "active_id":move_id[0], } } def get_total(self,ids,context={}): vals={} for obj in self.browse(ids): total=0 amt=0 fee=obj.fee or 0 for line in obj.lines: total+=line.amount or 0 fee=0 # XXX amt=total+fee vals[obj.id]={ "total": total, "amount": amt, } return vals def view_hdcase(self,ids,context={}): return { 'name': 'clinic_hd_case', 'mode': 'form', 'form_view_xml': 'clinic_hd_case_form', 'active_id': ids[0], } def delete(self,ids,context={}): for obj in self.browse(ids): if obj.state != 'draft': raise Exception("Can not delete HD Case %s because state is not draft"%obj.number) super().delete(ids) def onchange_fee_type(self,context={}): data=context['data'] fee_type=data.get("fee_type","") if fee_type in ("mg","sc","nhso"): data['fee']=1500 else: data['fee']=0.0 return data def new_visit(self,ids,context={}): obj=self.browse(ids[0]) skip_plan_id=0 plane=None for plane in obj.planes: if plane.state=='close': continue break # XXX if not plane or plane.state=='close': raise Exception("No Planing") print("found plane ",plane) vals={ 'date_visit': plane.date, 'time_start': plane.time_start, 'time_stop': plane.time_stop, 'patient_id': obj.patient_id.id, 'doctor_id': obj.doctor_id.id, 'nurse_id': obj.nurse_id.id, 'department_id': obj.department_id.id, 'planes': [], 'dialyzers': [], #'state': 'waiting_treatment', 'state': 'draft', # XXX need to gen new doc } plane.write({ 'state': 'close', }) visit_id=get_model("clinic.visit").create(vals) visit=get_model('clinic.visit').browse(visit_id) skip_plan_id=plane.id # copy remaining plane to new visit new_planes=[] for pl in obj.planes: # XXX if skip_plan_id==pl.id: continue if pl.state=='close': continue print("pl state ", pl.state) line={ 'date': pl.date, 'time_start': pl.time_start, 'time_stop': pl.time_stop, 'visit_id': visit_id, } new_planes.append(('create',line)) new_dialyzers=[] for dlz in obj.dialyzers: if dlz.use_time > dlz.dialyzer_id.max_use_time: continue use_time=(dlz.use_time or 0)+1 line={ "visit_id": visit_id, "dialyzer_id": dlz.dialyzer_id.id, "description": dlz.description, "use_time": use_time, "max_use_time": dlz.max_use_time, "member_type": dlz.member_type, "dialyzer_type": dlz.dialyzer_type, "bid_flow_rate": dlz.bid_flow_rate, "ultrafittration": dlz.ultrafittration, } new_dialyzers.append(('create',line)) vals={ 'planes': new_planes, 'dialyzers': new_dialyzers, } visit.write(vals) return { 'next': { 'name': "clinic_visit", 'mode': 'form', 'active_id': visit_id, }, 'flash': "New visit is created", } def onchange_dateplane(self,context={}): data=context["data"] path=context["path"] line=get_data_path(data,path,parent=True) start_date=line.get("date") if not start_date: return {} timenow=time.strftime("%H:%M:%S") timevisit=timenow # XXX timevisit=data['time_start'][11:20] fmt_date="%Y-%m-%d %H:%M:%S" fix_hrs=data['total_time'] or 0 # XXX time_start=datetime.strptime("%s %s"%(start_date,timevisit),fmt_date) seconds=fix_hrs*3600 time_stop=time_start+timedelta(seconds=seconds) line['time_start']=time_start.strftime(fmt_date) line['time_stop']=time_stop.strftime(fmt_date) if not line.get('state'): line['state']='open' return data HDcase.register()