Ticket #559: wsgi.2.py

# This should work just like Tool, copying also the middleware args to make
# them easily introspectable, all args configured for the registered middleware
# are passed as keyword arguments to __call__ to configure the middleware
# FIXME this should be moved inside CP like Tool
class Middleware(object):
    def __init__(self, middleware):
        self.middleware = middleware

    def __call__(self, **conf):
        return lambda app: self.middleware(app, **conf)

class ChangeCase(object):
    def __init__(self, app, to=None):
        self.app = app
        self.to = to

    def __call__(self, environ, start_response):
        response_iter = self.app(environ, start_response)
        response_string = ''.join(response_iter)
        if self.to == "upper":
            return [response_string.upper()]
        elif self.to == "lower":
            return [response_string.lower()]
        else:
            return [response_string]

import cherrypy

class Root(object):
    def index(self):
        dumb = 50
        return "HeLlO WoRlD!"
    index.exposed = True

global_conf = {
    'log.screen': True,
    'request.throw_errors': True,
}

# configure the middleware pipeline and our middleware
# TODO fix the namespace, wsgi? middleware? wsgi.middleware.*?
app_conf = {
    '/': {
        'wsgi.middleware_pipeline': ['changecase'],
        'wsgi.changecase.to': 'lower',
    }
}

# register our middleware like you do with a tool
cherrypy.wsgi.changecase = Middleware(ChangeCase)

cherrypy.config.update(global_conf)
cherrypy.quickstart(Root(), script_name='/', conf=app_conf)