GOLEM/golem_mail/wizard/mail_compose_message.py

# -*- coding: utf-8 -*-

#    Copyright 2018 Youssef El Ouahby <youssef@yaltik.com>
#    Copyright 2018 Fabien Bourgeois <fabien@yaltik.com>
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 of the
#    License, or (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
""" Mail Compose Message Adaptations """

from odoo import _, api, fields, models, SUPERUSER_ID, tools
from odoo.tools.safe_eval import safe_eval



class MailComposer(models.TransientModel):
    """ Mail Compose Message Adaptations """
    _inherit = 'mail.compose.message'

    from_presend_wizard = fields.Boolean()

    @api.multi
    def send_mail(self, auto_commit=False):
        """ Process the wizard content and proceed with sending the related
            email(s), rendering any template patterns on the fly if needed. """
        for wizard in self:
            # Duplicate attachments linked to the email.template.
            # Indeed, basic mail.compose.message wizard duplicates attachments in mass
            # mailing mode. But in 'single post' mode, attachments of an email template
            # also have to be duplicated to avoid changing their ownership.
            if wizard.attachment_ids and wizard.composition_mode != 'mass_mail' and wizard.template_id:
                new_attachment_ids = []
                for attachment in wizard.attachment_ids:
                    if attachment in wizard.template_id.attachment_ids:
                        new_attachment_ids.append(attachment.copy({'res_model': 'mail.compose.message', 'res_id': wizard.id}).id)
                    else:
                        new_attachment_ids.append(attachment.id)
                    wizard.write({'attachment_ids': [(6, 0, new_attachment_ids)]})

            # Mass Mailing
            mass_mode = wizard.composition_mode in ('mass_mail', 'mass_post')

            Mail = self.env['mail.mail']
            ActiveModel = self.env[wizard.model if wizard.model else 'mail.thread']
            if wizard.template_id:
                # template user_signature is added when generating body_html
                # mass mailing: use template auto_delete value -> note, for emails mass mailing only
                Mail = Mail.with_context(mail_notify_user_signature=False)
                ActiveModel = ActiveModel.with_context(mail_notify_user_signature=False, mail_auto_delete=wizard.template_id.auto_delete)
            if not hasattr(ActiveModel, 'message_post'):
                ActiveModel = self.env['mail.thread'].with_context(thread_model=wizard.model)
            if wizard.composition_mode == 'mass_post':
                # do not send emails directly but use the queue instead
                # add context key to avoid subscribing the author
                ActiveModel = ActiveModel.with_context(mail_notify_force_send=False, mail_create_nosubscribe=True)
            # wizard works in batch mode: [res_id] or active_ids or active_domain
            if mass_mode and wizard.use_active_domain and wizard.model:
                res_ids = self.env[wizard.model].search(safe_eval(wizard.active_domain)).ids
            elif mass_mode and wizard.model and self._context.get('active_ids'):
                if wizard.from_presend_wizard:
                    res_ids = self._context['partners_from_wizard']
                else:
                    res_ids = self._context['active_ids']
            else:
                res_ids = [wizard.res_id]

            batch_size = int(self.env['ir.config_parameter'].sudo().get_param('mail.batch_size')) or self._batch_size
            sliced_res_ids = [res_ids[i:i + batch_size] for i in range(0, len(res_ids), batch_size)]

            if wizard.composition_mode == 'mass_mail' or wizard.is_log or (wizard.composition_mode == 'mass_post' and not wizard.notify):  # log a note: subtype is False
                subtype_id = False
            elif wizard.subtype_id:
                subtype_id = wizard.subtype_id.id
            else:
                subtype_id = self.sudo().env.ref('mail.mt_comment', raise_if_not_found=False).id
            batch_mails = Mail
            for res_ids in sliced_res_ids:
                all_mail_values = wizard.get_mail_values(res_ids)
                for res_id, mail_values in all_mail_values.iteritems():
                    if wizard.composition_mode == 'mass_mail':
                        batch_mails |= Mail.create(mail_values)
                    else:
                        ActiveModel.browse(res_id).message_post(
                            message_type=wizard.message_type,
                            subtype_id=subtype_id,
                            **mail_values)
            if self.env.context.get('emails_from_wizard', False):
                mails = self._context['emails_from_wizard']
                emails_values = wizard.get_mail_values_from_mails(mails)
                for email_val in emails_values:
                    batch_mails |= Mail.create(email_val)

            if wizard.composition_mode == 'mass_mail':
                batch_mails.send(auto_commit=auto_commit)

        return {'type': 'ir.actions.act_window_close'}

    @api.multi
    def get_mail_values_from_mails(self, mails):
        self.ensure_one()
        mail_values = {
            'subject': self.subject,
            'body': self.body or '',
            'parent_id': self.parent_id and self.parent_id.id,
            'partner_ids': [partner.id for partner in self.partner_ids],
            'attachment_ids': [attach.id for attach in self.attachment_ids],
            'author_id': self.author_id.id,
            'email_from': self.email_from,
            'record_name': self.record_name,
            'no_auto_thread': self.no_auto_thread,
            'mail_server_id': self.mail_server_id.id,
            }
        mail_values.update(notification=not self.auto_delete_message,
                           model=self.model,
                           record_name=False)
        # auto deletion of mail_mail
        if self.auto_delete or self.template_id.auto_delete:
            mail_values['auto_delete'] = True
        mail_values['reply_to'] = mail_values['email_from']
        mail_values['body_html'] = mail_values.get('body', '')
        attachment_ids = []
        for attach_id in mail_values.pop('attachment_ids'):
            new_attach_id = self.env['ir.attachment'].browse(attach_id).copy({
                'res_model': self._name,
                'res_id': self.id
                })
            attachment_ids.append(new_attach_id.id)
        mail_values['attachment_ids'] = self.env['mail.thread']._message_preprocess_attachments(
            mail_values.pop('attachments', []),
            attachment_ids, 'mail.message', 0)
        emails_result = []
        for email in mails:
            mail_values['email_to'] = email
            emails_result.append(dict(mail_values))
        return emails_result