From 9206847d7017346e9ad032703879c3b8b8850400 Mon Sep 17 00:00:00 2001
From: Kaushal Prajapati <kaushal@techreceptives.com>
Date: Mon, 25 Dec 2017 11:09:48 +0530
Subject: [PATCH] [IMP] 'website_partner' compatible with Multi-Website

---
 addons/website_partner/data/website_partner_demo.xml | 11 +++++++++++
 addons/website_partner/models/res_partner.py         |  5 +++++
 addons/website_partner/views/res_partner_views.xml   |  3 +++
 3 files changed, 19 insertions(+)

diff --git a/addons/website_partner/data/website_partner_demo.xml b/addons/website_partner/data/website_partner_demo.xml
index ae529dfd..66f6d897 100644
--- a/addons/website_partner/data/website_partner_demo.xml
+++ b/addons/website_partner/data/website_partner_demo.xml
@@ -3,45 +3,56 @@
 
     <record id="base.partner_root" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
     <record id="base.partner_demo" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_12" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_12" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_2" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_1" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_12" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_3" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_3" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_2" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_18" model="res.partner">
         <field name="website_published">True</field>
+        <field name="website_ids" eval="[(4,ref('website.default_website'))]"/>
     </record>
 
     <record id="base.res_partner_10" model="res.partner">
diff --git a/addons/website_partner/models/res_partner.py b/addons/website_partner/models/res_partner.py
index bfbd5988..261b3ab6 100644
--- a/addons/website_partner/models/res_partner.py
+++ b/addons/website_partner/models/res_partner.py
@@ -10,6 +10,11 @@ class WebsiteResPartner(models.Model):
 
     website_description = fields.Html('Website Partner Full Description', strip_style=True)
     website_short_description = fields.Text('Website Partner Short Description')
+    website_ids = fields.Many2many('website', 'website_partner_pub_rel',
+                                   'website_id', 'partner_id',
+                                   string='Websites', copy=False,
+                                   help='List of websites in which '
+                                        'Partner is published.')
 
     @api.multi
     def _compute_website_url(self):
diff --git a/addons/website_partner/views/res_partner_views.xml b/addons/website_partner/views/res_partner_views.xml
index fc7e1bf7..778e8080 100644
--- a/addons/website_partner/views/res_partner_views.xml
+++ b/addons/website_partner/views/res_partner_views.xml
@@ -13,6 +13,9 @@
                         <field name="website_published" widget="website_button"/>
                     </button>
                 </button>
+                <xpath expr="//field[@name='lang']" position="after">
+                    <field name="website_ids" widget="many2many_tags"/>
+                </xpath>
             </data>
         </field>
     </record>