open-synergy/opnsynid-hr

# -*- coding: utf-8 -*-
# Copyright 2018 OpenSynergy Indonesia
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

from openerp import api, fields, models


class CancelParticipant(models.TransientModel):
    _name = "hr.cancel_participant"
    _description = "Cancel Participant"

    @api.model
    def _default_training_id(self):
        return self.env.context.get("active_id", False)

    @api.model
    def _default_detail_ids(self):
        training_id = self.env.context.get("active_id", False)
        obj_participant = self.env["hr.training_partisipant"]
        criteria = [("training_id", "=", training_id), ("state", "=", "draft")]
        participants = obj_participant.search(criteria)
        result = []
        for participant in participants:
            data = {
                "participant_id": participant.id,
                "employee_id": participant.partisipant_id.id,
            }
            result.append((0, 0, data))
        return result

    training_id = fields.Many2one(
        string="Training",
        comodel_name="hr.training",
        required=True,
        default=lambda self: self._default_training_id(),
    )

    detail_ids = fields.One2many(
        string="Participants",
        comodel_name="hr.cancel_participant_detail",
        inverse_name="wizard_id",
        default=lambda self: self._default_detail_ids(),
    )

    @api.multi
    def action_confirm(self):
        for wiz in self:
            wiz._confirm_cancel()

    @api.multi
    def _confirm_cancel(self):
        self.ensure_one()
        for detail in self.detail_ids:
            detail._confirm_cancel()


class CancelParticipantDetail(models.TransientModel):
    _name = "hr.cancel_participant_detail"
    _description = "Detail Cancel Participant"

    wizard_id = fields.Many2one(
        string="Wizard",
        comodel_name="hr.cancel_participant",
    )
    participant_id = fields.Many2one(
        string="Participant",
        comodel_name="hr.training_partisipant",
    )
    employee_id = fields.Many2one(
        string="Employee",
        comodel_name="hr.employee",
        related="participant_id.partisipant_id",
        store=False,
    )
    cancel_reason_id = fields.Many2one(
        string="Cancel Reason",
        comodel_name="hr.training_participant_cancel_reason",
        required=True,
    )
    substitute_by_id = fields.Many2one(
        string="Substitute By",
        comodel_name="hr.employee",
    )

    @api.multi
    def _confirm_cancel(self):
        self.ensure_one()
        if self.substitute_by_id:
            self.participant_id._substitute(self.substitute_by_id)
        self.participant_id.button_cancel(self.cancel_reason_id.id)