1
0
Fork 0
mirror of https://github.com/veekun/pokedex.git synced 2024-08-20 18:16:34 +00:00

Multi-language markdown move properties

Singular property name for English text
Plural property name for dict of texts keyed by language
This commit is contained in:
Petr Viktorin 2011-02-04 06:22:44 +02:00 committed by Eevee
parent d813e24b25
commit 30faa46ab6
2 changed files with 50 additions and 6 deletions
pokedex/db

View file

@ -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.