addons/website_crm_partner_assign/controllers/main.py

   1 # -*- coding: utf-8 -*-
   2
   3 import werkzeug
   4
   5 import openerp
   6 from openerp import SUPERUSER_ID
   7 from openerp.addons.web import http
   8 from openerp.tools.translate import _
   9 from openerp.addons.web.http import request
  10 from openerp.addons.website_partner.controllers import main as website_partner
  11
  12 class WebsiteCrmPartnerAssign(http.Controller):
  13     _references_per_page = 20
  14
  15     @http.route([
  16         '/partners',
  17         '/partners/page/<int:page>',
  18
  19         '/partners/grade/<int:grade_id>',
  20         '/partners/grade/<int:grade_id>/page/<int:page>',
  21
  22         '/partners/country/<int:country_id>',
  23         '/partners/country/<country_name>-<int:country_id>',
  24         '/partners/country/<int:country_id>/page/<int:page>',
  25         '/partners/country/<country_name>-<int:country_id>/page/<int:page>',
  26
  27         '/partners/grade/<int:grade_id>/country/<int:country_id>',
  28         '/partners/grade/<int:grade_id>/country/<country_name>-<int:country_id>',
  29         '/partners/grade/<int:grade_id>/country/<int:country_id>/page/<int:page>',
  30         '/partners/grade/<int:grade_id>/country/<country_name>-<int:country_id>/page/<int:page>',
  31
  32     ], type='http', auth="public", website=True, multilang=True)
  33     def partners(self, country_id=0, grade_id=0, page=0, **post):
  34         country_obj = request.registry['res.country']
  35         partner_obj = request.registry['res.partner']
  36         post_name = post.get('search', '')
  37         country = None
  38
  39         # format displayed membership lines domain
  40         base_partner_domain = [('is_company', '=', True), ('grade_id', '!=', False), ('website_published', '=', True)]
  41         partner_domain = list(base_partner_domain)
  42         if post_name:
  43             partner_domain += ['|', ('name', 'ilike', post_name), ('website_description', 'ilike', post_name)]
  44         if grade_id and grade_id != "all":
  45             partner_domain += [('grade_id', '=', int(grade_id))]  # try/catch int
  46
  47         # group by country
  48         countries = partner_obj.read_group(
  49             request.cr, openerp.SUPERUSER_ID, partner_domain, ["id", "country_id"],
  50             groupby="country_id", orderby="country_id", context=request.context)
  51         countries_partners = partner_obj.search(
  52             request.cr, openerp.SUPERUSER_ID, partner_domain,
  53             context=request.context, count=True)
  54
  55         if country_id:
  56             country = country_obj.browse(request.cr, request.uid, country_id, request.context)
  57             partner_domain += [('country_id', '=', country_id)]
  58             if not any(x['country_id'][0] == country_id for x in countries):
  59                 countries.append({
  60                     'country_id_count': 0,
  61                     'country_id': (country_id, country.name)
  62                 })
  63                 countries.sort(key=lambda d: d['country_id'][1])
  64
  65         countries.insert(0, {
  66             'country_id_count': countries_partners,
  67             'country_id': (0, _("All Countries"))
  68         })
  69
  70
  71         # format pager
  72         partner_ids = partner_obj.search(
  73             request.cr, openerp.SUPERUSER_ID, partner_domain,
  74             context=request.context)
  75         pager = request.website.pager(url="/partners", total=len(partner_ids), page=page, step=self._references_per_page, scope=7, url_args=post)
  76
  77         # search for partners to display
  78         partners_data = partner_obj.search_read(request.cr, openerp.SUPERUSER_ID,
  79             domain=partner_domain,
  80             fields=request.website.get_partner_white_list_fields(),
  81             offset=pager['offset'],
  82             limit=self._references_per_page,
  83             order="grade_id DESC,partner_weight DESC",
  84             context=request.context)
  85         google_map_partner_ids = ",".join([str(p['id']) for p in partners_data])
  86
  87         # group by grade
  88         grades = partner_obj.read_group(
  89             request.cr, openerp.SUPERUSER_ID, base_partner_domain, ["id", "grade_id"],
  90             groupby="grade_id", orderby="grade_id", context=request.context)
  91         grades_partners = partner_obj.search(
  92             request.cr, openerp.SUPERUSER_ID, base_partner_domain,
  93             context=request.context, count=True)
  94         grades.insert(0, {
  95             'grade_id_count': grades_partners,
  96             'grade_id': (0, _("All Categories"))
  97         })
  98
  99         values = {
 100             'countries': countries,
 101             'current_country_id': country_id,
 102             'current_country': country,
 103             'grades': grades,
 104             'grade_id': grade_id,
 105             'partners_data': partners_data,
 106             'google_map_partner_ids': google_map_partner_ids,
 107             'pager': pager,
 108             'searches': post,
 109             'search_path': "?%s" % werkzeug.url_encode(post),
 110         }
 111         return request.website.render("website_crm_partner_assign.index", values)
 112
 113     # Do not use semantic controller due to SUPERUSER_ID
 114     @http.route(['/partners/<int:partner_id>', '/partners/<partner_name>-<int:partner_id>'], type='http', auth="public", website=True, multilang=True)
 115     def partners_ref(self, partner_id, **post):
 116         partner = request.registry['res.partner'].browse(request.cr, SUPERUSER_ID, partner_id, context=request.context)
 117         values = website_partner.get_partner_template_value(partner)
 118         if not values:
 119             return self.partners(**post)
 120         values['main_object'] = values['partner']
 121         return request.website.render("website_crm_partner_assign.partner", values)