addons/website_crm/controllers/main.py

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