cooperative-market/beesdoo_shift/wizard/validate_attendance_sheet.py

# -*- coding: utf-8 -*-
from openerp import _, api, exceptions, fields, models
from openerp.exceptions import UserError, ValidationError


class ValidateAttendanceSheet(models.TransientModel):
    _name = "beesdoo.shift.sheet.validate"
    _description = """Check the user name and validate sheet.
    Useless for users in group_cooperative_admin"""
    _inherit = ["barcodes.barcode_events_mixin"]

    def _get_card_support_setting(self):
        return (
            self.env["ir.config_parameter"].get_param(
                "beesdoo_shift.card_support"
            )
            == "True"
        )

    @api.multi
    def _default_annotation(self):
        """
        The annotation is pre-filled with a warning message
        if a regular worker is added and should have been expected.
        """

        sheet_id = self._context.get("active_id")
        sheet_model = self._context.get("active_model")
        sheet = self.env[sheet_model].browse(sheet_id)
        warning_message = ""

        for added_shift in sheet.added_shift_ids:
            is_regular_worker = added_shift.worker_id.working_mode == "regular"
            is_compensation = added_shift.is_compensation

            if is_regular_worker and not is_compensation:
                warning_message += (
                    _(
                        "Warning : %s attended its shift as a normal one but was not expected. "
                        "Something may be wrong in his/her personnal informations.\n\n"
                    )
                    % added_shift.worker_id.name
                )
        return warning_message

    card_support = fields.Boolean(default=_get_card_support_setting)
    user_id = fields.Many2one("res.users", string="Login")
    password = fields.Char(string="Password")
    barcode = fields.Char(string="Barcode")
    annotation = fields.Text(
        "Important information requiring permanent member assistance",
        default=_default_annotation,
    )
    feedback = fields.Text("General feedback")
    worker_nb_feedback = fields.Selection(
        [
            ("not_enough", "Not enough"),
            ("enough", "Enough"),
            ("too_many", "Too many"),
        ],
        string="Number of workers",
        required=True,
    )

    def on_barcode_scanned(self, barcode):
        self.barcode = barcode

    @api.multi
    def validate_sheet(self):
        sheet_id = self._context.get("active_id")
        sheet_model = self._context.get("active_model")
        sheet = self.env[sheet_model].browse(sheet_id)

        if self.card_support:
            # Login with barcode
            card = self.env["member.card"].search(
                [("barcode", "=", self.barcode)]
            )
            if not len(card):
                raise UserError(_("Please set a correct barcode."))
            partner = card[0].partner_id
        else:
            # Login with credentials
            if not self.user_id:
                raise UserError(_("Please enter an user name."))
            self.user_id.sudo(self.user_id.id).check_credentials(self.password)
            partner = self.user_id.partner_id

        is_admin = partner.user_ids.has_group(
            "beesdoo_shift.group_cooperative_admin"
        )

        if not partner.super and not is_admin:
            raise UserError(
                _(
                    "Only super-cooperators and administrators can validate attendance sheets."
                )
            )
        if self.annotation:
            sheet.annotation += self.annotation
        if sheet.feedback:
            sheet.feedback += self.feedback
        sheet.worker_nb_feedback = self.worker_nb_feedback
        sheet._validate(partner or self.env.user.partner_id)