clinic/netforce_clinic/models/staff_rotation.py

import time

from netforce.model import Model, fields
from netforce.access import get_active_company

class StaffRotation(Model):
    _name="clinic.staff.rotation"
    _string="Staff Rotation"
    _multi_company=True

    _fields={
        "type": fields.Selection([['staff','Staff'],["doctor","Doctor"],["nurse","Nurse"]],"Type",search=True),
        "staff_id": fields.Many2One("clinic.staff","Staff", search=True),
        'categ_id': fields.Many2One("clinic.staff.categ","Category",search=True),
        "level_id": fields.Many2One("clinic.staff.level","Level", search=True),
        "hire_date": fields.Date("Hire Date", search=True),
        "resign_date": fields.Date("Resign Date", search=True),
        "wage": fields.Float("Wage"),
        "max_cycle": fields.Integer("Max Cycle"),
        "ot_per_cycle": fields.Float("OT/Cycle"),
        "note": fields.Text("Note"),
        'company_id': fields.Many2One("company","Company"),
        'state': fields.Selection([['draft','Draft'],['approved','Approved']],'State'),
    }

    _defaults={
        "company_id": lambda *a: get_active_company(),
        'hire_date': time.strftime("%Y-%m-%d"),
        'state': 'draft',
        'type': 'nurse',
    }

    def to_draft(self,ids,context={}):
        for obj in self.browse(ids):
            obj.write({
                'state': 'draft',
            })

    def approve(self,ids,context={}):
        for obj in self.browse(ids):
            obj.write({
                'state': 'approved',
            })

StaffRotation.register()