mirror of
https://gitlab.com/flectra-community/partner-contact.git
synced 2024-11-15 02:32:04 +00:00
31 lines
1002 B
Python
31 lines
1002 B
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):
|
||
|
|
return ["delivery", "invoice"]
|
||
|
|
|
||
|
|
def address_get(self, adr_pref=None):
|
||
|
|
res = super().address_get(adr_pref)
|
||
|
|
default_address_type_list = 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)]
|
||
|
|
if default_address_id:
|
||
|
|
res[addr_type] = default_address_id
|
||
|
|
return res
|