import time from netforce.model import Model, fields, get_model from netforce.access import get_active_company, get_active_user, set_active_user class Staff(Model): _name="clinic.staff" _string="Staff" _audit_log=True _multi_company=True _key=["name","branch_id"] def _get_age(self,ids,context={}): res={} year_now=int(time.strftime("%Y")) for obj in self.browse(ids): age=0 if obj.birthday: year_bd=int(obj.birthday[0:4]) age=year_now-year_bd res[obj.id]=age return res def _get_level(self,ids,context={}): res={} for obj in self.browse(ids): level_id=None for rotate in obj.rotations: level_id=rotate.level_id.id res[obj.id]=level_id return res def _get_cycle(self,ids,context={}): res={} for obj in self.browse(ids): cycle_id=None for line in obj.cycles: cycle=line.cycle_id cycle_id=cycle.id res[obj.id]=cycle_id return res def _get_base(self,ids,context={}): res={} for obj in self.browse(ids): wage=0 max_cycle=0 for rotate in obj.rotations: wage=rotate.wage max_cycle=rotate.max_cycle res[obj.id]={ 'wage': wage, 'max_cycle': max_cycle, } return res def _get_hdcase(self,ids,context={}): # problem slow to load res={} for obj in self.browse(ids): hids=[] if obj.type=='doctor': hids=get_model("clinic.hd.case.staff").search([['staff_id','=',obj.id]]) elif obj.type=='nurse': item_ids=[item['cycle_item_id'][0] for item in get_model("clinic.cycle.item.line").search_read([['nurse_id','=',obj.id]],['cycle_item_id'])] for item in get_model("clinic.cycle.item").browse(item_ids): for hdcase in item.hd_cases: hids.append(hdcase.id) res[obj.id]=hids return res _fields={ 'employee_id': fields.Many2One("hr.employee","Employee"), "number": fields.Char("Number",required=True,search=True), "name": fields.Char("Name",required=True,search=True), "name_eng": fields.Char("Eng Name",search=True), "nick_name": fields.Char("Nick Name",search=True), "identification" : fields.Char("Identification Card"), "expiry_card" : fields.Date("Expired Card"), "birthday": fields.Date("Birthday",search=True), "age": fields.Integer("Age", function="_get_age"), "state": fields.Selection([["part_time","Part Time"],["full_time","Full Time"]],"Working Status",search=True), "type": fields.Selection([['staff','Staff'],["doctor","Doctor"],["nurse","Nurse"]],"Type"), "gender": fields.Selection([["male","Male"],["female","Female"]],"Gender"), "nation_id": fields.Many2One("clinic.nation","Nationality"), "mobile": fields.Char("Mobile",required=False,search=True), "phone": fields.Char("Phone",search=True), 'email': fields.Char("Email"), "prof_license" : fields.Char("License No."), "prof_license_date" : fields.Date("License Date"), "prof_license_expiry" : fields.Date("Expired License"), "birthday": fields.Date("BirthDay",search=True), "department_id": fields.Many2One("clinic.department", "Department",search=True), "patients": fields.One2Many("clinic.patient","doctor_id","Patients"), "addresses": fields.One2Many("address","staff_id","Addresses"), "comments": fields.One2Many("message","related_id","Comments"), "nurse_visits": fields.One2Many("clinic.visit","nurse_id","Visits"), "doctor_visits": fields.One2Many("clinic.visit","doctor_id","Visits"), "company_id": fields.Many2One("company","Company",search=True), "user_id": fields.Many2One("base.user","User",search=True), 'image': fields.File("Image"), 'note': fields.Text("Note"), 'categ_id': fields.Many2One("clinic.staff.categ", "Category"), 'level_id': fields.Many2One("clinic.staff.level", "Level", function="_get_level"), 'active': fields.Boolean("Active"), 'date': fields.Date("Register Date"), 'wage': fields.Float("Wage", function="_get_base",function_multi=True), 'max_cycle': fields.Integer("Max Cycle", function="_get_base",function_multi=True), 'check_max_cycle': fields.Boolean("Check Max Cycle"), 'hire_date': fields.Date("Hire Date"), 'resign_date': fields.Date("Resign Date"), "documents": fields.One2Many("document","related_id","Documents"), "rotations": fields.One2Many("clinic.staff.rotation","staff_id","Staff Rotation"), "cycles": fields.One2Many("clinic.staff.cycle","staff_id","Cycles"), 'cycle_id': fields.Many2One('clinic.cycle','Last Cycle',function="_get_cycle"), "hd_case_staffs": fields.One2Many("clinic.hd.case.staff","staff_id","HD Cases"), "hd_cases": fields.Many2Many("clinic.hd.case","HD Cases",function="_get_hdcase"), # not need to use (it's slow to load) "cycle_item_nurses": fields.One2Many("clinic.cycle.item.line","nurse_id","Cycle Items"), 'branch_id': fields.Many2One("clinic.branch","Branch", search=True), } def _get_number(self,context={}): while 1: seq_id=get_model("sequence").find_sequence(name="Clinic Staff") if not seq_id: return "/" 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_branch(self,context={}): b_ids=get_model('clinic.branch').search([]) if b_ids: return b_ids[0] _defaults={ 'active': True, "state": "part_time", "type": lambda self,ctx: ctx.get("type") or "staff", "date": lambda *a: time.strftime("%Y-%m-%d"), "number": _get_number, "company_id": lambda *a: get_active_company(), 'branch_id': _get_branch, } _order="date desc,number desc" def check_emp(self,name="", employee_id=None): names=name.split(" ") first_name=names[0] last_name=names[-1] if first_name==last_name: first_name="" vals={ 'first_name': first_name, 'last_name': last_name, } emp_obj=get_model("hr.employee") emp_ids=emp_obj.search([['last_name','=',last_name]]) emp_id=None if not emp_ids and not employee_id: emp_id=emp_obj.create(vals) elif employee_id: emp_obj.browse(employee_id).write(vals) emp_id=employee_id else: emp_id=emp_ids[0] return emp_id def create(self,vals,**kw): name=vals['name'] employee_id=vals.get('employee_id') emp_id=self.check_emp(name,employee_id) if emp_id: vals['employee_id']=emp_id new_id=super().create(vals,**kw) return new_id def write(self,ids,vals,**kw): obj=self.browse(ids)[0] name=vals.get("name","") if not name: name=obj.name emp_id=self.check_emp(name,obj.employee_id.id) if emp_id: vals['employee_id']=emp_id super().write(ids,vals,**kw) def name_get(self,ids,context={}): vals=[] for obj in self.browse(ids): level=obj.level_id name=obj.name or "" if level: name+=" (%s)"%(level.name or "") vals.append((obj.id,name)) return vals def name_search(self,name,domain=None,context={},**kw): dom=[["name","ilike","%"+name+"%"]] if domain: dom=[dom,domain] ids1=self.search(dom) dom=[["name","ilike","%"+name+"%"]] level_ids=get_model('clinic.staff.level').search(dom) ids2=[] for x in self.search_read([domain],['level_id']): if x['level_id']: level_id=x['level_id'][0] if level_id in level_ids: ids2.append(x['id']) ids=list(set(ids1+ids2)) return self.name_get(ids,context=context) Staff.register()