golem_mail_family/models/golem_family.py

@@ -0,0 +1,51 @@
+# -*- 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/>.
+
+""" GOLEM Family adaptations """
+
+from odoo import models, fields, api
+
+class GolemFamily(models.Model):
+    """ GOLEM Family adaptations """
+    _inherit = 'golem.family'
+
+    @api.multi
+    def get_mass_mailing_action(self, action_type):
+        """ Call family mass mailing presend wizard """
+        presend_wizard = self.env['golem.mail.presend.wizard'].create({})
+        if action_type == 'family':
+            emails = self.mapped('email')
+            for email in emails:
+                self.env['golem.mail.recipient'].create({'email': email,
+                                                         'presend_wizard_id': presend_wizard.id})
+                act_name = 'Family mass mailing'
+        elif action_type == 'members':
+            partner_ids = self.mapped('member_ids').ids
+            for partner in partner_ids:
+                self.env['golem.mail.recipient'].create({'partner_id': partner,
+                                                         'presend_wizard_id': presend_wizard.id})
+                act_name = 'Family members mass mailing'
+        return {
+            'type': 'ir.actions.act_window',
+            'name': act_name,
+            'res_model': 'golem.mail.presend.wizard',
+            'res_id': presend_wizard.id,
+            'view_mode': 'form',
+            'view_type': 'form',
+            'target': 'new'
+            }