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