Python实现简易版的Web服务器(推荐)

python wsgiserver.py app:run

#encoding:utf-8
import socket
import StringIO
import sys
import logging
from datetime import datetime
logger = logging.getLogger(__name__)
class WSGIServer(object):
 address_family = socket.AF_INET
 socket_type = socket.SOCK_STREAM
 request_queue_size = 30
 recv_size = 1024
 def __init__(self, server_address):
  self._listen_socket = _listen_socket = socket.socket(self.address_family,
               self.socket_type)  _listen_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR,1)  _listen_socket.bind(server_address) 
  _listen_socket.listen(self.request_queue_size)  
  _host, _port = _listen_socket.getsockname()
  self._server_name = socket.getfqdn(_host)
  self._server_port = _port
  self._headers_set = []
  self._application = None
  self._client = None
  self._request_data = None
  self._request_method = None
  self._path = None
  self._request_version = None
  self._start_response = None
 def set_application(self, application):
  self._application = application
 def server_forever(self): 
  _listen_socket = self._listen_socket
  logger.info('listen on %s:%s', self._server_name, self._server_port)  while 1:  
   try:
    self._client, _addr = _listen_socket.accept()
    self._handle_request(_addr)
   except KeyboardInterrupt as e:
    logger.info('interrupt')
    break
   except BaseException as e:
    logger.error(e)
 def _handle_request(self, client_addr):
  self._request_data = _request_data = self._client.recv(self.recv_size)
  self._parse_request_data(_request_data)  
  _env = self._get_environment(client_addr) 
  _result = self._application(_env, self.start_response)
  self._finish_response(_result)
 def _parse_request_data(self, request_data): 
  _request_line = str(request_data.splitlines()[0]).rstrip('\r\n')
  (self._request_method, self._path, self._request_version) = _request_line.split()
 def _get_environment(self, client_addr):  
  _env = {}  
  _env['wsgi.version'] = (1, 0)  
  _env['wsgi.url_scheme'] = 'http'
  _env['wsgi.input'] = StringIO.StringIO(self._request_data)    _env['wsgi.errors'] = sys.stderr  
  _env['wsgi.multithread'] = False
  _env['wsgi.multiprocess'] = False
  _env['wsgi.run_once'] = False
  _env['REQUEST_METHOD'] = self._request_method.upper()  
  _env['PATH_INFO'] = self._path
  _env['SERVER_NAME'] = self._server_name
  _env['SERVER_PORT'] = self._server_port
  _env['HTTP_CLIENT_IP'] = client_addr[0]
  logger.info('%s %s %s %s', _env['HTTP_CLIENT_IP'], datetime.now().strftime('%Y-%m-%d %H:%M:%S'), _env['REQUEST_METHOD'], _env['PATH_INFO'])
  return _env
 def start_response(self, status, response_headers, exc_info=None):   _server_headers = [
   ('Date', 'Sun, 7 Jun 2015 23:07:04 GMT'),
   ('Server', 'WSGIServer 0.1')
   ]
  self._headers_set = [status, response_headers + _server_headers]
 def _finish_response(self, result):  
  _status, _response_headers = self._headers_set
  _response = 'HTTP/1.1 {status}\r\n'.format(status=_status)    for _header in _response_headers:  
  _response += '{0}:{1}\r\n'.format(*_header)  
  _response += '\r\n'
  for _data in result:   
   _response += _data
  self._client.sendall(_response)
  self._client.close()
def make_server(server_address, application):
 server = WSGIServer(server_address)
 server.set_application(application)
 return server
if __name__ == '__main__':
 logging.basicConfig(level=logging.DEBUG)
 server_addr= ('0.0.0.0', 43002)
 app_path = sys.argv[1]
 module, application = app_path.split(':')
 module = __import__(module)
 application = getattr(module, application)
 httpd = make_server(server_addr, application)
 httpd.server_forever()

#encoding:utf-8
import os
class PageNotFoundException(BaseException):
 pass
def render(filename, dirname='html'):
 _path = os.path.join(dirname, filename)
 if os.path.exists(_path):  
  with open(_path, 'rb') as handler:  
   return handler.read()
 raise PageNotFoundException('file not found:%s' % _path)
def run(env, start_response):
 _path = env.get('PATH_INFO')
 response = ''
 try:
  _path = 'index.html' if _path == '/' else _path[1:]
  if _path.endswith('.css'):
   start_response('200 OK', [('Content-Type', 'text/css')])
  elif _path.endswith('.js'):
   start_response('200 OK', [('Content-Type', 'text/javascript')]
  elif _path.endswith('.html'):
   start_response('200 OK', [('Content-Type', 'text/html')])
  else:
   start_response('200 OK', [('Content-Type', 'text/plain'), ('Content-Disposition', 'attachment; filename=%s' % os.path.basename(_path))])
  response = render(_path) 
 except PageNotFoundException as e:
  response = render('404.html') 
 return [response, '\r\n']