forked from github/pelican
Fixing #1038 by allowing nice URLs.
This commit is contained in:
parent
472abb6fd9
commit
3b315fbc53
1 changed files with 24 additions and 5 deletions
|
|
@ -1,5 +1,7 @@
|
|||
from __future__ import print_function
|
||||
import os
|
||||
import sys
|
||||
import logging
|
||||
try:
|
||||
import SimpleHTTPServer as srvmod
|
||||
except ImportError:
|
||||
|
|
@ -11,19 +13,36 @@ except ImportError:
|
|||
import socketserver # NOQA
|
||||
|
||||
PORT = len(sys.argv) == 2 and int(sys.argv[1]) or 8000
|
||||
SUFFIXES = ['','.html','/index.html']
|
||||
|
||||
Handler = srvmod.SimpleHTTPRequestHandler
|
||||
class ComplexHTTPRequestHandler(srvmod.SimpleHTTPRequestHandler):
|
||||
def do_GET(self):
|
||||
# we are trying to detect the file by having a fallback mechanism
|
||||
r = None
|
||||
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):
|
||||
r = srvmod.SimpleHTTPRequestHandler.do_GET(self)
|
||||
if r is not None:
|
||||
break
|
||||
logging.warning("Unable to find %s file." % self.path)
|
||||
return r
|
||||
|
||||
Handler = ComplexHTTPRequestHandler
|
||||
|
||||
try:
|
||||
httpd = socketserver.TCPServer(("", PORT), Handler)
|
||||
except OSError as e:
|
||||
print("Could not listen on port", PORT)
|
||||
logging.error("Could not listen on port %s" % PORT)
|
||||
sys.exit(getattr(e, 'exitcode', 1))
|
||||
|
||||
|
||||
print("serving at port", PORT)
|
||||
logging.info("serving at port %s" % PORT)
|
||||
try:
|
||||
httpd.serve_forever()
|
||||
except KeyboardInterrupt as e:
|
||||
print("shutting down server")
|
||||
httpd.socket.close()
|
||||
logging.info("shutting down server")
|
||||
httpd.socket.close()
|
||||
Loading…
Add table
Add a link
Reference in a new issue