Source

cssutils / restserver.py

Full commit
"""
ReSTserver
==========
:Author: Christof Hoeke
:Date: 060111
:Description: a simple reStructuredText Server
:Version: 1.3
:License: This work is licensed under a `Creative Commons License <http://creativecommons.org/licenses/by/2.0/>`_.

* ReST (files with extension RST_EXT) requests will be converted 
  to HTML
* .html requests will be checked if RST_EXT file with same name is present, 
  if yes the request will be redirected to the RST_EXT file (and then 
  converted as above)
* other requests use default SimpleHTTPServer behaviour
* supports command line options same as those for rst2html.py 
  with a patch from Felix Wiemann
* if no path is given "index." + RST_EXT is tried to prevent a maybe older index.html to be shown (from v1.3)
"""

PATH = ''
PORT = 8082
RST_EXT = '.txt'

import BaseHTTPServer
import SimpleHTTPServer

import os.path
import sys
import traceback

import docutils.core
import docutils.io


class Handler(SimpleHTTPServer.SimpleHTTPRequestHandler):

    def do_GET(self):
        """
        RST_EXT files will be transformed to HTML with docutils
        .html request will look for RST_EXT with same name
        """
        p = str(self.path)
        if p == '/':
            p = '/index.html'

        # should be generated from RST_EXT?
        if p.endswith('.html'):
            temp = p[:-5] + RST_EXT
            if os.path.isfile(temp[1:]):
                self.send_response(302)
                self.send_header("Location", temp)
                self.end_headers()
                return

        if p.endswith(RST_EXT):
            html = output = ''
            try:
                pub = docutils.core.Publisher(
                    source_class=docutils.io.FileInput,
                    destination_class=docutils.io.StringOutput)
                pub.set_components('standalone', 'rst', 'html')
                settings_overrides = {'halt_level': 5}
                pub.process_command_line(**settings_overrides)
                # Source and destination path should not be overriden
                # by process_command_line() to avoid getting wrong results.
                pub.set_source(None, p[1:])
                pub.set_destination(None, p[1:])
                html = pub.publish()
            except:
                output = traceback.format_exc()
                print >>sys.stderr, output
                self.send_response(500)
                self.send_header("Content-Type", "text/plain; charset=utf-8")
            else:
                self.send_response(200)
                self.send_header("Content-Type", "text/html; charset=utf-8")
                # following is correct but would conflict with Docutils META element:
                # self.send_header("Content-Type", "application/xhtml+xml; charset=utf-8")
            output += html
            self.send_header("Content-Length", len(output))
            self.end_headers()
            self.wfile.write(output)
        else:
            SimpleHTTPServer.SimpleHTTPRequestHandler.do_GET(self)


def run(server_class=BaseHTTPServer.HTTPServer,
        handler_class=Handler):
    server_address = (PATH, PORT)
    httpd = server_class(server_address, handler_class)
    httpd.serve_forever()


print 'serving on (localhost) %s:%s' % (PATH, PORT)
print 'stop with CTRL + PAUSE'
run()