partner-contact/partner_pricelist_search/models/res_partner.py

33 lines
1.1 KiB
Python
Raw Normal View History

2024-05-03 08:58:43 +00:00
# Copyright 2021 Tecnativa - Carlos Dauden
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from flectra import api, fields, models
class ResPartner(models.Model):
_inherit = "res.partner"
property_product_pricelist = fields.Many2one(
search="_search_property_product_pricelist"
)
@api.model
def search(self, args, offset=0, limit=None, order=None):
# Substitute pricelist tuple
partner_domain = [
(1, "=", 1)
if (isinstance(x, (list | tuple)) and x[0] == "property_product_pricelist")
else x
for x in args
]
return super(
ResPartner, self.with_context(search_partner_domain=partner_domain)
).search(args, offset=offset, limit=limit, order=order)
@api.model
def _search_property_product_pricelist(self, operator, value):
domain = self.env.context.get("search_partner_domain", [])
partners = self.with_context(prefetch_fields=False).search(domain)
key = "property_product_pricelist"
return [("id", "in", partners.filtered_domain([(key, operator, value)]).ids)]