mirror of
https://gitlab.com/flectra-community/partner-contact.git
synced 2024-11-15 02:32:04 +00:00
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
# Copyright 2020 Tecnativa - Carlos Dauden
|
|
# Copyright 2020 Tecnativa - Sergio Teruel
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
|
|
from flectra import fields, models
|
|
|
|
|
|
class ResPartner(models.Model):
|
|
_inherit = "res.partner"
|
|
|
|
partner_delivery_id = fields.Many2one(
|
|
comodel_name="res.partner",
|
|
string="Shipping address",
|
|
)
|
|
partner_invoice_id = fields.Many2one(
|
|
comodel_name="res.partner",
|
|
string="Invoice address",
|
|
)
|
|
|
|
def get_address_default_type(self):
|
|
"""This will be the extension method for other contact types"""
|
|
return ["delivery", "invoice"]
|
|
|
|
def address_get(self, adr_pref=None):
|
|
"""Force the delivery or invoice addresses. It will try to default
|
|
to the one set in the commercial partner if any"""
|
|
res = super().address_get(adr_pref)
|
|
adr_pref = adr_pref or []
|
|
default_address_type_list = {
|
|
x for x in adr_pref if x in self.get_address_default_type()
|
|
}
|
|
for partner in self:
|
|
for addr_type in default_address_type_list:
|
|
default_address_id = (
|
|
partner["partner_{}_id".format(addr_type)]
|
|
or partner.commercial_partner_id["partner_{}_id".format(addr_type)]
|
|
)
|
|
if default_address_id:
|
|
res[addr_type] = default_address_id.id
|
|
return res
|
|
|
|
def write(self, vals):
|
|
"""We want to prevent archived contacts as default addresses"""
|
|
if vals.get("active") is False:
|
|
self.search([("partner_delivery_id", "in", self.ids)]).write(
|
|
{"partner_delivery_id": False}
|
|
)
|
|
self.search([("partner_invoice_id", "in", self.ids)]).write(
|
|
{"partner_invoice_id": False}
|
|
)
|
|
return super().write(vals)
|