81 lines
2.3 KiB
Python
Executable file
81 lines
2.3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
'''
|
|
Use this in the same way as Python's SimpleHTTPServer:
|
|
|
|
./ssi_server.py [port]
|
|
|
|
The only difference is that, for files ending in '.html', ssi_server will
|
|
inline SSI (Server Side Includes) of the form:
|
|
|
|
<!-- #include virtual="fragment.html" -->
|
|
|
|
Run ./ssi_server.py in this directory and visit localhost:8000 for an exmaple.
|
|
'''
|
|
|
|
import os
|
|
import ssi
|
|
try:
|
|
# This works for Python 2
|
|
from SimpleHTTPServer import SimpleHTTPRequestHandler
|
|
import SimpleHTTPServer
|
|
except ImportError:
|
|
# This works for Python 3
|
|
from http.server import SimpleHTTPRequestHandler
|
|
import http.server as SimpleHTTPServer
|
|
import tempfile
|
|
|
|
|
|
class SSIRequestHandler(SimpleHTTPRequestHandler):
|
|
"""Adds minimal support for <!-- #include --> directives.
|
|
|
|
The key bit is translate_path, which intercepts requests and serves them
|
|
using a temporary file which inlines the #includes.
|
|
"""
|
|
|
|
def __init__(self, request, client_address, server):
|
|
self.temp_files = []
|
|
SimpleHTTPRequestHandler.__init__(self, request, client_address, server)
|
|
|
|
def do_GET(self):
|
|
SimpleHTTPRequestHandler.do_GET(self)
|
|
self.delete_temp_files()
|
|
|
|
def do_HEAD(self):
|
|
SimpleHTTPRequestHandler.do_HEAD(self)
|
|
self.delete_temp_files()
|
|
|
|
def translate_path(self, path):
|
|
fs_path = SimpleHTTPRequestHandler.translate_path(self, path)
|
|
if self.path.endswith('/'):
|
|
for index in "index.html", "index.htm", "index.shtml":
|
|
index = os.path.join(fs_path, index)
|
|
if os.path.exists(index):
|
|
fs_path = index
|
|
break
|
|
|
|
if (fs_path.endswith('.html') or fs_path.endswith(".shtml")) and os.path.exists(fs_path):
|
|
content = ssi.InlineIncludes(fs_path, path)
|
|
fs_path = self.create_temp_file(fs_path, content)
|
|
return fs_path
|
|
|
|
def delete_temp_files(self):
|
|
for temp_file in self.temp_files:
|
|
os.remove(temp_file)
|
|
|
|
def create_temp_file(self, original_path, content):
|
|
_, ext = os.path.splitext(original_path)
|
|
if ext == ".shtml":
|
|
ext = ".html"
|
|
fd, path = tempfile.mkstemp(suffix=ext)
|
|
try:
|
|
os.write(fd, content) # This works for Python 2
|
|
except TypeError:
|
|
os.write(fd, bytes(content, 'UTF-8')) # This works for Python 3
|
|
os.close(fd)
|
|
|
|
self.temp_files.append(path)
|
|
return path
|
|
|
|
|
|
if __name__ == '__main__':
|
|
SimpleHTTPServer.test(HandlerClass=SSIRequestHandler)
|