X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/b3cded6f3d341f91c4ec4ad3c92afecad48cb917..04322887b0e8c5d846b1d56cdc89e45f47b43c75:/pokedex/db/markdown.py?ds=inline 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.