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"]

    @api.multi
    def _get_active_sheet(self):
        sheet_id = self._context.get("active_id")
        sheet_model = self._context.get("active_model")

        if sheet_id and sheet_model:
            return self.env[sheet_model].browse(sheet_id)

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

    @api.multi
    def _get_warning_regular_workers(self):
        """
        A warning is shown if some regular workers were not expected
        but should be doing their regular shifts. This warning is added
        to sheet's annotation at validation.
        """
        sheet = self._get_active_sheet()
        warning_message = ""
        if sheet:
            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 += (
                        _(
                            "\n%s attended its shift as a normal one but was not expected. "
                            "Something may be wrong in his/her personnal informations.\n"
                        )
                        % added_shift.worker_id.name
                    )
        return warning_message

    @api.multi
    def _get_default_annotation(self):
        if self._get_active_sheet():
            return self._get_active_sheet().annotation

    @api.multi
    def _get_default_feedback(self):
        if self._get_active_sheet():
            return self._get_active_sheet().feedback

    @api.multi
    def _get_default_worker_nb_feedback(self):
        if self._get_active_sheet():
            return self._get_active_sheet().worker_nb_feedback

    card_support = fields.Boolean(default=_get_card_support_setting)
    login = fields.Char(string="Login")
    password = fields.Char(string="Password")
    barcode = fields.Char(string="Barcode")
    warning_regular_workers = fields.Text("Warning",
        default=_get_warning_regular_workers,
        help="Is any regular worker doing its regular shift as an added one ?"
    )
    annotation = fields.Text(
        "Important information requiring permanent member assistance",
        default=_get_default_annotation,
    )
    feedback = fields.Text("General feedback", default=_get_default_feedback)
    worker_nb_feedback = fields.Selection(
        [
            ("not_enough", "Not enough"),
            ("enough", "Enough"),
            ("too_many", "Too many"),
        ],
        string="Number of workers",
        default=_get_default_worker_nb_feedback,
        required=True,
    )

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

    @api.multi
    def save(self):
        """
        Save modifications onto attendance sheet.
        """
        sheet = self._get_active_sheet()

        sheet.annotation = self.annotation
        sheet.feedback = self.feedback
        sheet.worker_nb_feedback = self.worker_nb_feedback

    @api.multi
    def validate_sheet(self):
        sheet = self._get_active_sheet()

        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.login:
                raise UserError(_("Please enter your login."))
            user = self.env["res.users"].search([("login", "=", self.login)])
            user.sudo(user.id).check_credentials(self.password)
            partner = user.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."
                )
            )

        self.annotation += self.warning_regular_workers
        self.save()
        sheet._validate(partner or self.env.user.partner_id)