cooperative-market/beesdoo_base/wizard/member_card.py

# -*- coding: utf-8 -*-
from openerp import models, fields, api


class NewMemberCardWizard(models.TransientModel):
    """
        A transient model for the creation of a new card.
        The user can only define the raison why a new card is
        needed and the eater/worker that is concerned.
    """

    _name = "membercard.new.wizard"

    def _get_default_partner(self):
        return self.env.context["active_id"]

    new_comment = fields.Char("Reason", required=True)
    partner_id = fields.Many2one("res.partner", default=_get_default_partner)
    force_barcode = fields.Char(
        "Force Barcode", groups="beesdoo_base.group_force_barcode"
    )

    @api.one
    def create_new_card(self):
        client = self.partner_id.sudo()
        client._deactivate_active_cards()
        client._new_card(
            self.new_comment, self.env.uid, barcode=self.force_barcode
        )
        client.member_card_to_be_printed = True


class RequestMemberCardPrintingWizard(models.TransientModel):

    _name = "membercard.requestprinting.wizard"

    def _get_selected_partners(self):
        return self.env.context["active_ids"]

    partner_ids = fields.Many2many(
        "res.partner", default=_get_selected_partners
    )

    @api.one
    def request_printing(self):
        self.partner_ids.write({"member_card_to_be_printed": True})


class SetAsPrintedWizard(models.TransientModel):

    _name = "membercard.set_as_printed.wizard"

    def _get_selected_partners(self):
        return self.env.context["active_ids"]

    partner_ids = fields.Many2many(
        "res.partner", default=_get_selected_partners
    )

    @api.one
    def set_as_printed(self):
        self.partner_ids.write(
            {
                "member_card_to_be_printed": False,
                "last_printed": fields.Datetime.now(),
            }
        )