X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/4041d85dff793268e3e80287d2c2fcc7618182bb..acec17f23a8d18407ca183cb8064636cf5183237:/pokedex/db/rst.py diff --git a/pokedex/db/rst.py b/pokedex/db/rst.py index 474a275..d3d6322 100644 --- 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 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 , @@ -138,7 +140,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 @@ -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)) + +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)