initial commit 2

3 files changed, 38 insertions, 0 deletions

diff --git a/addons/website_membership/models/__init__.py b/addons/website_membership/models/__init__.py
new file mode 100644
index 00000000..6627227f
--- /dev/null
+++ b/addons/website_membership/models/__init__.py
@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import membership
+from . import website
diff --git a/addons/website_membership/models/membership.py b/addons/website_membership/models/membership.py
new file mode 100644
index 00000000..794c9732
--- /dev/null
+++ b/addons/website_membership/models/membership.py
@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import models
+
+class MembershipLine(models.Model):
+
+    _inherit = 'membership.membership_line'
+
+    def get_published_companies(self, limit=None):
+        if not self.ids:
+            return []
+        limit_clause = '' if limit is None else ' LIMIT %d' % limit
+        self.env.cr.execute("""
+            SELECT DISTINCT p.id
+            FROM res_partner p INNER JOIN membership_membership_line m
+            ON  p.id = m.partner
+            WHERE is_published AND is_company AND m.id IN %s """ + limit_clause, (tuple(self.ids),))
+        return [partner_id[0] for partner_id in self.env.cr.fetchall()]
diff --git a/addons/website_membership/models/website.py b/addons/website_membership/models/website.py
new file mode 100644
index 00000000..71540ac3
--- /dev/null
+++ b/addons/website_membership/models/website.py
@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import models, _
+from odoo.addons.http_routing.models.ir_http import url_for
+
+
+class Website(models.Model):
+    _inherit = "website"
+
+    def get_suggested_controllers(self):
+        suggested_controllers = super(Website, self).get_suggested_controllers()
+        suggested_controllers.append((_('Members'), url_for('/members'), 'website_membership'))
+        return suggested_controllers