request.py
155 lines
| 5.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 | # | ||
|
r8225 | # This software may be used and distributed according to the terms of the | ||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r2355 | |||
|
r27046 | from __future__ import absolute_import | ||
import cgi | ||||
import errno | ||||
import socket | ||||
from .common import ( | ||||
ErrorResponse, | ||||
HTTP_NOT_MODIFIED, | ||||
statusmessage, | ||||
) | ||||
from .. import ( | ||||
|
r34515 | pycompat, | ||
|
r27046 | util, | ||
) | ||||
|
r2355 | |||
|
r6774 | shortcuts = { | ||
'cl': [('cmd', ['changelog']), ('rev', None)], | ||||
'sl': [('cmd', ['shortlog']), ('rev', None)], | ||||
'cs': [('cmd', ['changeset']), ('node', None)], | ||||
'f': [('cmd', ['file']), ('filenode', None)], | ||||
'fl': [('cmd', ['filelog']), ('filenode', None)], | ||||
'fd': [('cmd', ['filediff']), ('node', None)], | ||||
'fa': [('cmd', ['annotate']), ('filenode', None)], | ||||
'mf': [('cmd', ['manifest']), ('manifest', None)], | ||||
'ca': [('cmd', ['archive']), ('node', None)], | ||||
'tags': [('cmd', ['tags'])], | ||||
'tip': [('cmd', ['changeset']), ('node', ['tip'])], | ||||
'static': [('cmd', ['static']), ('file', None)] | ||||
} | ||||
|
r10261 | def normalize(form): | ||
# first expand the shortcuts | ||||
|
r34514 | for k in shortcuts: | ||
|
r6774 | if k in form: | ||
for name, value in shortcuts[k]: | ||||
if value is None: | ||||
value = form[k] | ||||
form[name] = value | ||||
del form[k] | ||||
|
r10261 | # And strip the values | ||
for k, v in form.iteritems(): | ||||
form[k] = [i.strip() for i in v] | ||||
|
r6774 | return form | ||
|
r5566 | class wsgirequest(object): | ||
|
r26132 | """Higher-level API for a WSGI request. | ||
WSGI applications are invoked with 2 arguments. They are used to | ||||
instantiate instances of this class, which provides higher-level APIs | ||||
for obtaining request parameters, writing HTTP output, etc. | ||||
""" | ||||
|
r5566 | def __init__(self, wsgienv, start_response): | ||
|
r34513 | version = wsgienv[r'wsgi.version'] | ||
|
r3673 | if (version < (1, 0)) or (version >= (2, 0)): | ||
|
r4633 | raise RuntimeError("Unknown and unsupported WSGI version %d.%d" | ||
|
r2506 | % version) | ||
|
r34513 | self.inp = wsgienv[r'wsgi.input'] | ||
self.err = wsgienv[r'wsgi.errors'] | ||||
self.threaded = wsgienv[r'wsgi.multithread'] | ||||
self.multiprocess = wsgienv[r'wsgi.multiprocess'] | ||||
self.run_once = wsgienv[r'wsgi.run_once'] | ||||
|
r2506 | self.env = wsgienv | ||
|
r10261 | self.form = normalize(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) | ||||
|
r7180 | def drain(self): | ||
'''need to read all data from request, httplib is half-duplex''' | ||||
|
r13600 | length = int(self.env.get('CONTENT_LENGTH') or 0) | ||
|
r7180 | for s in util.filechunkiter(self.inp, limit=length): | ||
pass | ||||
|
r18352 | def respond(self, status, type, filename=None, body=None): | ||
|
r34515 | if not isinstance(type, str): | ||
type = pycompat.sysstr(type) | ||||
|
r5888 | if self._start_response is not None: | ||
|
r18348 | self.headers.append(('Content-Type', type)) | ||
if filename: | ||||
r26846 | filename = (filename.rpartition('/')[-1] | |||
|
r18348 | .replace('\\', '\\\\').replace('"', '\\"')) | ||
self.headers.append(('Content-Disposition', | ||||
'inline; filename="%s"' % filename)) | ||||
|
r18352 | if body is not None: | ||
self.headers.append(('Content-Length', str(len(body)))) | ||||
|
r5888 | |||
|
r5926 | for k, v in self.headers: | ||
if not isinstance(v, str): | ||||
|
r18348 | raise TypeError('header value must be string: %r' % (v,)) | ||
|
r5926 | |||
|
r5888 | if isinstance(status, ErrorResponse): | ||
|
r18348 | self.headers.extend(status.headers) | ||
|
r12739 | if status.code == HTTP_NOT_MODIFIED: | ||
# RFC 2616 Section 10.3.5: 304 Not Modified has cases where | ||||
# it MUST NOT include any headers other than these and no | ||||
# body | ||||
self.headers = [(k, v) for (k, v) in self.headers if | ||||
k in ('Date', 'ETag', 'Expires', | ||||
'Cache-Control', 'Vary')] | ||||
|
r26200 | status = statusmessage(status.code, str(status)) | ||
|
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 = [] | ||||
|
r18352 | if body is not None: | ||
self.write(body) | ||||
self.server_write = None | ||||
|
r5888 | |||
|
r5993 | def write(self, thing): | ||
|
r18351 | if thing: | ||
try: | ||||
self.server_write(thing) | ||||
|
r25660 | except socket.error as inst: | ||
|
r18351 | 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 | ||||
|
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 | ||