You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
131 lines
4.3 KiB
131 lines
4.3 KiB
# -*- 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)
|