clinic/netforce_clinic/models/staff.py

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_name=''
            title=obj.title_id
            name_check=""
            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_check+=obj.first_name or ""
                name+=" "
            if obj.last_name:
                name+=obj.last_name or ""
                name_check+=obj.last_name or ""
            if not obj.active:
                name+='__'
            elif context.get('active'):
                name+='__'
            res[obj.id]={
                'name': name,
                'name_check': name_check, # remove all space for make sure
            }
        return res
    
    def _get_location(self,ids,context={}):
        res={}
        user_id=get_active_user()
        set_active_user(1)
        for obj in self.browse(ids):
            st_codes=set([d.code for d in obj.departments])
            res[obj.id]=",".join([dcode for dcode in sorted(st_codes)])
        set_active_user(user_id)
        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):
            dpt_st=set([dpt.name for dpt in obj.departments])
            res[obj.id]=','.join([dpt_name for dpt_name in sorted(dpt_st)])
        set_active_user(user_id)
        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),
        "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,
    }

    _sql_constraints=("clinic_staff_key_uniq","unique(name_check)","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()