1 files changed, 19 insertions, 0 deletions
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()]