import time import datetime from netforce.model import Model, fields, get_model from netforce.access import get_active_company, get_active_user, set_active_user from netforce.utils import get_data_path class Visit(Model): _name="clinic.visit" _string="Visit" _audit_log=True _name_field="number" _multi_company=True def _get_visit_late(self,ids,context={}): res={} #oneday=datetime.timedelta(days=1) #yesterday=datetime.datetime.now()-oneday datenow=time.strftime("%Y-%m-%d") for obj in self.browse(ids): late=0 if datenow > obj.date_visit and obj.state=='waiting_treatment': late=1 res[obj.id]=late 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",search=True), "nurse_id": fields.Many2One("clinic.nurse","Nurse",search=True), "department_id": fields.Many2One("clinic.department", "Department",search=True), "cycle": fields.Selection([("1","One"),("2","Two"),("3","Tree"),("4","Four")],"Cycle",required=True), "state": fields.Selection([("draft","Draft"),("waiting_treatment","Waiting Treatment"),('treatment','Treatment'),("cancelled","Cancelled")],"Status",required=True), "comments": fields.One2Many("message","related_id","Comments"), "company_id": fields.Many2One("company","Company"), 'time_use': fields.Integer("Fix Hour"), 'late_visit': fields.Integer("Late Vist",function="_get_visit_late"), "dialyzers": fields.One2Many("clinic.dialyzer.line","visit_id","Dialyzers"), "date_visit": fields.Date("Visit Date",required=True,search=True), "time_start": fields.DateTime("Time Start"), "time_stop": fields.DateTime("Time Stop"), 'planes': fields.One2Many("clinic.visit.plane","visit_id","Planning"), 'hd_cases': fields.One2Many('clinic.hd.case','visit_id',"HD Cases",readonly=True), } def _get_number(self,context={}): while 1: seq_id=get_model("sequence").find_sequence(name="Clinic Visit") 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 def _get_time_start(self,context={}): now=datetime.datetime.now() starttime=now.strftime("%Y-%m-%d %H:%M:%S") return starttime def _get_time_stop(self,context={}): hr=datetime.timedelta(seconds=3600) now=datetime.datetime.now() stoptime=(now+hr).strftime("%Y-%m-%d %H:%M:%S") return stoptime _defaults={ "state": "draft", "cycle": "1", 'date_visit': lambda *a: time.strftime("%Y-%m-%d"), 'time_start': _get_time_start, 'time_stop': _get_time_stop, "number": "/", "company_id": lambda *a: get_active_company(), 'nurse_id': _get_nurse, 'time_use': 1, } _order="id desc" def get_dialyzer(self): return def confirm(self,ids,context={}): obj=self.browse(ids[0]) number=obj.number if number=="/": number=self._get_number(context) vals={ 'number': number, 'state': 'waiting_treatment', } obj.write(vals) def do_treatment(self,ids,context={}): hd_case_obj=get_model("clinic.hd.case") dt=datetime.datetime datenow=dt.now().strftime("%Y-%m-%d") obj=self.browse(ids)[0] # cannot treatment if current date is not same visit date if obj.date_visit!=datenow: raise Exception("Today is not treament date!") if not obj.dialyzers: raise Exception("Please select dialyzer!") if len(obj.dialyzers)>1: raise Exception("Can select only 1 dialyzer!") include_fee=obj.patient_id.type in ('mg','sc','nhso') and True or False vals={ 'patient_id': obj.patient_id.id, 'doctor_id': obj.doctor_id.id, 'nurse_id': obj.nurse_id.id, 'department_id': obj.department_id.id, 'time_start': obj.time_start, 'time_stop': obj.time_stop, 'visit_id': obj.id, 'fee': include_fee and 1500.00 or 0.0, 'fee_type': obj.patient_id.type, 'lines':[], 'dialyzers': [], 'state': 'in_progress', } patient_type={ "mg":"Medical Government", "sc":"Social Security", "nhso":"NHSO (30฿)", "personal": "Personal", "others": "Others", } categ_name=patient_type.get(obj.patient_id.type) categ_ids=get_model("partner.categ").search([['name','=',categ_name]]) if not categ_ids: raise Exception("Category: %s not found"%categ_name) partner_id=None if obj.patient_id.type in ("mg","sc","nhso"): partner_obj=get_model("partner") for partner in partner_obj.search_browse([]): if partner.categ_id.id in categ_ids: partner_id=partner.id vals['fee_partner_id']=partner_id break # find dialyser if not obj.dialyzers: raise Exception("%s don't have dialyzer for treatment yet" % obj.patient_id.name) hd_case_id=hd_case_obj.create(vals) for dlz in obj.dialyzers: # TODO check max time and expire day also dlz.dialyzer_id.write({'use_time': dlz.use_time}) # count use time dlz.write({ 'hd_case_id': hd_case_id, }) for plan in obj.planes: plan.write({ 'hd_case_id': hd_case_id, }) obj.write({"state":"treatment"}) return { 'next': { 'name': 'clinic_hd_case', 'mode': 'form', 'active_id': hd_case_id, }, 'flash': 'Visit %s has been confirmed'%obj.number, } def discard(self,ids,context={}): obj=self.browse(ids)[0] obj.write({"state":"cancelled"}) def reopen(self,ids,context={}): obj=self.browse(ids)[0] obj.write({"state":"draft"}) def onchange_patient(self,context={}): data=context['data'] patient_id=data['patient_id'] visits=self.search_browse([['patient_id','=',patient_id]],order="number desc") if visits: #print([(v.id, v.date_visit,v.cycle) for v in visits]) visit=visits[0] data['doctor_id']=visit.doctor_id.id data['department_id']=visit.department_id.id cycle=int(visit.cycle)+1 data['cycle']=str(cycle) if cycle>4: data['cycle']='1' else: data['doctor_id']=None data['department_id']=None return data def copy(self,ids,context={}): obj=self.browse(ids[0]) vals={ 'patient_id': obj.patient_id.id, 'doctor_id': obj.doctor_id.id, 'nurse_id': obj.nurse_id.id, 'department_id': obj.department_id.id, } # XXX update cycle visits=self.search_browse([['patient_id','=',obj.patient_id.id]],order="number desc") if visits: visit=visits[0] # order desc already cycle=int(visit.cycle)+1 vals['cycle']=str(cycle) if cycle>4: vals['cycle']='1' new_id=self.create(vals,context=context) new_obj=self.browse(new_id) return { 'next':{ 'name': 'clinic_visit', 'mode': 'form', 'active_id': new_id, }, 'flash': 'Visit %s is copy to %s'%(obj.number,new_obj.number), } def get_visit_demo(self,context={}): if not context.get('number'): return {} visit_id=int(context['number']) visit=self.browse(visit_id) data={ 'number' : visit.number } return data def delete(self,ids,context={}): for obj in self.browse(ids): if obj.state!='draft': raise Exception("Can not delete visit %s because state is not draft!"%obj.number) super().delete(ids,context=context) def get_data(self,context={}): ref_id=context.get("refer_id") if not ref_id: return {} ref_id=int(ref_id) obj=self.browse(ref_id) cycle_list={'1': 'One','2': 'Two','3': 'Tree','4': 'Four'} data={ 'number': obj.number, 'date_visit': obj.date_visit, 'department_name': obj.department_id.name or "", 'patient_name': obj.patient_id.name or "", 'doctor_name': obj.doctor_id.name or "", 'nurse_name': obj.nurse_id.name or "", 'cycle': cycle_list.get(obj.cycle,""), 'print_date': time.strftime("%d/%m/%Y"), } return data 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) use_time=dialyzer.use_time or 0 use_time+=1 line["description"]=dialyzer.description or "" line["use_time"]=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_timeuse(self,context={}): data=context["data"] time_start=data['time_start'] fmt_date="%Y-%m-%d %H:%M:%S" time_start=datetime.datetime.strptime(time_start,fmt_date) time_use=data['time_use'] seconds=(time_use or 1)*3600 time_stop=time_start+datetime.timedelta(seconds=seconds) data['time_stop']=time_stop.strftime(fmt_date) return data def onchange_date_visit(self,context={}): data=context["data"] timenow=time.strftime("%H:%M:%S") date_visit=data['date_visit'] fmt_date="%Y-%m-%d %H:%M:%S" time_start=datetime.datetime.strptime("%s %s"%(date_visit,timenow),fmt_date) print("time_start ", time_start) time_use=data['time_use'] seconds=(time_use or 1)*3600 time_stop=time_start+datetime.timedelta(seconds=seconds) data['time_start']=time_start.strftime(fmt_date) data['time_stop']=time_stop.strftime(fmt_date) return data 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] if not timevisit: timevisit=timenow fmt_date="%Y-%m-%d %H:%M:%S" fix_hrs=data['time_use'] or 0 time_start=datetime.datetime.strptime("%s %s"%(start_date,timevisit),fmt_date) seconds=fix_hrs*3600 time_stop=time_start+datetime.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 def create_dialyzer(self,ids,context={}): return { 'next': { 'name': 'clinic_dialyzer', 'mode': 'form', }, } Visit.register()