openerp/addons/base/ir/ir_http.py

   1 #----------------------------------------------------------
   2 # ir_http modular http routing
   3 #----------------------------------------------------------
   4 import logging
   5 import re
   6 import sys
   7
   8 import werkzeug.exceptions
   9 import werkzeug.routing
  10
  11 import openerp
  12 from openerp import http
  13 from openerp.http import request
  14 from openerp.osv import osv, orm
  15
  16 _logger = logging.getLogger(__name__)
  17
  18
  19 # FIXME: replace by proxy on request.uid?
  20 _uid = object()
  21
  22 class ModelConverter(werkzeug.routing.BaseConverter):
  23
  24     def __init__(self, url_map, model=False):
  25         super(ModelConverter, self).__init__(url_map)
  26         self.model = model
  27         self.regex = '([0-9]+)'
  28
  29     def to_python(self, value):
  30         m = re.match(self.regex, value)
  31         return request.registry[self.model].browse(
  32             request.cr, _uid, int(m.group(1)), context=request.context)
  33
  34     def to_url(self, value):
  35         return value.id
  36
  37 class ModelsConverter(werkzeug.routing.BaseConverter):
  38
  39     def __init__(self, url_map, model=False):
  40         super(ModelsConverter, self).__init__(url_map)
  41         self.model = model
  42         # TODO add support for slug in the form [A-Za-z0-9-] bla-bla-89 -> id 89
  43         self.regex = '([0-9,]+)'
  44
  45     def to_python(self, value):
  46         # TODO:
  47         # - raise routing.ValidationError() if no browse record can be createdm
  48         # - support slug
  49         return request.registry[self.model].browse(request.cr, _uid, [int(i) for i in value.split(',')], context=request.context)
  50
  51     def to_url(self, value):
  52         return ",".join(i.id for i in value)
  53
  54 class ir_http(osv.AbstractModel):
  55     _name = 'ir.http'
  56     _description = "HTTP routing"
  57
  58     def _get_converters(self):
  59         return {'model': ModelConverter, 'models': ModelsConverter}
  60
  61     def _find_handler(self):
  62         return self.routing_map().bind_to_environ(request.httprequest.environ).match()
  63
  64     def _auth_method_user(self):
  65         request.uid = request.session.uid
  66         if not request.uid:
  67             raise http.SessionExpiredException("Session expired")
  68
  69     def _auth_method_admin(self):
  70         if not request.db:
  71             raise http.SessionExpiredException("No valid database for request %s" % request.httprequest)
  72         request.uid = openerp.SUPERUSER_ID
  73
  74     def _auth_method_none(self):
  75         request.disable_db = True
  76         request.uid = None
  77
  78     def _authenticate(self, auth_method='user'):
  79         if request.session.uid:
  80             try:
  81                 request.session.check_security()
  82                 # what if error in security.check()
  83                 #   -> res_users.check()
  84                 #   -> res_users.check_credentials()
  85             except Exception:
  86                 request.session.logout()
  87         getattr(self, "_auth_method_%s" % auth_method)()
  88         return auth_method
  89
  90     def _handle_exception(self, exception):
  91         raise exception
  92
  93     def _dispatch(self):
  94         # locate the controller method
  95         try:
  96             func, arguments = self._find_handler()
  97         except werkzeug.exceptions.NotFound, e:
  98             return self._handle_exception(e)
  99
 100         # check authentication level
 101         try:
 102             auth_method = self._authenticate(getattr(func, "auth", None))
 103         except Exception:
 104             # force a Forbidden exception with the original traceback
 105             return self._handle_exception(
 106                 convert_exception_to(
 107                     werkzeug.exceptions.Forbidden))
 108
 109         # post process arg to set uid on browse records
 110         for arg in arguments.itervalues():
 111             if isinstance(arg, orm.browse_record) and arg._uid is _uid:
 112                 arg._uid = request.uid
 113
 114         # set and execute handler
 115         try:
 116             request.set_handler(func, arguments, auth_method)
 117             result = request.dispatch()
 118             if isinstance(result, Exception):
 119                 raise result
 120         except Exception, e:
 121             return self._handle_exception(e)
 122
 123         return result
 124
 125     def routing_map(self):
 126         if not hasattr(self, '_routing_map'):
 127             _logger.info("Generating routing map")
 128             cr = request.cr
 129             m = request.registry.get('ir.module.module')
 130             ids = m.search(cr, openerp.SUPERUSER_ID, [('state', '=', 'installed'), ('name', '!=', 'web')], context=request.context)
 131             installed = set(x['name'] for x in m.read(cr, 1, ids, ['name'], context=request.context))
 132             mods = ['', "web"] + sorted(installed)
 133             self._routing_map = http.routing_map(mods, False, converters=self._get_converters())
 134
 135         return self._routing_map
 136
 137 def convert_exception_to(to_type, with_message=False):
 138     """ Should only be called from an exception handler. Fetches the current
 139     exception data from sys.exc_info() and creates a new exception of type
 140     ``to_type`` with the original traceback.
 141
 142     If ``with_message`` is ``True``, sets the new exception's message to be
 143     the stringification of the original exception. If ``False``, does not
 144     set the new exception's message. Otherwise, uses ``with_message`` as the
 145     new exception's message.
 146
 147     :type with_message: str|bool
 148     """
 149     etype, original, tb = sys.exc_info()
 150     try:
 151         if with_message is False:
 152             message = None
 153         elif with_message is True:
 154             message = str(original)
 155         else:
 156             message = str(with_message)
 157
 158         raise to_type, message, tb
 159     except to_type, e:
 160         return e
 161
 162 # vim:et: