open-synergy/opnsynid-hr

# Copyright 2018 OpenSynergy Indonesia
# Copyright 2020 PT. Simetri Sinergi Indonesia
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from openerp import api, models


class HrTerminationTransition(models.Model):
    _name = "hr.termination_transition"
    _inherit = ["hr.career_transition"]
    _description = "Career Transition - Termination"
    _table = "hr_career_transition"

    @api.model
    def _default_type_id(self):
        return self.env.ref(
            "hr_termination_transition." "career_transition_termination"
        ).id

    @api.model
    def search(self, args, offset=0, limit=None, order=None, count=False):
        type_id = (
            self.env.ref(
                "hr_termination_transition." "career_transition_termination", False
            )
            and self.env.ref(
                "hr_termination_transition." "career_transition_termination"
            )
            or self.env["hr." "career_transition_" "type"]
        )
        args.append(("type_id", "=", type_id.id))
        return super(HrTerminationTransition, self).search(
            args=args, offset=offset, limit=limit, order=order, count=count
        )

    @api.multi
    def _prepare_valid_data(self):
        _super = super(HrTerminationTransition, self)
        res = _super._prepare_valid_data()
        self.ensure_one()
        check_employee = self.employee_id
        check_employee.write({"active": False})
        check_user = check_employee.user_id
        if check_user:
            check_user.write({"active": False})
        return res

    @api.multi
    def _prepare_cancel_data(self):
        _super = super(HrTerminationTransition, self)
        res = _super._prepare_cancel_data()
        self.ensure_one()
        check_employee = self.employee_id
        check_employee.write({"active": True})
        check_user = check_employee.user_id
        if check_user:
            check_user.write({"active": True})
        return res