You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
3.8 KiB
103 lines
3.8 KiB
# -*- coding: utf-8 -*-
|
|
# Copyright 2016 Antonio Espinosa <antonio.espinosa@tecnativa.com>
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
from openerp import api, fields, models, _
|
|
from openerp.exceptions import Warning as UserError
|
|
|
|
|
|
class MailMassMailing(models.Model):
|
|
_inherit = 'mail.mass_mailing'
|
|
|
|
state = fields.Selection(selection_add=[
|
|
('sending', "Sending"),
|
|
])
|
|
pending_count = fields.Integer(
|
|
string="Pending", compute='_compute_pending_count')
|
|
mass_mailing_sending_ids = fields.One2many(
|
|
comodel_name='mail.mass_mailing.sending', readonly=True,
|
|
inverse_name='mass_mailing_id', string="Sending tasks")
|
|
|
|
@api.model
|
|
def read_group(self, domain, fields, groupby, **kwargs):
|
|
# Add 'sending' state group, even if no results.
|
|
# This is needed for kanban view, columns are showed always
|
|
res = super(MailMassMailing, self).read_group(
|
|
domain, fields, groupby, **kwargs)
|
|
if groupby and groupby[0] == "state":
|
|
group_domain = domain + [('state', '=', 'sending')]
|
|
count = self.search_count(group_domain)
|
|
res.append({
|
|
'__context': {'group_by': groupby[1:]},
|
|
'__domain': group_domain,
|
|
'state': ('sending', _("Sending")),
|
|
'state_count': count,
|
|
})
|
|
return res
|
|
|
|
def _sendings_get(self):
|
|
self.ensure_one()
|
|
return self.env['mail.mass_mailing.sending'].search([
|
|
('mass_mailing_id', '=', self.id),
|
|
('state', 'in', ('enqueued', 'sending')),
|
|
])
|
|
|
|
@api.multi
|
|
def send_mail(self):
|
|
if not self.env.context.get('sending_queue_enabled', False):
|
|
return super(MailMassMailing, self).send_mail()
|
|
for mailing in self:
|
|
m_sending = self.env['mail.mass_mailing.sending']
|
|
sendings = mailing._sendings_get()
|
|
if sendings:
|
|
raise UserError(_(
|
|
"There is another sending task running. "
|
|
"Please, be patient. You can see all the sending tasks in "
|
|
"'Sending tasks' tab"
|
|
))
|
|
res_ids = mailing.get_recipients(mailing)
|
|
batch_size = m_sending.batch_size_get()
|
|
if not res_ids:
|
|
raise UserError(_("Please select recipients."))
|
|
sending = m_sending.create({
|
|
'state': 'draft',
|
|
'mass_mailing_id': mailing.id,
|
|
})
|
|
sending_state = 'enqueued'
|
|
if len(res_ids) < (batch_size / 2):
|
|
mailing.with_context(
|
|
mass_mailing_sending_id=sending.id,
|
|
sending_queue_enabled=False).send_mail()
|
|
sending_state = 'sending'
|
|
sending.state = sending_state
|
|
mailing.write({
|
|
'sent_date': fields.Datetime.now(),
|
|
'state': 'sending',
|
|
})
|
|
return True
|
|
|
|
@api.model
|
|
def get_recipients(self, mailing):
|
|
sending = False
|
|
sending_id = self.env.context.get('mass_mailing_sending_id', False)
|
|
if sending_id:
|
|
sending = self.env['mail.mass_mailing.sending'].browse(sending_id)
|
|
try:
|
|
res_ids = super(MailMassMailing, self).get_recipients(mailing)
|
|
except UserError as e:
|
|
if sending:
|
|
sending._send_error(e)
|
|
else:
|
|
raise
|
|
return []
|
|
if sending:
|
|
res_ids = sending.get_recipient_batch(res_ids)
|
|
return res_ids
|
|
|
|
@api.multi
|
|
def _compute_pending_count(self):
|
|
for mailing in self:
|
|
sendings = mailing._sendings_get()
|
|
mailing.pending_count = (
|
|
sum(sendings.mapped('pending_count')) +
|
|
sum(sendings.mapped('sending_count')))
|