403 lines
16 KiB
Python
403 lines
16 KiB
Python
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
|
|
|
|
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_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_rotation(self,ids,context={}):
|
|
res={}
|
|
for obj in self.browse(ids):
|
|
wage=0
|
|
max_cycle=0
|
|
ot_per_cycle=0
|
|
level_id=None
|
|
categ_id=None
|
|
for rotate in sorted(obj.rotations,key=lambda x: x.hire_date or ""):
|
|
if rotate.state=='approved':
|
|
level_id=rotate.level_id.id
|
|
wage=rotate.wage
|
|
max_cycle=rotate.max_cycle
|
|
ot_per_cycle=rotate.ot_per_cycle or 0
|
|
categ_id=rotate.categ_id.id
|
|
res[obj.id]={
|
|
'wage': wage,
|
|
'max_cycle': max_cycle,
|
|
'ot_per_cycle': ot_per_cycle,
|
|
'level_id': level_id,
|
|
'categ_id': categ_id,
|
|
}
|
|
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
|
|
|
|
def _get_name(self,ids,context={}):
|
|
res={}
|
|
for obj in self.browse(ids):
|
|
name=''
|
|
title=obj.title_id
|
|
if title:
|
|
title_name=title.name or ""
|
|
title_name=title_name.replace(" ","")
|
|
if title_name.lower()!='notitle':
|
|
name+=obj.title_id.name or ""
|
|
if obj.first_name:
|
|
name+=obj.first_name or ""
|
|
name+=" "
|
|
if obj.last_name:
|
|
name+=obj.last_name or ""
|
|
if not obj.active:
|
|
name+='__'
|
|
elif context.get('active'):
|
|
name+='__'
|
|
res[obj.id]={
|
|
'name': name,
|
|
'name_check': name.replace(" ",""), # remove all space for make sure
|
|
}
|
|
return res
|
|
|
|
def _get_location(self,ids,context={}):
|
|
res={}
|
|
for obj in self.browse(ids):
|
|
res[obj.id]=",".join([d.code for d in obj.departments])
|
|
return res
|
|
|
|
def _get_department_names(self,ids,context={}):
|
|
res={}
|
|
user_id=get_active_user()
|
|
set_active_user(1)
|
|
for obj in self.browse(ids):
|
|
res[obj.id]=','.join([dpt.name for dpt in obj.departments])
|
|
set_active_user(user_id)
|
|
return res
|
|
|
|
def _get_all(self,ids,context={}):
|
|
res={}
|
|
for obj in self.browse(ids):
|
|
pt_ids=[]
|
|
if obj.type=='doctor':
|
|
pt_ids=get_model("clinic.patient").search([['doctor_id','=',obj.id]])
|
|
elif obj.type=='nurse':
|
|
p=set()
|
|
for cline in obj.cycle_item_lines:
|
|
citem=cline.cycle_item_id
|
|
for hdcase in citem.hd_cases:
|
|
p.update({hdcase.patient_id.id})
|
|
pt_ids=list(p)
|
|
res[obj.id]=pt_ids
|
|
return res
|
|
|
|
_fields={
|
|
'employee_id': fields.Many2One("hr.employee","Employee"),
|
|
"number": fields.Char("Number",required=True,search=True),
|
|
"title_id": fields.Many2One("clinic.name.title","Title"),
|
|
"first_name": fields.Char("First Name"),
|
|
"last_name": fields.Char("Last Name"),
|
|
"name": fields.Char("Name",function="_get_name",function_multi=True,store=True,required=True,search=True),
|
|
"name_check": fields.Char("Name",function="_get_name",function_multi=True,store=True),
|
|
"name_eng": fields.Char("Eng Name",search=True),
|
|
"nick_name": fields.Char("Nick Name",search=True),
|
|
"identification" : fields.Char("ID 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"),
|
|
"patients": fields.Many2Many("clinic.patient","Patients",function="_get_all"),
|
|
"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",function="_get_rotation",function_multi=True,store=True),
|
|
'level_id': fields.Many2One("clinic.staff.level", "Level", function="_get_rotation",function_multi=True,store=True),
|
|
'active': fields.Boolean("Active"),
|
|
'date': fields.Date("Register Date"),
|
|
'wage': fields.Float("Wage", function="_get_rotation",function_multi=True,store=True),
|
|
'max_cycle': fields.Integer("Max Cycle", function="_get_rotation",function_multi=True,store=True),
|
|
'ot_per_cycle': fields.Integer("OT Per Cycle", function="_get_rotation",function_multi=True,store=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_lines": fields.One2Many("clinic.cycle.item.line","nurse_id","Cycle Items"),
|
|
'branch_id': fields.Many2One("clinic.branch","Branch", search=True),
|
|
"partner_id": fields.Many2One("partner","Contact"),
|
|
'departments': fields.Many2Many("clinic.department","Departments"),
|
|
'department_names': fields.Text("Departments",function="_get_department_names"),
|
|
"location": fields.Char("Location",function="_get_location",store=True),
|
|
}
|
|
|
|
def _get_number(self,context={}):
|
|
while 1:
|
|
seq_type='clinic_staff'
|
|
seq_id=get_model("sequence").find_sequence(type=seq_type)
|
|
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]
|
|
|
|
def _get_departments(self,context={}):
|
|
res=get_model('select.company').get_select()
|
|
if res:
|
|
if res.get("department_ids"):
|
|
return res['department_ids']
|
|
else:
|
|
return [res['department_id']]
|
|
return get_model("clinic.department").search([])
|
|
|
|
def _get_default_location(self,context={}):
|
|
res=get_model('select.company').get_select()
|
|
code=''
|
|
if res:
|
|
dpt_ids=[]
|
|
if res.get("department_ids"):
|
|
dpt_ids=res['department_ids']
|
|
else:
|
|
dpt_ids=[res['department_id']]
|
|
dpts=get_model("clinic.department").browse(dpt_ids)
|
|
code=','.join([dpt.code for dpt in dpts])
|
|
return code
|
|
|
|
def _get_department(self,context={}):
|
|
res=get_model('select.company').get_select()
|
|
if res:
|
|
if res.get("department_ids"):
|
|
return res['department_ids'][0]
|
|
else:
|
|
return res['department_id']
|
|
|
|
_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,
|
|
'departments': _get_departments,
|
|
'department_id': _get_department,
|
|
'location': _get_default_location,
|
|
}
|
|
|
|
#_key=["name_check","branch_id"] #not working
|
|
_sql_constraints=("clinic_staff_key_uniq","unique(name_check,branch_id)","name should be unique"),
|
|
_order="date desc,number desc"
|
|
|
|
def create_contact(self,code,name):
|
|
partner_id=get_model("partner").create({
|
|
'code': code,
|
|
'name': name,
|
|
'last_name': name,
|
|
'type': 'person',
|
|
'is_staff': True,
|
|
})
|
|
address_id=get_model('address').create({
|
|
'type': 'shipping',
|
|
'partner_id': partner_id,
|
|
'address': 'your address',
|
|
'address2': 'your address2',
|
|
'city': 'your city',
|
|
'postal_code': 'your zip',
|
|
'country_id': 1,
|
|
|
|
})
|
|
return partner_id, address_id
|
|
|
|
def check_contact(self,name=""):
|
|
partner_id=None
|
|
dom=[
|
|
['name','=',name],
|
|
['is_staff','=',True],
|
|
]
|
|
for partner in get_model("partner").search_browse(dom):
|
|
if partner.name==name:
|
|
partner_id=partner.id
|
|
break
|
|
return partner_id
|
|
|
|
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):
|
|
title_id=vals['title_id']
|
|
title=get_model("clinic.name.title").browse(title_id)
|
|
title_name=title.name or ""
|
|
vals['name']=''
|
|
if title_name.lower()=='notitle':
|
|
vals['name']='%s%s %s'%(title_name,vals['first_name'],vals['last_name'])
|
|
else:
|
|
vals['name']='%s %s'%(vals['first_name'],vals['last_name'])
|
|
partner_name='%s %s'%(vals['first_name'],vals['last_name']) #XXX
|
|
employee_id=vals.get('employee_id')
|
|
name=vals['name']
|
|
emp_id=self.check_emp(name,employee_id)
|
|
if emp_id:
|
|
vals['employee_id']=emp_id
|
|
partner_id=get_model("clinic.staff").check_contact(partner_name)
|
|
address_id=None
|
|
if not partner_id:
|
|
vals['partner_id'],address_id=self.create_contact(vals['number'],partner_name)
|
|
new_id=super().create(vals,**kw)
|
|
self.function_store([new_id])
|
|
for obj in self.browse([new_id]):
|
|
if address_id:
|
|
addr=get_model('address').browse(address_id)
|
|
addr.write({
|
|
'partner_id': vals['partner_id'],
|
|
'staff_id': new_id,
|
|
})
|
|
if not obj.departments:
|
|
dpt_ids=get_model("clinic.department").search([])
|
|
for st in get_model("clinic.staff").search_browse([]):
|
|
if not st.departments:
|
|
st.write({
|
|
'departments': [['set',dpt_ids]],
|
|
})
|
|
print('set all department for ', st.name)
|
|
|
|
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
|
|
partner_name='%s %s'%(obj.first_name or "",obj.last_name or "")
|
|
partner_id=get_model("clinic.staff").check_contact(partner_name)
|
|
address_id=None
|
|
if not partner_id:
|
|
vals['partner_id'],address_id=self.create_contact(partner_name,partner_name) #XXX
|
|
super().write(ids,vals,**kw)
|
|
self.function_store(ids)
|
|
if address_id:
|
|
addr=get_model('address').browse(address_id)
|
|
addr.write({
|
|
'partner_id': vals['partner_id'],
|
|
'staff_id': ids[0],
|
|
})
|
|
if not obj.departments:
|
|
dpt_ids=get_model("clinic.department").search([])
|
|
for st in get_model("clinic.staff").search_browse([]):
|
|
if not st.departments:
|
|
st.write({
|
|
'departments': [['set',dpt_ids]],
|
|
})
|
|
print('set all department for ', st.name)
|
|
|
|
def name_get(self,ids,context={}):
|
|
vals=[]
|
|
for obj in self.browse(ids):
|
|
name=obj.name or ""
|
|
nickname=obj.nick_name or ""
|
|
if nickname:
|
|
name+=" (%s)"%(nickname)
|
|
vals.append((obj.id,name))
|
|
return vals
|
|
|
|
def name_search(self,name,domain=None,condition=[],context={},**kw):
|
|
dom=[["name","ilike","%"+name+"%"]]
|
|
if domain:
|
|
dom=[dom,domain]
|
|
ids1=self.search(dom)
|
|
dom=[["nick_name","ilike","%"+name+"%"]]
|
|
if domain:
|
|
dom=[dom,domain]
|
|
ids2=self.search(dom)
|
|
ids=list(set(ids1+ids2))
|
|
return self.name_get(ids,context=context)
|
|
|
|
Staff.register()
|