Factored Pokémon colors out into their own table.

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

index 474a275..80ac87e 100644 (file)
--- a/pokedex/db/rst.py
+++ b/pokedex/db/rst.py
@@ -40,6 +40,8 @@ from docutils.parsers.rst import Parser, roles
  import docutils.utils
  from docutils.writers.html4css1 import Writer as HTMLWriter
  
  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>,
  ### 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>,
@@ -85,7 +87,7 @@ roles.register_local_role('data', data_role)
  
  class RstString(object):
      """Wraps a reStructuredText string.  Stringifies to the original text, but
  
  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={}):
      """
  
      def __init__(self, source_text, document_properties={}):
@@ -138,7 +140,7 @@ class RstString(object):
  class MoveEffectProperty(object):
      """Property that wraps a move effect.  Used like this:
  
  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
  
          some_move.effect            # returns an RstString
          some_move.effect.as_html    # returns a chunk of HTML
@@ -162,3 +164,17 @@ class MoveEffectProperty(object):
          return RstString(getattr(move.move_effect, self.effect_column),
                           document_properties=dict(
                               _pokedex_handle_data=data_role_func))
          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)