From 8d9f7f64a1fe08cc67e32ac09828e9104f1200d2 Mon Sep 17 00:00:00 2001
From: Youssef ELOUAHBY <youssef@yaltik.com>
Date: Thu, 22 Nov 2018 23:12:53 +0100
Subject: [PATCH] Add button to family form

---
 golem_mail_family/models/golem_family.py         |  8 ++++++--
 .../views/golem_mail_family_views.xml            | 16 ++++++++++++++++
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/golem_mail_family/models/golem_family.py b/golem_mail_family/models/golem_family.py
index 8627ebf..5a04188 100644
--- a/golem_mail_family/models/golem_family.py
+++ b/golem_mail_family/models/golem_family.py
@@ -24,6 +24,10 @@ class GolemFamily(models.Model):
     """ GOLEM Family adaptations """
     _inherit = 'golem.family'
 
+    @api.multi
+    def family_members_mass_mailing(self):
+        return self.get_mass_mailing_action('members')
+
     @api.multi
     def get_mass_mailing_action(self, action_type):
         """ Call family mass mailing presend wizard """
@@ -33,13 +37,13 @@ class GolemFamily(models.Model):
             for email in emails:
                 self.env['golem.mail.recipient'].create({'email': email,
                                                          'presend_wizard_id': presend_wizard.id})
-                act_name = 'Family mass mailing'
+            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'
+            act_name = 'Family members mass mailing'
         return {
             'type': 'ir.actions.act_window',
             'name': act_name,
diff --git a/golem_mail_family/views/golem_mail_family_views.xml b/golem_mail_family/views/golem_mail_family_views.xml
index 89d9bcc..811fa91 100644
--- a/golem_mail_family/views/golem_mail_family_views.xml
+++ b/golem_mail_family/views/golem_mail_family_views.xml
@@ -18,6 +18,22 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 -->
 <odoo>
   <data>
+    <!-- golem family form adaptations -->
+    <record id="golem_family_form_inherit_golem_mail_family"
+      model="ir.ui.view">
+      <field name="name">Golem Family form adaptations golem mail family</field>
+      <field name="model">golem.family</field>
+      <field name="inherit_id" ref="golem_family.golem_family_form"/>
+      <field name="arch" type="xml">
+        <sheet position="before">
+          <header>
+            <button name="family_members_mass_mailing" string="Family members mass mailing"
+              type="object" class="oe_highlight"
+              attrs="{'invisible': [('member_ids', '=', False)]}"/>
+          </header>
+        </sheet>
+      </field>
+    </record>
     <!-- Family member mass mailing action-->
     <record id="action_family_members_mass_mailing" model="ir.actions.server">
       <field name="name">Family members mass mailing</field>