mirror of
https://github.com/python/cpython.git
synced 2024-11-28 12:31:14 +08:00
b6d4a8e4de
docs and support for more client-generated CGI variables. (This should complete the WSGI 1.0.1 compliance changes for Python 3.x.)
157 lines
4.7 KiB
Python
157 lines
4.7 KiB
Python
"""BaseHTTPServer that implements the Python WSGI protocol (PEP 3333)
|
|
|
|
This is both an example of how WSGI can be implemented, and a basis for running
|
|
simple web applications on a local machine, such as might be done when testing
|
|
or debugging an application. It has not been reviewed for security issues,
|
|
however, and we strongly recommend that you use a "real" web server for
|
|
production use.
|
|
|
|
For example usage, see the 'if __name__=="__main__"' block at the end of the
|
|
module. See also the BaseHTTPServer module docs for other API information.
|
|
"""
|
|
|
|
from http.server import BaseHTTPRequestHandler, HTTPServer
|
|
import sys
|
|
import urllib.parse
|
|
from wsgiref.handlers import SimpleHandler
|
|
|
|
__version__ = "0.2"
|
|
__all__ = ['WSGIServer', 'WSGIRequestHandler', 'demo_app', 'make_server']
|
|
|
|
|
|
server_version = "WSGIServer/" + __version__
|
|
sys_version = "Python/" + sys.version.split()[0]
|
|
software_version = server_version + ' ' + sys_version
|
|
|
|
|
|
class ServerHandler(SimpleHandler):
|
|
|
|
server_software = software_version
|
|
|
|
def close(self):
|
|
try:
|
|
self.request_handler.log_request(
|
|
self.status.split(' ',1)[0], self.bytes_sent
|
|
)
|
|
finally:
|
|
SimpleHandler.close(self)
|
|
|
|
|
|
|
|
class WSGIServer(HTTPServer):
|
|
|
|
"""BaseHTTPServer that implements the Python WSGI protocol"""
|
|
|
|
application = None
|
|
|
|
def server_bind(self):
|
|
"""Override server_bind to store the server name."""
|
|
HTTPServer.server_bind(self)
|
|
self.setup_environ()
|
|
|
|
def setup_environ(self):
|
|
# Set up base environment
|
|
env = self.base_environ = {}
|
|
env['SERVER_NAME'] = self.server_name
|
|
env['GATEWAY_INTERFACE'] = 'CGI/1.1'
|
|
env['SERVER_PORT'] = str(self.server_port)
|
|
env['REMOTE_HOST']=''
|
|
env['CONTENT_LENGTH']=''
|
|
env['SCRIPT_NAME'] = ''
|
|
|
|
def get_app(self):
|
|
return self.application
|
|
|
|
def set_app(self,application):
|
|
self.application = application
|
|
|
|
|
|
|
|
class WSGIRequestHandler(BaseHTTPRequestHandler):
|
|
|
|
server_version = "WSGIServer/" + __version__
|
|
|
|
def get_environ(self):
|
|
env = self.server.base_environ.copy()
|
|
env['SERVER_PROTOCOL'] = self.request_version
|
|
env['SERVER_SOFTWARE'] = self.server_version
|
|
env['REQUEST_METHOD'] = self.command
|
|
if '?' in self.path:
|
|
path,query = self.path.split('?',1)
|
|
else:
|
|
path,query = self.path,''
|
|
|
|
env['PATH_INFO'] = urllib.parse.unquote_to_bytes(path).decode('iso-8859-1')
|
|
env['QUERY_STRING'] = query
|
|
|
|
host = self.address_string()
|
|
if host != self.client_address[0]:
|
|
env['REMOTE_HOST'] = host
|
|
env['REMOTE_ADDR'] = self.client_address[0]
|
|
|
|
if self.headers.get('content-type') is None:
|
|
env['CONTENT_TYPE'] = self.headers.get_content_type()
|
|
else:
|
|
env['CONTENT_TYPE'] = self.headers['content-type']
|
|
|
|
length = self.headers.get('content-length')
|
|
if length:
|
|
env['CONTENT_LENGTH'] = length
|
|
|
|
for k, v in self.headers.items():
|
|
k=k.replace('-','_').upper(); v=v.strip()
|
|
if k in env:
|
|
continue # skip content length, type,etc.
|
|
if 'HTTP_'+k in env:
|
|
env['HTTP_'+k] += ','+v # comma-separate multiple headers
|
|
else:
|
|
env['HTTP_'+k] = v
|
|
return env
|
|
|
|
def get_stderr(self):
|
|
return sys.stderr
|
|
|
|
def handle(self):
|
|
"""Handle a single HTTP request"""
|
|
|
|
self.raw_requestline = self.rfile.readline()
|
|
if not self.parse_request(): # An error code has been sent, just exit
|
|
return
|
|
|
|
handler = ServerHandler(
|
|
self.rfile, self.wfile, self.get_stderr(), self.get_environ()
|
|
)
|
|
handler.request_handler = self # backpointer for logging
|
|
handler.run(self.server.get_app())
|
|
|
|
|
|
|
|
def demo_app(environ,start_response):
|
|
from io import StringIO
|
|
stdout = StringIO()
|
|
print("Hello world!", file=stdout)
|
|
print(file=stdout)
|
|
h = sorted(environ.items())
|
|
for k,v in h:
|
|
print(k,'=',repr(v), file=stdout)
|
|
start_response("200 OK", [('Content-Type','text/plain; charset=utf-8')])
|
|
return [stdout.getvalue().encode("utf-8")]
|
|
|
|
|
|
def make_server(
|
|
host, port, app, server_class=WSGIServer, handler_class=WSGIRequestHandler
|
|
):
|
|
"""Create a new WSGI server listening on `host` and `port` for `app`"""
|
|
server = server_class((host, port), handler_class)
|
|
server.set_app(app)
|
|
return server
|
|
|
|
|
|
if __name__ == '__main__':
|
|
httpd = make_server('', 8000, demo_app)
|
|
sa = httpd.socket.getsockname()
|
|
print("Serving HTTP on", sa[0], "port", sa[1], "...")
|
|
import webbrowser
|
|
webbrowser.open('http://localhost:8000/xyz?abc')
|
|
httpd.handle_request() # serve one request, then exit
|