|
|
# -*- coding: utf-8 -*- from openerp import models, fields, api, _ from openerp.exceptions import ValidationError
class Partner(models.Model):
_inherit = "res.partner"
eater = fields.Selection( [("eater", "Eater"), ("worker_eater", "Worker and Eater")], string="Eater/Worker", ) child_eater_ids = fields.One2many( "res.partner", "parent_eater_id", domain=[("customer", "=", True), ("eater", "=", "eater")], ) parent_eater_id = fields.Many2one( "res.partner", string="Parent Worker", readonly=True ) barcode = fields.Char( compute="_get_bar_code", string="Barcode", store=True ) parent_barcode = fields.Char( compute="_get_bar_code", string="Parent Barcode", store=True ) member_card_ids = fields.One2many("member.card", "partner_id") country_id = fields.Many2one( required=True, default=lambda self: self.env.ref("base.be") )
member_card_to_be_printed = fields.Boolean("Print BEES card?") last_printed = fields.Datetime("Last printed on") cooperator_type = fields.Selection( [ ("share_a", "Share A"), ("share_b", "Share B"), ("share_c", "Share C"), ], store=True, compute=None, )
@api.one @api.depends( "parent_eater_id", "parent_eater_id.barcode", "eater", "member_card_ids", ) def _get_bar_code(self): if self.eater == "eater": self.parent_barcode = self.parent_eater_id.barcode elif self.member_card_ids: for c in self.member_card_ids: if c.valid: self.barcode = c.barcode
@api.one @api.constrains("child_eater_ids", "parent_eater_id") def _check_number_of_eaters(self): """The owner of an A share can have a maximum of two eaters but
the owner of a B share can have a maximum of three eaters. """
# Get the default_code of the share for the current eater and his parent share_type_code = self.cooperator_type parent_share_type_code = self.parent_eater_id.cooperator_type # Raise exception if share_type_code == "share_b" or parent_share_type_code == "share_b": if ( len(self.child_eater_ids) > 3 or len(self.parent_eater_id.child_eater_ids) > 3 ): raise ValidationError( _("You can only set three additional eaters per worker") ) else: if ( len(self.child_eater_ids) > 2 or len(self.parent_eater_id.child_eater_ids) > 2 ): raise ValidationError( _("You can only set two additional eaters per worker") )
@api.multi def write(self, values): if values.get("parent_eater_id") and self.parent_eater_id: raise ValidationError( _( "You try to assign a eater to a worker but this easer is alread assign to %s please remove it before" ) % self.parent_eater_id.name ) # replace many2many command when writing on child_eater_ids to just remove the link if "child_eater_ids" in values: for command in values["child_eater_ids"]: if command[0] == 2: command[0] = 3 return super(Partner, self).write(values)
@api.one def _deactivate_active_cards(self): for card in self.member_card_ids.filtered("valid"): card.valid = False card.end_date = fields.Date.today()
@api.multi def _new_card(self, reason, user_id, barcode=False): card_data = { "partner_id": self.id, "responsible_id": user_id, "comment": reason, } if barcode: card_data["barcode"] = barcode self.env["member.card"].create(card_data)
@api.multi def _new_eater(self, surname, name, email): partner_data = { "lastname": name, "firstname": surname, "is_customer": True, "eater": "eater", "parent_eater_id": self.id, "email": email, "country_id": self.country_id.id, } return self.env["res.partner"].create(partner_data)
|