diff options
author | Allan Saddi <allan@saddi.com> | 2007-06-05 09:42:39 -0700 |
---|---|---|
committer | Allan Saddi <allan@saddi.com> | 2007-06-05 09:42:39 -0700 |
commit | cbe792618a1e6b1aeddd8d751dbeefe6a981ba23 (patch) | |
tree | 803d38bc8a5f7c10b86efd36742a925ad5a6960b /flup/server/cgi.py | |
parent | ac40930383daac77cb44bd75b2a899fa2934a81e (diff) | |
download | flup-cbe792618a1e6b1aeddd8d751dbeefe6a981ba23.tar.gz |
Remove publisher and middleware packages. Add cgi server for completeness.1.0
Diffstat (limited to 'flup/server/cgi.py')
-rw-r--r-- | flup/server/cgi.py | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/flup/server/cgi.py b/flup/server/cgi.py new file mode 100644 index 0000000..17cc3ca --- /dev/null +++ b/flup/server/cgi.py @@ -0,0 +1,71 @@ +# Taken from <http://www.python.org/dev/peps/pep-0333/> +# which was placed in the public domain. + +import os, sys + + +__all__ = ['WSGIServer'] + + +class WSGIServer(object): + + def __init__(self, application): + self.application = application + + def run(self): + + environ = dict(os.environ.items()) + environ['wsgi.input'] = sys.stdin + environ['wsgi.errors'] = sys.stderr + environ['wsgi.version'] = (1,0) + environ['wsgi.multithread'] = False + environ['wsgi.multiprocess'] = True + environ['wsgi.run_once'] = True + + if environ.get('HTTPS','off') in ('on','1'): + environ['wsgi.url_scheme'] = 'https' + else: + environ['wsgi.url_scheme'] = 'http' + + headers_set = [] + headers_sent = [] + + def write(data): + if not headers_set: + raise AssertionError("write() before start_response()") + + elif not headers_sent: + # Before the first output, send the stored headers + status, response_headers = headers_sent[:] = headers_set + sys.stdout.write('Status: %s\r\n' % status) + for header in response_headers: + sys.stdout.write('%s: %s\r\n' % header) + sys.stdout.write('\r\n') + + sys.stdout.write(data) + sys.stdout.flush() + + def start_response(status,response_headers,exc_info=None): + if exc_info: + try: + if headers_sent: + # Re-raise original exception if headers sent + raise exc_info[0], exc_info[1], exc_info[2] + finally: + exc_info = None # avoid dangling circular ref + elif headers_set: + raise AssertionError("Headers already set!") + + headers_set[:] = [status,response_headers] + return write + + result = self.application(environ, start_response) + try: + for data in result: + if data: # don't send headers until body appears + write(data) + if not headers_sent: + write('') # send headers now if body was empty + finally: + if hasattr(result,'close'): + result.close() |