wsgicgi.py
90 lines
| 2.8 KiB
| text/x-python
|
PythonLexer
Eric Hopper
|
r2506 | # hgweb/wsgicgi.py - CGI->WSGI translator | ||
# | ||||
# Copyright 2006 Eric Hopper <hopper@omnifarious.org> | ||||
# | ||||
Martin Geisler
|
r8225 | # This software may be used and distributed according to the terms of the | ||
Matt Mackall
|
r10263 | # GNU General Public License version 2 or any later version. | ||
Eric Hopper
|
r2506 | # | ||
# This was originally copied from the public domain code at | ||||
# http://www.python.org/dev/peps/pep-0333/#the-server-gateway-side | ||||
Yuya Nishihara
|
r27046 | from __future__ import absolute_import | ||
from .. import ( | ||||
Pulkit Goyal
|
r30636 | encoding, | ||
Yuya Nishihara
|
r27046 | util, | ||
) | ||||
from . import ( | ||||
common, | ||||
) | ||||
Eric Hopper
|
r2506 | |||
def launch(application): | ||||
Yuya Nishihara
|
r30473 | util.setbinary(util.stdin) | ||
util.setbinary(util.stdout) | ||||
Eric Hopper
|
r2506 | |||
Pulkit Goyal
|
r30636 | environ = dict(encoding.environ.iteritems()) | ||
Dirkjan Ochtman
|
r5580 | environ.setdefault('PATH_INFO', '') | ||
Patrick Mezard
|
r10201 | if environ.get('SERVER_SOFTWARE', '').startswith('Microsoft-IIS'): | ||
Mads Kiilerich
|
r17424 | # IIS includes script_name in PATH_INFO | ||
Patrick Mezard
|
r10201 | scriptname = environ['SCRIPT_NAME'] | ||
if environ['PATH_INFO'].startswith(scriptname): | ||||
environ['PATH_INFO'] = environ['PATH_INFO'][len(scriptname):] | ||||
Dirkjan Ochtman
|
r7406 | |||
Yuya Nishihara
|
r30473 | stdin = util.stdin | ||
Augie Fackler
|
r13570 | if environ.get('HTTP_EXPECT', '').lower() == '100-continue': | ||
Yuya Nishihara
|
r30473 | stdin = common.continuereader(stdin, util.stdout.write) | ||
Augie Fackler
|
r13570 | |||
environ['wsgi.input'] = stdin | ||||
Yuya Nishihara
|
r30473 | environ['wsgi.errors'] = util.stderr | ||
Thomas Arendsen Hein
|
r3673 | environ['wsgi.version'] = (1, 0) | ||
environ['wsgi.multithread'] = False | ||||
Eric Hopper
|
r2506 | environ['wsgi.multiprocess'] = True | ||
Thomas Arendsen Hein
|
r3673 | environ['wsgi.run_once'] = True | ||
Eric Hopper
|
r2506 | |||
Benoit Boissinot
|
r10339 | if environ.get('HTTPS', 'off').lower() in ('on', '1', 'yes'): | ||
Eric Hopper
|
r2506 | environ['wsgi.url_scheme'] = 'https' | ||
else: | ||||
environ['wsgi.url_scheme'] = 'http' | ||||
headers_set = [] | ||||
headers_sent = [] | ||||
Yuya Nishihara
|
r30473 | out = util.stdout | ||
Eric Hopper
|
r2506 | |||
def write(data): | ||||
if not headers_set: | ||||
Thomas Arendsen Hein
|
r3673 | raise AssertionError("write() before start_response()") | ||
Eric Hopper
|
r2506 | |||
elif not headers_sent: | ||||
Thomas Arendsen Hein
|
r3673 | # Before the first output, send the stored headers | ||
status, response_headers = headers_sent[:] = headers_set | ||||
out.write('Status: %s\r\n' % status) | ||||
for header in response_headers: | ||||
out.write('%s: %s\r\n' % header) | ||||
out.write('\r\n') | ||||
Eric Hopper
|
r2506 | |||
Alexis S. L. Carvalho
|
r2558 | out.write(data) | ||
out.flush() | ||||
Eric Hopper
|
r2506 | |||
Thomas Arendsen Hein
|
r3673 | def start_response(status, response_headers, exc_info=None): | ||
Eric Hopper
|
r2506 | if exc_info: | ||
try: | ||||
if headers_sent: | ||||
# Re-raise original exception if headers sent | ||||
Peter Ruibal
|
r7008 | raise exc_info[0](exc_info[1], exc_info[2]) | ||
Eric Hopper
|
r2506 | finally: | ||
exc_info = None # avoid dangling circular ref | ||||
elif headers_set: | ||||
raise AssertionError("Headers already set!") | ||||
Thomas Arendsen Hein
|
r3673 | headers_set[:] = [status, response_headers] | ||
Eric Hopper
|
r2506 | return write | ||
Dirkjan Ochtman
|
r6922 | content = application(environ, start_response) | ||
Konstantin Zemlyak
|
r10753 | try: | ||
for chunk in content: | ||||
write(chunk) | ||||
Mads Kiilerich
|
r18552 | if not headers_sent: | ||
write('') # send headers now if body was empty | ||||
Konstantin Zemlyak
|
r10753 | finally: | ||
Alex Gaynor
|
r34487 | getattr(content, 'close', lambda: None)() | ||