X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/b3cded6f3d341f91c4ec4ad3c92afecad48cb917..14001d50d3e995a611a492d2bda21b4b75f5ac56:/pokedex/db/markdown.py?ds=sidebyside

diff --git a/pokedex/db/markdown.py b/pokedex/db/markdown.py
index 928c3d4..2bef876 100644
--- a/pokedex/db/markdown.py
+++ b/pokedex/db/markdown.py
@@ -57,6 +57,26 @@ class MarkdownString(object):
         return self.source_text
 
 
+class _MoveEffects(object):
+    def __init__(self, effect_column, move):
+        self.effect_column = effect_column
+        self.move = move
+
+    def __contains__(self, lang):
+        return lang in self.move.move_effect.prose
+
+    def __getitem__(self, lang):
+        try:
+            effect_text = getattr(self.move.move_effect.prose[lang], self.effect_column)
+        except AttributeError:
+            return None
+        effect_text = effect_text.replace(
+            u'$effect_chance',
+            unicode(self.move.effect_chance),
+        )
+
+        return MarkdownString(effect_text)
+
 class MoveEffectProperty(object):
     """Property that wraps a move effect.  Used like this:
 
@@ -73,13 +93,28 @@ class MoveEffectProperty(object):
         self.effect_column = effect_column
 
     def __get__(self, move, move_class):
-        effect_text = getattr(move.move_effect, self.effect_column)
-        effect_text = effect_text.replace(
-            u'$effect_chance',
-            unicode(move.effect_chance),
-        )
+        if move is None:
+            # Don't crash with getattr on the class
+            return NotImplemented
+        return _MoveEffects(self.effect_column, move)['en']
 
-        return MarkdownString(effect_text)
+class MoveEffectsProperty(object):
+    """Property that wraps move effects.  Used like this:
+
+        MoveClass.effects = MoveEffectProperty('effect')
+
+        some_move.effects[lang]            # returns a MarkdownString
+        some_move.effects[lang].as_html    # returns a chunk of HTML
+
+    This class also performs simple substitution on the effect, replacing
+    `$effect_chance` with the move's actual effect chance.
+    """
+
+    def __init__(self, effect_column):
+        self.effect_column = effect_column
+
+    def __get__(self, move, move_class):
+        return _MoveEffects(self.effect_column, move)
 
 class MarkdownColumn(sqlalchemy.types.TypeDecorator):
     """Generic SQLAlchemy column type for Markdown text.