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:
parent
d813e24b25
commit
30faa46ab6
2 changed files with 50 additions and 6 deletions
pokedex/db
|
@ -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.
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue