# $Id: WSGIHandler.py 2133 2006-09-06 18:52:56Z dairiki $
# WSGIHandler.py - handles WSGI requests for Myghty
# Copyright (C) 2004, 2005 Michael Bayer mike_mp@zzzcomputing.com
#
# This module is part of Myghty and is released under
# the MIT License: http://www.opensource.org/licenses/mit-license.php
#
#
import myghty.interp
import myghty.request
import myghty.resolver
import myghty.buffer
import myghty.escapes as escapes
import myghty.session as session
import myghty.exception as exception
import myghty.http.HTTPHandler as HTTPHandler
import myghty.http.HTTPHandler as http
from myghty.util import *
import os, sys, cgi
def handle(environ, start_response, **params):
return HTTPHandler.handle_http(WSGIHandler, environ = environ, start_response = start_response, **params)
def get_handler(interpreter_name = None, **params):
return HTTPHandler.get_handler(WSGIHandler, interpreter_name = interpreter_name, **params)
def application(environ, start_response):
return get_handler(**environ.get('myghty.application', {})).handle(environ, start_response, **environ.get('myghty.request', {}))
class WSGIHandler(HTTPHandler.HTTPHandler):
def __init__(self, **params):
HTTPHandler.HTTPHandler.__init__(self, None, myghty.buffer.LinePrinter(sys.stderr), **params)
def do_get_init_params(self, httpreq, **params):
return params
def do_get_resolver(self, **params):
return WSGIResolver(**params)
def do_make_request_impl(self, httpreq, **params):
return WSGIRequestImpl(httpreq, **params)
def do_get_component(self, httpreq, **params):
return httpreq.path_info
def do_handle_result(self, httpreq, status_code, reason):
httpreq.status = status_code
httpreq.status_message = reason
httpreq.send_http_header()
return [httpreq.out_buffer.getvalue()]
def handle(self, environ, start_response, httpreq = None, interp = None, request_impl = None, component = None, **params):
if httpreq is None:
httpreq = WSGIRequest(environ, start_response)
return HTTPHandler.HTTPHandler.handle(self, httpreq, interp, request_impl, component, **params)
class WSGIRequest(HTTPHandler.HTTPRequest):
def __init__(self, environ, start_response):
HTTPHandler.HTTPRequest.__init__(self)
self.start_response = start_response
self.environ = environ
for key in environ.keys():
if key[0:4] == 'HTTP':
self.headers_in.add(key[5:], environ[key])
self.content_type = 'text/html'
self.method = environ['REQUEST_METHOD']
self.path_info = environ.get('PATH_INFO', '/')
self.args = environ.get('QUERY_STRING', None)
self.status_message = None
self.filename = environ.get('PATH_TRANSLATED', None)
self.out_buffer = StringIO()
self.fieldstorage = cgi.FieldStorage(
fp = environ['wsgi.input'],
environ = environ,
keep_blank_values = True
)
def do_send_headers(self):
headers = [(key, value) for key, value in self.get_response_headers().iteritems()]
if self.status_message is None:
reason = "HTTP return code"
else:
reason = self.status_message
self.start_response(str(self.status) + " " + reason, headers)
class WSGIResolver(HTTPHandler.HTTPResolver): pass
class WSGIWriter(HTTPHandler.HTTPWriter):pass
class WSGIRequestImpl(HTTPHandler.HTTPRequestImpl):
def __init__(self, httpreq, **params):
logger = myghty.buffer.LinePrinter(httpreq.environ['wsgi.errors'])
HTTPHandler.HTTPRequestImpl.__init__(self, httpreq, logger, logger, **params)
def do_make_request_args(self, httpreq, **params):
return self.request_args_from_fieldstorage(httpreq.fieldstorage)
def do_get_out_buffer(self, httpreq, out_buffer = None, **params):
if out_buffer is None:
return WSGIWriter(httpreq, httpreq.out_buffer)
else:
return WSGIWriter(httpreq, out_buffer)
|