rst2bs.py

#! /usr/bin/env python
# https://gitorious.org/mles/mles/source/5c575ae60df0b483d079120c17881d29c3bca7b3:rst2bs.py
# http://grissiom.blogspot.in/2009/12/some-twist-on-blog.html
#
# Generate an HTML Snippet for BlogSpot from reStructuredText.
#
# Modified from Matthias Friedrich's script that
# Generate an HTML Snippet for WordPress Blogs from reStructuredText.
# http://unmaintainable.wordpress.com/2008/03/22/using-rst-with-wordpress/
#
# This is a modification of the standard HTML writer that leaves out
# the header, the body tag, and several CSS classes that have no use
# in wordpress. What is left is an incomplete HTML document suitable
# for pasting into the WordPress online editor.
#
# Note: This is a quick hack, so it probably won't work for the more
#       advanced features of rst.
#
# Copyright (c) 2008 Matthias Friedrich <matt@mafr.de>
# Copyright (c) 2009 Grissiom <chaos.proton@gmail.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the Artistic License.
#


import sys
import docutils
import docutils.nodes
from docutils.writers import html4css1
from docutils import frontend, writers
from docutils.core import publish_cmdline, default_description


class Writer(html4css1.Writer):
	supported = ('bshtml', )

	settings_spec = html4css1.Writer.settings_spec + ( )

	def __init__(self):
		html4css1.Writer.__init__(self)
		self.translator_class = BsHtmlTranslator


class BsHtmlTranslator(html4css1.HTMLTranslator):
	"""An HTML emitting visitor."""

	doctype = ('')

	def __init__(self, *args):
		html4css1.HTMLTranslator.__init__(self, *args)
		self.stylesheet = [ ]
		self.meta = [ ]
		self.head = [ ]
		self.head_prefix = [ ]
		self.body_prefix = [ ]
		self.body_suffix = [ ]
		self.section_level = 3
		self.compact_simple = True
		self.literal_block = False


	def visit_document(self, node):
		pass

	def depart_document(self, node):
		pass

	def visit_section(self, node):
		self.section_level += 1

	def depart_section(self, node):
		self.section_level -= 1

	def visit_paragraph(self, node):
		if self.should_be_compact_paragraph(node):
			self.context.append('')
		else:
			self.body.append('<p>')
			self.context.append('</p>\n\n')

	def depart_paragraph(self, node):
		self.body.append(self.context.pop())

        #def visit_reference(self, node):
                #attrs = { }
                #if node.has_key('refuri'):
                        #attrs['href'] = node['refuri']
                #else:
                        #assert node.has_key('refid'), 'Invalid internal link'
                        #attrs['href'] = '#' + node['refid']
                #self.body.append(self.starttag(node, 'a', '', **attrs))

	def visit_Text(self, node):
		if self.literal_block:
			text = node.astext()
		else:
			text = node.astext().replace('\n', ' ')
		encoded = self.encode(text)
		if self.in_mailto and self.settings.cloak_email_addresses:
			encoded = self.cloak_email(encoded)
		self.body.append(encoded)

        def visit_block_quote(self, node):
                self.body.append('<blockquote>')

        def depart_block_quote(self, node):
                self.body.append('</blockquote>\n\n')

	#def visit_list_item(self, node):
		#self.body.append('  ' + self.starttag(node, 'li', ''))

	#def visit_title(self, node):
		#h_level = self.section_level + self.initial_header_level - 1
		#self.body.append(
			#self.starttag(node, 'h%s' % h_level, '', **{ }))
		#self.context.append('</h%s>\n\n' % (h_level, ))

	def depart_title(self, node):
		self.body.append(self.context.pop())

	def visit_literal_block(self, node):
		self.literal_block = True
		self.body.append(self.starttag(node, 'pre'))

	def depart_literal_block(self, node):
		self.body.append('\n</pre>\n\n')
		self.literal_block = False

	#def visit_literal(self, node):
		#self.body.append('<code>')

	#def depart_literal(self, node):
		#self.body.append('</code>')


if __name__ == '__main__':
	# docutils tries to load the module 'bshtml' below, so we need an alias
	sys.modules['bshtml'] = sys.modules['__main__']

	try:
	    import locale
	    locale.setlocale(locale.LC_ALL, '')
	except:
	    pass

	description = ('Generates an HTML Snippet for Wordpress from'
			'standalone reStructuredText sources.  '
			+ default_description)

	publish_cmdline(writer_name='bshtml', description=description)

# EOF