partner-contact/partner_ref_unique/models/res_company.py
2021-03-23 20:13:18 +01:00

34 lines
1006 B
Python

# Copyright 2016 Antonio Espinosa
# Copyright 2020 Tecnativa - João Marques
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from flectra import fields, models
class ResCompany(models.Model):
_inherit = "res.company"
partner_ref_unique = fields.Selection(
selection=[
("none", "None"),
("companies", "Only companies"),
("all", "All partners"),
],
string="Unique partner reference for",
default="none",
)
def write(self, vals):
"""Launch manually the constraint check in partners as current ORM
doesn't trigger the constraint on related fields.
"""
res = super().write(vals)
if "partner_ref_unique" in vals:
partners = (
self.env["res.partner"]
.with_context(active_test=False)
.search([("company_id", "in", [False] + self.ids)])
)
partners._check_ref()
return res