forked from github/pelican
1b6f081619
The socket may remain in the TIME_WAIT state for some time after the server shuts down, which prevents another instance of the server from listening on the same port. This change allows the server to reuse the address even when it's still waiting.
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
from __future__ import print_function
|
|
import os
|
|
import sys
|
|
import logging
|
|
try:
|
|
import SimpleHTTPServer as srvmod
|
|
except ImportError:
|
|
import http.server as srvmod # NOQA
|
|
|
|
try:
|
|
import SocketServer as socketserver
|
|
except ImportError:
|
|
import socketserver # NOQA
|
|
|
|
PORT = len(sys.argv) == 2 and int(sys.argv[1]) or 8000
|
|
SUFFIXES = ['', '.html', '/index.html']
|
|
|
|
|
|
class ComplexHTTPRequestHandler(srvmod.SimpleHTTPRequestHandler):
|
|
def do_GET(self):
|
|
# we are trying to detect the file by having a fallback mechanism
|
|
found = False
|
|
for suffix in SUFFIXES:
|
|
if not hasattr(self,'original_path'):
|
|
self.original_path = self.path
|
|
self.path = self.original_path + suffix
|
|
path = self.translate_path(self.path)
|
|
if os.path.exists(path):
|
|
srvmod.SimpleHTTPRequestHandler.do_GET(self)
|
|
logging.info("Found: %s" % self.path)
|
|
found = True
|
|
break
|
|
logging.info("Tried to find file %s, but it doesn't exist. " % self.path)
|
|
if not found:
|
|
logging.warning("Unable to find file %s or variations." % self.path)
|
|
|
|
Handler = ComplexHTTPRequestHandler
|
|
|
|
socketserver.TCPServer.allow_reuse_address = True
|
|
try:
|
|
httpd = socketserver.TCPServer(("", PORT), Handler)
|
|
except OSError as e:
|
|
logging.error("Could not listen on port %s" % PORT)
|
|
sys.exit(getattr(e, 'exitcode', 1))
|
|
|
|
|
|
logging.info("serving at port %s" % PORT)
|
|
try:
|
|
httpd.serve_forever()
|
|
except KeyboardInterrupt as e:
|
|
logging.info("shutting down server")
|
|
httpd.socket.close()
|