server.py

#! /usr/bin/env python
import posixpath
import argparse
import urllib
import os

from http.server import SimpleHTTPRequestHandler
from http.server import HTTPServer


class RootedHTTPServer(HTTPServer):

    def __init__(self, base_path, *args, **kwargs):
        HTTPServer.__init__(self, *args, **kwargs)
        self.RequestHandlerClass.base_path = base_path


class RootedHTTPRequestHandler(SimpleHTTPRequestHandler):

    def translate_path(self, path):
        url_path = urllib.parse.urlparse(path).path
        path = posixpath.normpath(urllib.parse.unquote(url_path))
        words = [w for w in path.split('/') if w]
        path = self.base_path
        for word in words:
            drive, word = os.path.splitdrive(word)
            head, word = os.path.split(word)
            if word in (os.curdir, os.pardir):
                continue
            path = os.path.join(path, word)
        return path


def main(HandlerClass=RootedHTTPRequestHandler, ServerClass=RootedHTTPServer):
    parser = argparse.ArgumentParser()
    parser.add_argument('--port', '-p',
                        default=os.getenv('PORT', 5000),
                        type=int)
    parser.add_argument('--dir', '-d', default=os.getcwd(), type=str)
    args = parser.parse_args()

    server_address = ('', args.port)

    httpd = ServerClass(args.dir, server_address, HandlerClass)

    sa = httpd.socket.getsockname()
    print("Serving HTTP on", sa[0], "port", sa[1], "...")
    httpd.serve_forever()


if __name__ == '__main__':
    main()