request.py
99 lines
| 3.4 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.server_write = None | ||||
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 | ||
|
r2506 | self.headers = [] | ||
def __iter__(self): | ||||
return iter([]) | ||||
|
r2355 | |||
|
r2464 | def read(self, count=-1): | ||
return self.inp.read(count) | ||||
|
r5888 | def start_response(self, status): | ||
if self._start_response is not None: | ||||
if not self.headers: | ||||
raise RuntimeError("request.write called before headers sent" + | ||||
" (%s)." % thing) | ||||
if isinstance(status, ErrorResponse): | ||||
status = statusmessage(status.code) | ||||
elif isinstance(status, int): | ||||
status = statusmessage(status) | ||||
self.server_write = self._start_response(status, self.headers) | ||||
self._start_response = None | ||||
self.headers = [] | ||||
|
r5561 | def respond(self, status, *things): | ||
|
r5888 | if not things: | ||
self.start_response(status) | ||||
|
r2355 | for thing in things: | ||
if hasattr(thing, "__iter__"): | ||||
for part in thing: | ||||
|
r5561 | self.respond(status, part) | ||
|
r2355 | else: | ||
|
r2514 | thing = str(thing) | ||
|
r5888 | self.start_response(status) | ||
|
r2355 | try: | ||
|
r2514 | self.server_write(thing) | ||
|
r2355 | except socket.error, inst: | ||
if inst[0] != errno.ECONNRESET: | ||||
raise | ||||
|
r5760 | |||
|
r5561 | def write(self, *things): | ||
self.respond('200 Script output follows', *things) | ||||
|
r2355 | |||
|
r4246 | def writelines(self, lines): | ||
for line in lines: | ||||
self.write(line) | ||||
def flush(self): | ||||
return None | ||||
def close(self): | ||||
return None | ||||
|
r2355 | def header(self, headers=[('Content-type','text/html')]): | ||
|
r2506 | self.headers.extend(headers) | ||
|
r2355 | |||
|
r2466 | def httphdr(self, type, filename=None, length=0, headers={}): | ||
headers = headers.items() | ||||
headers.append(('Content-type', type)) | ||||
|
r2434 | if filename: | ||
headers.append(('Content-disposition', 'attachment; filename=%s' % | ||||
filename)) | ||||
if length: | ||||
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 | ||