server.py

#!/usr/bin/env python
"""
A server that works with the livereload chrome extension to automatically
refresh the browser whenver a set of files changes. The server degrades to a
SimpleHTTPServer w/o livereload capability if the livereload Python package is
not available.

The default port is 8000. Supply a commandline argument for a different port.

This serves files from the current directory on your local browser and is good
for viewing slides easily.

Usage:

python server.py

or

python server.py <port>
"""

import sys
import BaseHTTPServer
from SimpleHTTPServer import SimpleHTTPRequestHandler

HandlerClass = SimpleHTTPRequestHandler
ServerClass = BaseHTTPServer.HTTPServer
Protocol = "HTTP/1.0"
SERVER_ADDRESS = '127.0.0.1'

try:
    from livereload import Server as LiveServer
except ImportError:
    LiveServer = None


DEFAULT_PORT = 8000
patterns = [
    '*/*.html',
    '**/*.md',
    '*/images/',
    '*/css/',
]


def main(port):
    if LiveServer is not None:
        live_reload_server(port)
    else:
        print("Using SimpleHTTPServer. Follow the README for livereload if "
            "desired")
        simple_server(port)


def simple_server(port):
    HandlerClass.protocol_version = Protocol
    httpd = ServerClass((SERVER_ADDRESS, port), HandlerClass)
    sa = httpd.socket.getsockname()
    print("Serving HTTP on", sa[0], "port", sa[1], "...")
    httpd.serve_forever()


def live_reload_server(port):
    server = LiveServer()
    for pattern in patterns:
        server.watch(pattern)
    server.serve(port=port)


if __name__ == '__main__':
    port = int(sys.argv[1]) if sys.argv[1:] else 8000
    main(port)