web.py

#!/usr/bin/python
# coding:utf8

import os
import sys
import time
import traceback
from quixote.qwip import QWIP
from quixote.publish import Publisher

from vilya import views as controllers
from vilya.config import DEVELOP_MODE
from vilya.libs.gzipper import make_gzip_middleware
from vilya.libs.permdir import get_tmpdir
from vilya.libs.auth.check_auth import check_auth
from vilya.libs.import_obj import import_obj_set
from vilya.libs.template import st
from vilya.models.user import User
from vilya.views.util import is_mobile_device


PERFORMANCE_METRIC_MARKER = '<!-- _performtips_ -->'


def show_performance_metric(request, output):
    idx = output.find(PERFORMANCE_METRIC_MARKER)
    if idx > 0:
        pt = int((time.time() - request.start_time) * 1000)
        cls = pt > 250 and 'red' or pt > 100 and 'orange' or 'green'
        block = '<li class="hidden-phone"><span style="color:%s"> %d ms </span></li>' % (cls, pt)
        output = (output[:idx] + block + output[idx + len(PERFORMANCE_METRIC_MARKER):])
    return output


class CODEPublisher(Publisher):

    def __init__(self, *args, **kwargs):
        Publisher.__init__(self, *args, **kwargs)
        self.configure(DISPLAY_EXCEPTIONS='plain',
                       SECURE_ERRORS=0,
                       UPLOAD_DIR=get_tmpdir() + '/upload/')

    def start_request(self, request):
        Publisher.start_request(self, request)
        os.environ['SQLSTORE_SOURCE'] = request.get_url()

        resp = request.response
        resp.set_content_type('text/html; charset=utf-8')
        resp.set_header('Pragma', 'no-cache')
        resp.set_header('Cache-Control', 'must-revalidate, no-cache, private')
        # FIXME: quixote with origin?
        resp.set_header('Access-Control-Allow-Origin', '*')
        request.enable_ajax = False
        request.browser = request.guess_browser_version()
        request.method = request.get_method()
        request.url = request.get_path()
        request.is_mobile = is_mobile_device(request)
        request.start_time = time.time()
        request.user = User.check_session(request)

        import_obj_set("request", request)

    def try_publish(self, request, path):
        output = Publisher.try_publish(self, request, path)
        output = show_performance_metric(request, output)
        return output

    def finish_failed_request(self, request):
        if DEVELOP_MODE:
            exc_type, exc_value, tb = sys.exc_info()
            raise exc_type, exc_value, tb
        else:
            return Publisher.finish_failed_request(self, request)

    def _generate_cgitb_error(self, request, original_response,
                              exc_type, exc_value, tb):
        traceback.print_exc()
        return st('/errors/500.html', **locals())


def create_publisher():
    return CODEPublisher(controllers)


app = make_gzip_middleware(QWIP(create_publisher()))