[MIGR] mail: migration to the new API

1 files changed, 4 insertions, 2 deletions

diff --git a/addons/mass_mailing/wizard/mail_compose_message.py b/addons/mass_mailing/wizard/mail_compose_message.py
index 2990016abd0..de480f455c4 100644
--- a/addons/mass_mailing/wizard/mail_compose_message.py
+++ b/addons/mass_mailing/wizard/mail_compose_message.py
@@ -20,11 +20,13 @@ class MailComposeMessage(osv.TransientModel):
         ),
     }
 
-    def get_mail_values(self, cr, uid, wizard, res_ids, context=None):
+    def get_mail_values(self, cr, uid, ids, res_ids, context=None):
         """ Override method that generated the mail content by creating the
         mail.mail.statistics values in the o2m of mail_mail, when doing pure
         email mass mailing. """
-        res = super(MailComposeMessage, self).get_mail_values(cr, uid, wizard, res_ids, context=context)
+        res = super(MailComposeMessage, self).get_mail_values(cr, uid, ids, res_ids, context=context)
+        # TDE: arg was wiards, not ids - but new API -> multi with ensure_one
+        wizard = self.browse(cr, uid, ids[0], context=context)
         # use only for allowed models in mass mailing
         if wizard.composition_mode == 'mass_mail' and \
                 (wizard.mass_mailing_name or wizard.mass_mailing_id) and \