addons/website_membership/controllers/main.py

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