Make rst as_html handle errors a little more nicely.

[zzz-pokedex.git] / pokedex / db / rst.py
diff --git a/pokedex/db/rst.py b/pokedex/db/rst.py

index e70af2c..a77feef 100644 (file)
--- a/pokedex/db/rst.py
+++ b/pokedex/db/rst.py
@@ -33,6 +33,8 @@ are, apparently, global.
      return a reST node.
  """
  
+import cgi
+
  from docutils.frontend import OptionParser
  from docutils.io import Output
  import docutils.nodes
@@ -40,6 +42,8 @@ from docutils.parsers.rst import Parser, roles
  import docutils.utils
  from docutils.writers.html4css1 import Writer as HTMLWriter
  
+import sqlalchemy.types
+
  ### Subclasses of bits of docutils, to munge it into doing what I want
  class HTMLFragmentWriter(HTMLWriter):
      """Translates reST to HTML, but only as a fragment.  Enclosing <body>,
@@ -63,12 +67,13 @@ class UnicodeOutput(Output):
  ### Text roles
  
  def generic_role(name, rawtext, text, lineno, inliner, options={}, content=[]):
-    node = docutils.nodes.strong(text, rawtext, **options)
+    node = docutils.nodes.emphasis(rawtext, text, **options)
      return [node], []
  
  roles.register_local_role('ability', generic_role)
  roles.register_local_role('item', generic_role)
  roles.register_local_role('move', generic_role)
+roles.register_local_role('type', generic_role)
  roles.register_local_role('pokemon', generic_role)
  roles.register_local_role('mechanic', generic_role)
  
@@ -84,7 +89,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={}):
@@ -128,8 +133,17 @@ 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"""
+                <p><em>Error in markup!  Raw source is below.</em></p>
+                <pre>{0}</pre>
+            """.format( cgi.escape(self.source_text) )
+
+        destination = UnicodeOutput()
          writer = HTMLFragmentWriter()
          return writer.write(document, destination)
  
@@ -137,7 +151,7 @@ class RstString(object):
  class MoveEffectProperty(object):
      """Property that wraps a move effect.  Used like this:
  
-        MoveClass.effect = MoveEffectProperty()
+        MoveClass.effect = MoveEffectProperty('effect')
  
          some_move.effect            # returns an RstString
          some_move.effect.as_html    # returns a chunk of HTML
@@ -147,6 +161,9 @@ class MoveEffectProperty(object):
      lie and it doesn't yet.
      """
  
+    def __init__(self, effect_column):
+        self.effect_column = effect_column
+
      def __get__(self, move, move_class):
          # Attach a function for handling the `data` role
          # XXX make this a little more fault-tolerant..  maybe..
@@ -155,6 +172,20 @@ class MoveEffectProperty(object):
              newtext = getattr(move, text[5:])
              return docutils.nodes.Text(newtext, rawtext)
  
-        return RstString(move.move_effect.effect,
+        return RstString(getattr(move.move_effect, self.effect_column),
                           document_properties=dict(
                               _pokedex_handle_data=data_role_func))
+
+class RstTextColumn(sqlalchemy.types.TypeDecorator):
+    """Generic column type for reST text.
+
+    Do NOT use this for move effects!  They need to know what move they belong
+    to so they can fill in, e.g., effect chances.
+    """
+    impl = sqlalchemy.types.Unicode
+
+    def process_bind_param(self, value, dialect):
+        return unicode(value)
+
+    def process_result_value(self, value, dialect):
+        return RstString(value)