mirror of
https://gitlab.com/flectra-community/partner-contact.git
synced 2024-11-15 02:32:04 +00:00
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
# Copyright 2021 ForgeFlow, S.L.
|
|
# Copyright 2022 Vauxoo, S.A.
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
|
|
|
|
from flectra import _, api, fields, models
|
|
from flectra.exceptions import UserError
|
|
|
|
DOMAIN_SEARCH = {
|
|
("=", False): ("=", 0),
|
|
("=", True): (">=", 1),
|
|
("!=", False): (">=", 1),
|
|
("!=", True): ("=", 0),
|
|
}
|
|
|
|
|
|
class ResPartner(models.Model):
|
|
_inherit = "res.partner"
|
|
|
|
is_customer = fields.Boolean(
|
|
compute="_compute_is_customer",
|
|
inverse="_inverse_is_customer",
|
|
search="_search_is_customer",
|
|
string="Is a Customer",
|
|
default=lambda self: self._default_is_customer(),
|
|
)
|
|
is_supplier = fields.Boolean(
|
|
compute="_compute_is_supplier",
|
|
inverse="_inverse_is_supplier",
|
|
search="_search_is_supplier",
|
|
string="Is a Supplier",
|
|
default=lambda self: self._default_is_supplier(),
|
|
)
|
|
|
|
@api.depends("customer_rank")
|
|
def _compute_is_customer(self):
|
|
for partner in self:
|
|
if not partner.is_customer:
|
|
partner.is_customer = bool(partner.customer_rank)
|
|
|
|
@api.depends("supplier_rank")
|
|
def _compute_is_supplier(self):
|
|
for partner in self:
|
|
if not partner.is_supplier:
|
|
partner.is_supplier = bool(partner.supplier_rank)
|
|
|
|
def _inverse_is_customer(self):
|
|
self.filtered(lambda p: not p.is_customer).write({"customer_rank": 0})
|
|
self.filtered(lambda p: p.is_customer and not p.customer_rank).write(
|
|
{"customer_rank": 1}
|
|
)
|
|
|
|
def _search_is_customer(self, operator, value):
|
|
if operator not in ["=", "!="] or not isinstance(value, bool):
|
|
raise UserError(_("Operation not supported"))
|
|
operator, value = DOMAIN_SEARCH.get((operator, value))
|
|
return [("customer_rank", operator, value)]
|
|
|
|
def _inverse_is_supplier(self):
|
|
self.filtered(lambda p: not p.is_supplier).write({"supplier_rank": 0})
|
|
self.filtered(lambda p: p.is_supplier and not p.supplier_rank).write(
|
|
{"supplier_rank": 1}
|
|
)
|
|
|
|
def _search_is_supplier(self, operator, value):
|
|
if operator not in ["=", "!="] or not isinstance(value, bool):
|
|
raise UserError(_("Operation not supported"))
|
|
operator, value = DOMAIN_SEARCH.get((operator, value))
|
|
return [("supplier_rank", operator, value)]
|
|
|
|
def _default_is_customer(self):
|
|
return self.env.context.get("res_partner_search_mode") == "customer"
|
|
|
|
def _default_is_supplier(self):
|
|
return self.env.context.get("res_partner_search_mode") == "supplier"
|