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
|
@ -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.
|
||||
|
|
|
@ -1521,11 +1521,20 @@ Move.super_contest_combo_prev = association_proxy('super_contest_combo_second',
|
|||
Move.target = relation(MoveTarget, backref='moves')
|
||||
Move.type = relation(Type, back_populates='moves')
|
||||
|
||||
Move.effect = markdown.MoveEffectProperty('effect')
|
||||
Move.effects = markdown.MoveEffectsProperty('effect')
|
||||
Move.short_effect = markdown.MoveEffectProperty('short_effect')
|
||||
Move.short_effects = markdown.MoveEffectsProperty('short_effect')
|
||||
|
||||
MoveChangelog.changed_in = relation(VersionGroup, backref='move_changelog')
|
||||
MoveChangelog.move_effect = relation(MoveEffect, backref='move_changelog')
|
||||
MoveChangelog.type = relation(Type, backref='move_changelog')
|
||||
|
||||
MoveChangelog.effect = markdown.MoveEffectProperty('effect')
|
||||
MoveChangelog.effects = markdown.MoveEffectsProperty('effect')
|
||||
MoveChangelog.short_effect = markdown.MoveEffectProperty('short_effect')
|
||||
MoveChangelog.short_effects = markdown.MoveEffectsProperty('short_effect')
|
||||
|
||||
MoveEffect.category_map = relation(MoveEffectCategoryMap)
|
||||
MoveEffect.categories = association_proxy('category_map', 'category')
|
||||
MoveEffect.changelog = relation(MoveEffectChangelog,
|
||||
|
|
Loading…
Reference in a new issue