X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/1f0d231daf5b121dce85da6658b36d5ead980104..ee6df6ce9cf56afa4fe5fe24a7cb6fa2c474ce99:/pokedex/db/rst.py diff --git a/pokedex/db/rst.py b/pokedex/db/rst.py index d3d6322..9ee84ba 100644 --- a/pokedex/db/rst.py +++ b/pokedex/db/rst.py @@ -33,12 +33,15 @@ are, apparently, global. return a reST node. """ +import cgi + from docutils.frontend import OptionParser from docutils.io import Output import docutils.nodes from docutils.parsers.rst import Parser, roles import docutils.utils from docutils.writers.html4css1 import Writer as HTMLWriter +from docutils.writers import UnfilteredWriter import sqlalchemy.types @@ -52,6 +55,53 @@ class HTMLFragmentWriter(HTMLWriter): subs = self.interpolation_dict() return subs['body'] + +class TextishTranslator(docutils.nodes.SparseNodeVisitor): + """A simple translator that tries to return plain text that still captures + the spirit of the original (basic) formatting. + + This will probably not be useful for anything complicated; it's only meant + for extremely simple text. + """ + + def __init__(self, document): + self.document = document + self.translated = u'' + + def visit_Text(self, node): + """Text is left alone.""" + self.translated += node.astext() + + def depart_paragraph(self, node): + """Append a blank line after a paragraph, unless it's the last of its + siblings. + """ + if not node.parent: + return + + # Loop over siblings. If we see a sibling after we see this node, then + # append the blank line + seen_node = False + for sibling in node.parent: + if sibling is node: + seen_node = True + continue + + if seen_node: + self.translated += u'\n\n' + return + +class TextishWriter(UnfilteredWriter): + """Translates reST back into plain text, aka more reST. Difference is that + custom roles are handled, so you get "50% chance" instead of junk. + """ + + def translate(self): + visitor = TextishTranslator(self.document) + self.document.walkabout(visitor) + self.output = visitor.translated + + class UnicodeOutput(Output): """reST Unicode output. The distribution only has a StringOutput, and I want me some Unicode. @@ -87,7 +137,7 @@ roles.register_local_role('data', data_role) class RstString(object): """Wraps a reStructuredText string. Stringifies to the original text, but - may be translated to HTML with .to_html(). + may be translated to HTML with .as_html(). """ def __init__(self, source_text, document_properties={}): @@ -131,11 +181,30 @@ class RstString(object): """Returns the string as HTML4.""" document = self.rest_document - destination = UnicodeOutput() + # Check for errors; don't want to leave the default error message cruft + # in here + if document.next_node(condition=docutils.nodes.system_message): + # Boo! Cruft. + return u""" +

Error in markup! Raw source is below.

+
{0}
+ """.format( cgi.escape(self.source_text) ) + + destination = UnicodeOutput() writer = HTMLFragmentWriter() return writer.write(document, destination) + @property + def as_text(self): + """Returns the string mostly unchanged, save for our custom roles.""" + + document = self.rest_document + + destination = UnicodeOutput() + writer = TextishWriter() + return writer.write(document, destination) + class MoveEffectProperty(object): """Property that wraps a move effect. Used like this: