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

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.

View file

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