initial commit 2

1 files changed, 34 insertions, 0 deletions

diff --git a/addons/website_customer/models/res_partner.py b/addons/website_customer/models/res_partner.py
new file mode 100644
index 00000000..5ad57bc9
--- /dev/null
+++ b/addons/website_customer/models/res_partner.py
@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models
+
+
+class Partner(models.Model):
+
+    _inherit = 'res.partner'
+
+    website_tag_ids = fields.Many2many('res.partner.tag', 'res_partner_res_partner_tag_rel', 'partner_id', 'tag_id', string='Website tags')
+
+    def get_backend_menu_id(self):
+        return self.env.ref('contacts.menu_contacts').id
+
+
+class Tags(models.Model):
+
+    _name = 'res.partner.tag'
+    _description = 'Partner Tags - These tags can be used on website to find customers by sector, or ...'
+    _inherit = 'website.published.mixin'
+
+    @api.model
+    def get_selection_class(self):
+        classname = ['default', 'primary', 'success', 'warning', 'danger']
+        return [(x, str.title(x)) for x in classname]
+
+    name = fields.Char('Category Name', required=True, translate=True)
+    partner_ids = fields.Many2many('res.partner', 'res_partner_res_partner_tag_rel', 'tag_id', 'partner_id', string='Partners')
+    classname = fields.Selection(get_selection_class, 'Class', default='default', help="Bootstrap class to customize the color", required=True)
+    active = fields.Boolean('Active', default=True)
+
+    def _default_is_published(self):
+        return True