addons/website_membership/controllers/main.py

   1 # -*- coding: utf-8 -*-
   2
   3 import openerp
   4 from openerp import SUPERUSER_ID
   5 from openerp.addons.web import http
   6 from openerp.addons.web.http import request
   7 from openerp.addons.website_partner.controllers import main as website_partner
   8 from openerp.tools.translate import _
   9
  10 import werkzeug.urls
  11
  12
  13 class WebsiteMembership(http.Controller):
  14     _references_per_page = 20
  15
  16     @http.route([
  17         '/members/',
  18         '/members/page/<int:page>/',
  19         '/members/association/<int:membership_id>/',
  20         '/members/association/<int:membership_id>/page/<int:page>/',
  21
  22         '/members/country/<int:country_id>',
  23         '/members/country/<country_name>-<int:country_id>',
  24         '/members/country/<int:country_id>/page/<int:page>/',
  25         '/members/country/<country_name>-<int:country_id>/page/<int:page>/',
  26
  27         '/members/association/<int:membership_id>/country/<country_name>-<int:country_id>',
  28         '/members/association/<int:membership_id>/country/<int:country_id>',
  29         '/members/association/<int:membership_id>/country/<country_name>-<int:country_id>/page/<int:page>/',
  30         '/members/association/<int:membership_id>/country/<int:country_id>/page/<int:page>/',
  31     ], type='http', auth="public", website=True, multilang=True)
  32     def members(self, membership_id=None, country_name=None, country_id=0, page=0, **post):
  33         cr, uid, context = request.cr, request.uid, request.context
  34         product_obj = request.registry['product.product']
  35         country_obj = request.registry['res.country']
  36         membership_line_obj = request.registry['membership.membership_line']
  37         partner_obj = request.registry['res.partner']
  38         post_name = post.get('name', '')
  39         current_country = None
  40
  41         # base domain for groupby / searches
  42         base_line_domain = [("partner.website_published", "=", True),('state', 'in', ['free', 'paid'])]
  43         if membership_id:
  44             base_line_domain.append(('membership_id', '=', membership_id))
  45             membership = product_obj.browse(cr, uid, membership_id, context=context)
  46         else:
  47             membership = None
  48         if post_name:
  49             base_line_domain += ['|', ('partner.name', 'ilike', post_name),
  50                                       ('partner.website_description', 'ilike', post_name)]
  51
  52         # group by country, based on all customers (base domain)
  53         membership_line_ids = membership_line_obj.search(cr, uid, base_line_domain, context=context)
  54         countries = partner_obj.read_group(
  55             cr, uid, [('member_lines', 'in', membership_line_ids), ("website_published", "=", True)], ["id", "country_id"],
  56             groupby="country_id", orderby="country_id", context=request.context)
  57         countries_total = sum(country_dict['country_id_count'] for country_dict in countries)
  58
  59         line_domain = list(base_line_domain)
  60         if country_id:
  61             line_domain.append(('partner.country_id', '=', country_id))
  62             current_country = country_obj.read(cr, uid, country_id, ['id', 'name'], context)
  63             if not any(x['country_id'][0] == country_id for x in countries):
  64                 countries.append({
  65                     'country_id_count': 0,
  66                     'country_id': (country_id, current_country["name"])
  67                 })
  68                 countries.sort(key=lambda d: d['country_id'][1])
  69
  70         countries.insert(0, {
  71             'country_id_count': countries_total,
  72             'country_id': (0, _("All Countries"))
  73         })
  74
  75         # displayed membership lines
  76         membership_line_ids = membership_line_obj.search(cr, uid, line_domain, context=context)
  77         membership_lines = membership_line_obj.browse(cr, uid, membership_line_ids, context=context)
  78         membership_lines.sort(key=lambda x: x.membership_id.website_sequence)
  79         partner_ids = [m.partner and m.partner.id for m in membership_lines]
  80         google_map_partner_ids = ",".join(map(str, partner_ids))
  81
  82         partners_data = {}
  83         for partner in partner_obj.read(cr, openerp.SUPERUSER_ID, partner_ids, request.website.get_partner_white_list_fields(), context=context):
  84             partners_data[partner.get("id")] = partner
  85
  86         # format domain for group_by and memberships
  87         membership_ids = product_obj.search(cr, uid, [('membership', '=', True)], order="website_sequence", context=context)
  88         memberships = product_obj.browse(cr, uid, membership_ids, context=context)
  89
  90         # request pager for lines
  91         pager = request.website.pager(url="/members/", total=len(membership_line_ids), page=page, step=self._references_per_page, scope=7, url_args=post)
  92
  93         values = {
  94             'partners_data': partners_data,
  95             'membership_lines': membership_lines,
  96             'memberships': memberships,
  97             'membership': membership,
  98             'countries': countries,
  99             'current_country': current_country and [current_country['id'], current_country['name']] or None,
 100             'current_country_id': current_country and current_country['id'] or 0,
 101             'google_map_partner_ids': google_map_partner_ids,
 102             'pager': pager,
 103             'post': post,
 104             'search': "?%s" % werkzeug.url_encode(post),
 105         }
 106         return request.website.render("website_membership.index", values)
 107
 108     @http.route(['/members/<int:partner_id>/', '/members/<partner_name>-<int:partner_id>/'], type='http', auth="public", website=True, multilang=True)
 109     def partners_ref(self, partner_id, **post):
 110         partner = request.registry['res.partner'].browse(request.cr, SUPERUSER_ID, partner_id, context=request.context)
 111         values = website_partner.get_partner_template_value(partner)
 112         if not values:
 113             return self.members(**post)
 114         values['main_object'] = values['partner']
 115         return request.website.render("website_membership.partner", values)