addons/website_crm/controllers/main.py

   1 # -*- coding: utf-8 -*-
   2 import base64
   3
   4 import werkzeug
   5 import werkzeug.urls
   6
   7 from openerp import http, SUPERUSER_ID
   8 from openerp.http import request
   9 from openerp.tools.translate import _
  10
  11 class contactus(http.Controller):
  12
  13     def generate_google_map_url(self, street, city, city_zip, country_name):
  14         url = "http://maps.googleapis.com/maps/api/staticmap?center=%s&sensor=false&zoom=8&size=298x298" % werkzeug.url_quote_plus(
  15             '%s, %s %s, %s' % (street, city, city_zip, country_name)
  16         )
  17         return url
  18
  19     @http.route(['/page/website.contactus', '/page/contactus'], type='http', auth="public", website=True)
  20     def contact(self, **kwargs):
  21         values = {}
  22         for field in ['description', 'partner_name', 'phone', 'contact_name', 'email_from', 'name']:
  23             if kwargs.get(field):
  24                 values[field] = kwargs.pop(field)
  25         values.update(kwargs=kwargs.items())
  26         return request.website.render("website.contactus", values)
  27
  28     def create_lead(self, request, values, kwargs):
  29         """ Allow to be overrided """
  30         return request.registry['crm.lead'].create(request.cr, SUPERUSER_ID, values, request.context)
  31
  32     def preRenderThanks(self, request, values, kwargs):
  33         """ Allow to be overrided """
  34         company = request.website.company_id
  35         return {
  36             'google_map_url': self.generate_google_map_url(company.street, company.city, company.zip, company.country_id and company.country_id.name_get()[0][1] or ''),
  37             '_values': values,
  38             '_kwargs': kwargs,
  39         }
  40
  41     @http.route(['/crm/contactus'], type='http', auth="public", website=True)
  42     def contactus(self, **kwargs):
  43         def dict_to_str(title, dictvar):
  44             ret = "\n\n%s" % title
  45             for field in dictvar:
  46                 ret += "\n%s" % field
  47             return ret
  48
  49         _TECHNICAL = ['show_info', 'view_from', 'view_callback']  # Only use for behavior, don't stock it
  50         _BLACKLIST = ['id', 'create_uid', 'create_date', 'write_uid', 'write_date', 'user_id', 'active']  # Allow in description
  51         _REQUIRED = ['name', 'contact_name', 'email_from', 'description']  # Could be improved including required from model
  52
  53         post_file = []  # List of file to add to ir_attachment once we have the ID
  54         post_description = []  # Info to add after the message
  55         values = {}
  56
  57         for field_name, field_value in kwargs.items():
  58             if hasattr(field_value, 'filename'):
  59                 post_file.append(field_value)
  60             elif field_name in request.registry['crm.lead']._all_columns and field_name not in _BLACKLIST:
  61                 values[field_name] = field_value
  62             elif field_name not in _TECHNICAL:  # allow to add some free fields or blacklisted field like ID
  63                 post_description.append("%s: %s" % (field_name, field_value))
  64
  65         if "name" not in kwargs and values.get("contact_name"):  # if kwarg.name is empty, it's an error, we cannot copy the contact_name
  66             values["name"] = values.get("contact_name")
  67         # fields validation : Check that required field from model crm_lead exists
  68         error = set(field for field in _REQUIRED if not values.get(field))
  69
  70         values = dict(values, error=error)
  71         if error:
  72             values.update(kwargs=kwargs.items())
  73             return request.website.render(kwargs.get("view_from", "website.contactus"), values)
  74
  75         try:
  76             values['medium_id'] = request.registry['ir.model.data'].get_object_reference(request.cr, SUPERUSER_ID, 'crm', 'crm_tracking_medium_website')[1]
  77             values['section_id'] = request.registry['ir.model.data'].xmlid_to_res_id(request.cr, SUPERUSER_ID, 'website.salesteam_website_sales')
  78         except ValueError:
  79             pass
  80
  81         # description is required, so it is always already initialized
  82         if post_description:
  83             values['description'] += dict_to_str(_("Custom Fields: "), post_description)
  84
  85         if kwargs.get("show_info"):
  86             post_description = []
  87             environ = request.httprequest.headers.environ
  88             post_description.append("%s: %s" % ("IP", environ.get("REMOTE_ADDR")))
  89             post_description.append("%s: %s" % ("USER_AGENT", environ.get("HTTP_USER_AGENT")))
  90             post_description.append("%s: %s" % ("ACCEPT_LANGUAGE", environ.get("HTTP_ACCEPT_LANGUAGE")))
  91             post_description.append("%s: %s" % ("REFERER", environ.get("HTTP_REFERER")))
  92             values['description'] += dict_to_str(_("Environ Fields: "), post_description)
  93
  94         lead_id = self.create_lead(request, dict(values, user_id=False), kwargs)
  95         if lead_id:
  96             for field_value in post_file:
  97                 attachment_value = {
  98                     'name': field_value.filename,
  99                     'res_name': field_value.filename,
 100                     'res_model': 'crm.lead',
 101                     'res_id': lead_id,
 102                     'datas': base64.encodestring(field_value.read()),
 103                     'datas_fname': field_value.filename,
 104                 }
 105                 request.registry['ir.attachment'].create(request.cr, SUPERUSER_ID, attachment_value, context=request.context)
 106
 107         values = self.preRenderThanks(request, values, kwargs)
 108         return request.website.render(kwargs.get("view_callback", "website_crm.contactus_thanks"), values)