social/mass_mailing_partner/models/mailing_trace.py

38 lines
1.2 KiB
Python
Raw Normal View History

# Copyright 2016 Antonio Espinosa - <antonio.espinosa@tecnativa.com>
# Copyright 2020 Tecnativa - Manuel Calero
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from odoo import api, fields, models
class MailingTrace(models.Model):
_inherit = "mailing.trace"
partner_id = fields.Many2one(
string="Partner", comodel_name="res.partner", readonly=True
)
@api.model
def partner_id_from_obj(self, model, res_id):
partner_id = False
obj = self.env[model].browse(res_id)
if obj.exists():
if model == "res.partner":
partner_id = res_id
elif "partner_id" in obj._fields:
partner_id = obj.partner_id.id
return partner_id
def partner_link(self):
for stat in self.filtered(lambda r: r.model and r.res_id):
partner_id = self.partner_id_from_obj(stat.model, stat.res_id)
if partner_id != stat.partner_id.id:
stat.partner_id = partner_id
return True
@api.model
def create(self, vals):
stat = super(MailingTrace, self).create(vals)
stat.partner_link()
return stat