request.py
102 lines
| 3.5 KiB
| text/x-python
|
PythonLexer
|
r2391 | # hgweb/request.py - An http request from either CGI or the standalone server. | ||
|
r2355 | # | ||
# Copyright 21 May 2005 - (c) 2005 Jake Edge <jake@edge2.net> | ||||
|
r2859 | # Copyright 2005, 2006 Matt Mackall <mpm@selenic.com> | ||
|
r2355 | # | ||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
|
r3963 | import socket, cgi, errno | ||
|
r2355 | from mercurial.i18n import gettext as _ | ||
|
r5563 | from common import ErrorResponse, statusmessage | ||
|
r2355 | |||
|
r5566 | class wsgirequest(object): | ||
def __init__(self, wsgienv, start_response): | ||||
|
r2506 | version = wsgienv['wsgi.version'] | ||
|
r3673 | if (version < (1, 0)) or (version >= (2, 0)): | ||
|
r4633 | raise RuntimeError("Unknown and unsupported WSGI version %d.%d" | ||
|
r2506 | % version) | ||
self.inp = wsgienv['wsgi.input'] | ||||
self.err = wsgienv['wsgi.errors'] | ||||
self.threaded = wsgienv['wsgi.multithread'] | ||||
self.multiprocess = wsgienv['wsgi.multiprocess'] | ||||
self.run_once = wsgienv['wsgi.run_once'] | ||||
self.env = wsgienv | ||||
|
r2355 | self.form = cgi.parse(self.inp, self.env, keep_blank_values=1) | ||
|
r5888 | self._start_response = start_response | ||
|
r5993 | self.server_write = None | ||
|
r2506 | self.headers = [] | ||
def __iter__(self): | ||||
return iter([]) | ||||
|
r2355 | |||
|
r2464 | def read(self, count=-1): | ||
return self.inp.read(count) | ||||
|
r5993 | def respond(self, status, type=None, filename=None, length=0): | ||
|
r5888 | if self._start_response is not None: | ||
|
r5993 | |||
self.httphdr(type, filename, length) | ||||
|
r5888 | if not self.headers: | ||
|
r5922 | raise RuntimeError("request.write called before headers sent") | ||
|
r5888 | |||
|
r5926 | for k, v in self.headers: | ||
if not isinstance(v, str): | ||||
raise TypeError('header value must be string: %r' % v) | ||||
|
r5888 | if isinstance(status, ErrorResponse): | ||
status = statusmessage(status.code) | ||||
|
r5993 | elif status == 200: | ||
status = '200 Script output follows' | ||||
|
r5888 | elif isinstance(status, int): | ||
status = statusmessage(status) | ||||
self.server_write = self._start_response(status, self.headers) | ||||
self._start_response = None | ||||
self.headers = [] | ||||
|
r5993 | def write(self, thing): | ||
if hasattr(thing, "__iter__"): | ||||
for part in thing: | ||||
self.write(part) | ||||
else: | ||||
thing = str(thing) | ||||
try: | ||||
self.server_write(thing) | ||||
except socket.error, inst: | ||||
if inst[0] != errno.ECONNRESET: | ||||
raise | ||||
|
r2355 | |||
|
r4246 | def writelines(self, lines): | ||
for line in lines: | ||||
self.write(line) | ||||
def flush(self): | ||||
return None | ||||
def close(self): | ||||
return None | ||||
|
r5930 | def header(self, headers=[('Content-Type','text/html')]): | ||
|
r2506 | self.headers.extend(headers) | ||
|
r2355 | |||
|
r5993 | def httphdr(self, type=None, filename=None, length=0, headers={}): | ||
|
r2466 | headers = headers.items() | ||
|
r5993 | if type is not None: | ||
headers.append(('Content-Type', type)) | ||||
|
r2434 | if filename: | ||
|
r6137 | filename = (filename.split('/')[-1] | ||
.replace('\\', '\\\\').replace('"', '\\"')) | ||||
|
r6136 | headers.append(('Content-Disposition', | ||
|
r6137 | 'inline; filename="%s"' % filename)) | ||
|
r2434 | if length: | ||
|
r5930 | headers.append(('Content-Length', str(length))) | ||
|
r2355 | self.header(headers) | ||
|
r5566 | |||
def wsgiapplication(app_maker): | ||||
|
r5887 | '''For compatibility with old CGI scripts. A plain hgweb() or hgwebdir() | ||
can and should now be used as a WSGI application.''' | ||||
|
r5760 | application = app_maker() | ||
def run_wsgi(env, respond): | ||||
|
r5887 | return application(env, respond) | ||
|
r5760 | return run_wsgi | ||