# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.

from odoo import fields, models


class MailComposeMessage(models.TransientModel):
    _inherit = 'mail.compose.message'

    marketing_activity_id = fields.Many2one('marketing.activity', string='Marketing Activity')

    def _prepare_mail_values_mailing_traces(self, mail_values_all):
        """ Override method to link mail automation activity with mail statistics"""
        trace_values_all = super()._prepare_mail_values_mailing_traces(mail_values_all)

        # skip update if no marketing activity is linked to the mailing
        if not self.marketing_activity_id:
            return trace_values_all

        # retrieve trace linked to recipient
        traces = self.env['marketing.trace'].search([
            ('activity_id', '=', self.marketing_activity_id.id),
            ('res_id', 'in', list(trace_values_all.keys()))
        ])
        traces_mapping = dict((trace.res_id, trace.id) for trace in traces)

        # update statistics creation done in mass_mailing to include link between stat and trace
        for res_id, trace_values in trace_values_all.items():
            trace_values['marketing_trace_id'] = traces_mapping[res_id]

        return trace_values_all