odoo-extras/oerp_develop_test_mode/controllers/main.py

# -*- coding: utf-8 -*-
#/#############################################################################
#
#    BizzAppDev
#    Copyright (C) 2015-TODAY bizzappdev(<http://www.bizzappdev.com>).
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 of the
#    License, or (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
#/#############################################################################

import os
import re
import urllib
import urlparse

import openerp
from openerp.addons.web import http
from openerp.addons.web.controllers.main import Binary
from openerp.http import request, route
from openerp.tools import config

openerpweb = http


def db_monodb(req):
    # if only one db exists, return it else return False
    return db_redirect(req, True)[0]


def db_redirect(req, match_first_only_if_unique):
    db = False
    redirect = False

    # 1 try the db in the url
    db_url = req.params.get('db')
    if db_url:
        return (db_url, False)

    dbs = db_list(req, True)

    # 2 use the database from the cookie if it's listable and still listed
    cookie_db = req.httprequest.cookies.get('last_used_database')
    if cookie_db in dbs:
        db = cookie_db

    # 3 use the first db if user can list databases
    if dbs and not db and (not match_first_only_if_unique or len(dbs) == 1):
        db = dbs[0]

    # redirect to the chosen db if multiple are available
    if db and len(dbs) > 1:
        query = dict(urlparse.parse_qsl(
            req.httprequest.query_string, keep_blank_values=True))
        query.update({'db': db})
        redirect = req.httprequest.path + '?' + urllib.urlencode(query)
    return (db, redirect)


def db_list(req, force=False):
    proxy = req.session.proxy("db")
    dbs = proxy.list(force)
    h = req.httprequest.environ['HTTP_HOST'].split(':')[0]
    d = h.split('.')[0]
    r = openerp.tools.config['dbfilter'].replace('%h', h).replace('%d', d)
    dbs = [i for i in dbs if re.match(r, i)]
    return dbs


class mode(openerpweb.Controller):
    _cp_path = "/web/mode"

    @openerpweb.jsonrequest
    def get_mode(self, req, mode=False, db=False):
        if not db:
            dbs = db_list(req)
            db = dbs and dbs[0] or False
        return config.get(mode, {}).get(db, False)


class cp_Binary(Binary):
    _cp_path = "/web/binary"

    @http.route([
                '/web/binary/company_logo',
                '/logo',
                '/logo.png',
                ], type='http', auth="none")
    def company_logo(self, dbname=None, **kw):
        req = request
        if req.session._db:
            dbname = req.session._db
            uid = req.session._uid
        elif dbname is None:
            dbname = db_monodb(req)

        mode = config.get('develop', {}).get(
            dbname, False) and 'develop' or False
        mode = not mode and config.get('test', {}).get(
            dbname, False) and 'test' or mode
        if not mode:
            return super(cp_Binary, self).company_logo(dbname=dbname, **kw)

        # TODO add etag, refactor to use /image code for etag
        module_path = list(os.path.split(__file__)[
                           :-1]) + ["%s_logo.png" % mode]
        image = open(os.path.join(*module_path), 'r')
        image_data = image.read()
        image.close()
        headers = [
            ('Content-Type', 'image/png'),
            ('Content-Length', len(image_data)),
        ]
        return req.make_response(image_data, headers)


class SwitchController(http.Controller):
    @route('/easy_switch_user/switch', type='json', auth="none")
    def switch(self, login, password):
        request = http.request
        uid = request.session.authenticate(request.db, login, password)
        if uid is False:
            raise Exception('Login Failed')

# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: