# XXX: Check if "gametext" is set correctly everywhere
import collections
+from functools import partial
from sqlalchemy import Column, ForeignKey, MetaData, PrimaryKeyConstraint, Table, UniqueConstraint
from sqlalchemy.ext.declarative import (
)
from sqlalchemy.ext.associationproxy import association_proxy
from sqlalchemy.orm import (
- backref, eagerload_all, relation, class_mapper, synonym, mapper,
+ backref, compile_mappers, eagerload_all, relation, class_mapper, synonym, mapper,
)
from sqlalchemy.orm.session import Session, object_session
-from sqlalchemy.orm.collections import attribute_mapped_collection
-from sqlalchemy.ext.associationproxy import association_proxy
+from sqlalchemy.orm.interfaces import AttributeExtension
+from sqlalchemy.orm.collections import attribute_mapped_collection, MappedCollection, collection, collection_adapter
+from sqlalchemy.ext.associationproxy import _AssociationDict, association_proxy
from sqlalchemy.sql import and_
-from sqlalchemy.sql.expression import ColumnOperators
+from sqlalchemy.sql.expression import ColumnOperators, bindparam
from sqlalchemy.schema import ColumnDefault
from sqlalchemy.types import *
from inspect import isclass
-from pokedex.db import markdown
-
-# A list of all table classes will live in table_classes
-table_classes = []
-
-class TableMetaclass(DeclarativeMeta):
- def __init__(cls, name, bases, attrs):
- super(TableMetaclass, cls).__init__(name, bases, attrs)
- if hasattr(cls, '__tablename__'):
- table_classes.append(cls)
+from pokedex.db import markdown, multilang
class TableSuperclass(object):
"""Superclass for declarative tables, to give them some generic niceties
return unicode(self).encode('utf8')
metadata = MetaData()
-TableBase = declarative_base(metadata=metadata, cls=TableSuperclass, metaclass=TableMetaclass)
+TableBase = declarative_base(metadata=metadata, cls=TableSuperclass)
-### Helper classes
-class LanguageSpecific(object):
- """Mixin for prose and text tables"""
- @declared_attr
- def language_id(cls):
- return Column(Integer, ForeignKey('languages.id'), primary_key=True, nullable=False,
- info=dict(description="The language"))
-class LanguageSpecificColumn(object):
- """A column that will not appear in the table it's defined in, but in a related one"""
- _ordering = [1]
- def __init__(self, *args, **kwargs):
- self.args = args
- self.plural = kwargs.pop('plural')
- self.kwargs = kwargs
- self.order = self._ordering[0]
- self._ordering[0] += 1
+### Need Language first, to create the partial() below
- def makeSAColumn(self):
- return Column(*self.args, **self.kwargs)
-
-class ProseColumn(LanguageSpecificColumn):
- """A column that will appear in the corresponding _prose table"""
+class Language(TableBase):
+ u"""A language the Pokémon games have been transleted into
+ """
+ __tablename__ = 'languages'
+ __singlename__ = 'language'
+ id = Column(Integer, primary_key=True, nullable=False,
+ info=dict(description="A numeric ID"))
+ iso639 = Column(Unicode(2), nullable=False,
+ info=dict(description="The two-letter code of the country where this language is spoken. Note that it is not unique.", format='identifier'))
+ iso3166 = Column(Unicode(2), nullable=False,
+ info=dict(description="The two-letter code of the language. Note that it is not unique.", format='identifier'))
+ identifier = Column(Unicode(16), nullable=False,
+ info=dict(description="An identifier", format='identifier'))
+ official = Column(Boolean, nullable=False, index=True,
+ info=dict(description=u"True iff games are produced in the language."))
+ order = Column(Integer, nullable=True,
+ info=dict(description=u"Order for sorting in foreign name lists."))
-class TextColumn(LanguageSpecificColumn):
- """A column that will appear in the corresponding _text table"""
+create_translation_table = partial(multilang.create_translation_table, language_class=Language)
+create_translation_table('language_texts', Language, 'names',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
### The actual tables
info=dict(description="An identifier", format='identifier'))
generation_id = Column(Integer, ForeignKey('generations.id'), nullable=False,
info=dict(description="The ID of the generation this ability was introduced in", detail=True))
- effect = ProseColumn(markdown.MarkdownColumn(5120), plural='effects', nullable=False,
- info=dict(description="A detailed description of this ability's effect", format='markdown'))
- short_effect = ProseColumn(markdown.MarkdownColumn(255), plural='short_effects', nullable=False,
- info=dict(description="A short summary of this ability's effect", format='markdown'))
- name = TextColumn(Unicode(24), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('ability_texts', Ability, 'names',
+ name = Column(Unicode(24), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
+create_translation_table('ability_prose', Ability, 'prose',
+ effect = Column(markdown.MarkdownColumn(5120), nullable=False,
+ info=dict(description="A detailed description of this ability's effect", format='markdown')),
+ short_effect = Column(markdown.MarkdownColumn(255), nullable=False,
+ info=dict(description="A short summary of this ability's effect", format='markdown')),
+)
class AbilityChangelog(TableBase):
"""History of changes to abilities across main game versions."""
info=dict(description="The ID of the ability that changed"))
changed_in_version_group_id = Column(Integer, ForeignKey('version_groups.id'), nullable=False,
info=dict(description="The ID of the version group in which the ability changed"))
- effect = ProseColumn(markdown.MarkdownColumn(255), plural='effects', nullable=False,
+
+create_translation_table('ability_changelog_prose', AbilityChangelog, 'prose',
+ effect = Column(markdown.MarkdownColumn(255), nullable=False,
info=dict(description="A description of the old behavior", format='markdown'))
+)
-class AbilityFlavorText(TableBase, LanguageSpecific):
+class AbilityFlavorText(TableBase):
u"""In-game flavor text of an ability
"""
__tablename__ = 'ability_flavor_text'
info=dict(description="The ID of the ability"))
version_group_id = Column(Integer, ForeignKey('version_groups.id'), primary_key=True, nullable=False, autoincrement=False,
info=dict(description="The ID of the version group this flavor text is taken from"))
+ language_id = Column(Integer, ForeignKey('languages.id'), primary_key=True, nullable=False,
+ info=dict(description="The language"))
flavor_text = Column(Unicode(64), nullable=False,
info=dict(description="The actual flavor text", official=True, format='gametext'))
info=dict(description="A unique ID for this firmness"))
identifier = Column(Unicode(10), nullable=False,
info=dict(description="An identifier", format='identifier'))
- name = TextColumn(Unicode(10), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('berry_firmness_texts', BerryFirmness, 'names',
+ name = Column(Unicode(10), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class BerryFlavor(TableBase):
u"""A Berry flavor level.
info=dict(description="The base number of hearts the user of this move gets"))
jam = Column(SmallInteger, nullable=False,
info=dict(description="The base number of hearts the user's opponent loses"))
- flavor_text = ProseColumn(Unicode(64), plural='flavor_texts', nullable=False,
- info=dict(description="The in-game description of this effect", official=True, format='gametext'))
- effect = ProseColumn(Unicode(255), plural='effects', nullable=False,
- info=dict(description="A detailed description of the effect", format='plaintext'))
+
+create_translation_table('contest_effect_prose', ContestEffect, 'prose',
+ flavor_text = Column(Unicode(64), nullable=False,
+ info=dict(description="The in-game description of this effect", official=True, format='gametext')),
+ effect = Column(Unicode(255), nullable=False,
+ info=dict(description="A detailed description of the effect", format='plaintext')),
+)
class ContestType(TableBase):
u"""A Contest type, such as "cool" or "smart", and their associated Berry flavors and Pokéblock colors.
info=dict(description="A unique ID for this Contest type"))
identifier = Column(Unicode(6), nullable=False,
info=dict(description="An identifier", format='identifier'))
- flavor = TextColumn(Unicode(6), nullable=False, plural='flavors',
- info=dict(description="The name of the corresponding Berry flavor", official=True, format='plaintext'))
- color = TextColumn(Unicode(6), nullable=False, plural='colors',
- info=dict(description=u"The name of the corresponding Pokéblock color", official=True, format='plaintext'))
- name = TextColumn(Unicode(6), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('contest_type_texts', ContestType, 'names',
+ name = Column(Unicode(6), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+ flavor = Column(Unicode(6), nullable=False,
+ info=dict(description="The name of the corresponding Berry flavor", official=True, format='plaintext')),
+ color = Column(Unicode(6), nullable=False,
+ info=dict(description=u"The name of the corresponding Pokéblock color", official=True, format='plaintext')),
+)
class EggGroup(TableBase):
u"""An Egg group. Usually, two Pokémon can breed if they share an Egg Group.
info=dict(description="A unique ID for this group"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u"An identifier.", format='identifier'))
- name = ProseColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('egg_group_prose', EggGroup, 'names',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class Encounter(TableBase):
u"""Encounters with wild Pokémon.
info=dict(description="A unique ID for this condition"))
identifier = Column(Unicode(64), nullable=False,
info=dict(description="An identifier", format='identifier'))
- name = ProseColumn(Unicode(64), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('encounter_condition_prose', EncounterCondition, 'prose',
+ name = Column(Unicode(64), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class EncounterConditionValue(TableBase):
u"""A possible state for a condition; for example, the state of 'swarm' could be 'swarm' or 'no swarm'.
info=dict(description="An identifier", format='identifier'))
is_default = Column(Boolean, nullable=False,
info=dict(description='Set if this value is the default state for the condition'))
- name = ProseColumn(Unicode(64), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('encounter_condition_value_prose', EncounterConditionValue, 'prose',
+ name = Column(Unicode(64), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class EncounterConditionValueMap(TableBase):
u"""Maps encounters to the specific conditions under which they occur.
info=dict(description="A unique ID for the terrain"))
identifier = Column(Unicode(64), nullable=False,
info=dict(description="An identifier", format='identifier'))
- name = ProseColumn(Unicode(64), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('encounter_terrain_prose', EncounterTerrain, 'prose',
+ name = Column(Unicode(64), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class EncounterSlot(TableBase):
u"""An abstract "slot" within a terrain, associated with both some set of conditions and a rarity.
info=dict(description="A numeric ID"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description="An identifier", format='identifier'))
- name = ProseColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('evolution_trigger_prose', EvolutionTrigger, 'prose',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class Experience(TableBase):
u"""EXP needed for a certain level with a certain growth rate
info=dict(description=u"ID of the Pokédex this generation's main games use by default"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u'An identifier', format='identifier'))
- name = TextColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('generation_texts', Generation, 'names',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class GrowthRate(TableBase):
u"""Growth rate of a Pokémon, i.e. the EXP → level function.
info=dict(description="An identifier", format='identifier'))
formula = Column(Unicode(500), nullable=False,
info=dict(description="The formula", format='latex'))
- name = ProseColumn(Unicode(20), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('growth_rate_prose', GrowthRate, 'prose',
+ name = Column(Unicode(20), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class Item(TableBase):
u"""An Item from the games, like "Poké Ball" or "Bicycle".
info=dict(description=u"Power of the move Fling when used with this item."))
fling_effect_id = Column(Integer, ForeignKey('item_fling_effects.id'), nullable=True,
info=dict(description=u"ID of the fling-effect of the move Fling when used with this item. Note that these are different from move effects."))
- short_effect = ProseColumn(Unicode(256), plural='short_effects', nullable=False,
- info=dict(description="A short summary of the effect", format='plaintext'))
- effect = ProseColumn(markdown.MarkdownColumn(5120), plural='effects', nullable=False,
- info=dict(description=u"Detailed description of the item's effect.", format='markdown'))
- name = TextColumn(Unicode(20), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
@property
def appears_underground(self):
"""
return any(flag.identifier == u'underground' for flag in self.flags)
+create_translation_table('item_texts', Item, 'names',
+ name = Column(Unicode(20), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
+create_translation_table('item_prose', Item, 'prose',
+ short_effect = Column(Unicode(256), nullable=False,
+ info=dict(description="A short summary of the effect", format='plaintext')),
+ effect = Column(markdown.MarkdownColumn(5120), nullable=False,
+ info=dict(description=u"Detailed description of the item's effect.", format='markdown')),
+)
+
class ItemCategory(TableBase):
u"""An item category
"""
info=dict(description="ID of the pocket these items go to"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description="An identifier", format='identifier'))
- name = ProseColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('item_category_prose', ItemCategory, 'prose',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class ItemFlag(TableBase):
u"""An item attribute such as "consumable" or "holdable".
info=dict(description="A numeric ID"))
identifier = Column(Unicode(24), nullable=False,
info=dict(description="Identifier of the flag", format='identifier'))
- description = ProseColumn(Unicode(64), plural='descriptions', nullable=False,
- info=dict(description="Short description of the flag", format='plaintext'))
- name = ProseColumn(Unicode(24), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('item_flag_prose', ItemFlag, 'prose',
+ name = Column(Unicode(24), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+ description = Column(Unicode(64), nullable=False,
+ info=dict(description="Short description of the flag", format='plaintext')),
+)
class ItemFlagMap(TableBase):
u"""Maps an item flag to its item.
item_flag_id = Column(Integer, ForeignKey('item_flags.id'), primary_key=True, autoincrement=False, nullable=False,
info=dict(description="The ID of the item flag"))
-class ItemFlavorText(TableBase, LanguageSpecific):
+class ItemFlavorText(TableBase):
u"""An in-game description of an item
"""
__tablename__ = 'item_flavor_text'
info=dict(description="The ID of the item"))
version_group_id = Column(Integer, ForeignKey('version_groups.id'), primary_key=True, autoincrement=False, nullable=False,
info=dict(description="ID of the version group that sports this text"))
+ language_id = Column(Integer, ForeignKey('languages.id'), primary_key=True, nullable=False,
+ info=dict(description="The language"))
flavor_text = Column(Unicode(255), nullable=False,
info=dict(description="The flavor text itself", official=True, format='gametext'))
__singlename__ = 'item_fling_effect'
id = Column(Integer, primary_key=True, nullable=False,
info=dict(description="A numeric ID"))
- effect = ProseColumn(Unicode(255), plural='effects', nullable=False,
- info=dict(description="Description of the effect", format='plaintext'))
+
+create_translation_table('item_fling_effect_prose', ItemFlingEffect, 'prose',
+ effect = Column(Unicode(255), nullable=False,
+ info=dict(description="Description of the effect", format='plaintext')),
+)
class ItemInternalID(TableBase):
u"""The internal ID number a game uses for an item
info=dict(description="A numeric ID"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description="An identifier of this pocket", format='identifier'))
- name = TextColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
-
-class Language(TableBase):
- u"""A language the Pokémon games have been transleted into
- """
- __tablename__ = 'languages'
- __singlename__ = 'language'
- id = Column(Integer, primary_key=True, nullable=False,
- info=dict(description="A numeric ID"))
- iso639 = Column(Unicode(2), nullable=False,
- info=dict(description="The two-letter code of the country where this language is spoken. Note that it is not unique.", format='identifier'))
- iso3166 = Column(Unicode(2), nullable=False,
- info=dict(description="The two-letter code of the language. Note that it is not unique.", format='identifier'))
- identifier = Column(Unicode(16), nullable=False,
- info=dict(description="An identifier", format='identifier'))
- official = Column(Boolean, nullable=False, index=True,
- info=dict(description=u"True iff games are produced in the language."))
- order = Column(Integer, nullable=True,
- info=dict(description=u"Order for sorting in foreign name lists."))
- name = TextColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
- # Languages compare equal to its identifier, so a dictionary of
- # translations, with a Language as the key, can be indexed by the identifier
- def __eq__(self, other):
- try:
- return (
- self is other or
- self.identifier == other or
- self.identifier == other.identifier
- )
- except AttributeError:
- return NotImplemented
-
- def __ne__(self, other):
- return not (self == other)
-
- def __hash__(self):
- return hash(self.identifier)
+create_translation_table('item_pocket_texts', ItemPocket, 'names',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class Location(TableBase):
u"""A place in the Pokémon world
info=dict(description="ID of the region this location is in"))
identifier = Column(Unicode(64), nullable=False,
info=dict(description="An identifier", format='identifier'))
- name = TextColumn(Unicode(64), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('location_texts', Location, 'names',
+ name = Column(Unicode(64), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class LocationArea(TableBase):
u"""A sub-area of a location
info=dict(description="ID the games ude for this area"))
identifier = Column(Unicode(64), nullable=True,
info=dict(description="An identifier", format='identifier'))
- name = ProseColumn(Unicode(64), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('location_area_prose', LocationArea, 'prose',
+ name = Column(Unicode(64), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class LocationAreaEncounterRate(TableBase):
# XXX: What's this exactly? Someone add the docstring & revise the descriptions
info=dict(description="A numeric ID"))
identifier = Column(Unicode(8), nullable=False,
info=dict(description="An identifier", format='identifier'))
- name = ProseColumn(Unicode(8), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('move_battle_style_prose', MoveBattleStyle, 'prose',
+ name = Column(Unicode(8), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class MoveEffectCategory(TableBase):
u"""Category of a move effect
info=dict(description="An identifier", format='identifier'))
can_affect_user = Column(Boolean, nullable=False,
info=dict(description="Set if the user can be affected"))
- name = ProseColumn(Unicode(64), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('move_effect_category_prose', MoveEffectCategory, 'prose',
+ name = Column(Unicode(64), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+)
class MoveEffectCategoryMap(TableBase):
u"""Maps a move effect category to a move effect
info=dict(description="A numeric ID"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description="An identifier", format='identifier'))
- description = ProseColumn(Unicode(64), plural='descriptions', nullable=False,
- info=dict(description="A description of the class", format='plaintext'))
- name = ProseColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('move_damage_class_prose', MoveDamageClass, 'prose',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+ description = Column(Unicode(64), nullable=False,
+ info=dict(description="A description of the class", format='plaintext')),
+)
class MoveEffect(TableBase):
u"""An effect of a move
__singlename__ = 'move_effect'
id = Column(Integer, primary_key=True, nullable=False,
info=dict(description="A numeric ID"))
- short_effect = ProseColumn(Unicode(256), plural='short_effects', nullable=False,
- info=dict(description="A short summary of the effect", format='plaintext'))
- effect = ProseColumn(Unicode(5120), plural='effects', nullable=False,
- info=dict(description="A detailed description of the effect", format='plaintext'))
+
+create_translation_table('move_effect_prose', MoveEffect, 'prose',
+ short_effect = Column(Unicode(256), nullable=False,
+ info=dict(description="A short summary of the effect", format='plaintext')),
+ effect = Column(Unicode(5120), nullable=False,
+ info=dict(description="A detailed description of the effect", format='plaintext')),
+)
class MoveEffectChangelog(TableBase):
"""History of changes to move effects across main game versions."""
info=dict(description="The ID of the effect that changed"))
changed_in_version_group_id = Column(Integer, ForeignKey('version_groups.id'), nullable=False,
info=dict(description="The ID of the version group in which the effect changed"))
- effect = ProseColumn(markdown.MarkdownColumn(512), plural='effects', nullable=False,
- info=dict(description="A description of the old behavior", format='markdown'))
__table_args__ = (
UniqueConstraint(effect_id, changed_in_version_group_id),
{},
)
+create_translation_table('move_effect_changelog_prose', MoveEffectChangelog, 'prose',
+ effect = Column(markdown.MarkdownColumn(512), nullable=False,
+ info=dict(description="A description of the old behavior", format='markdown')),
+)
+
class MoveFlag(TableBase):
u"""Maps a move flag to a move
"""
info=dict(description="A numeric ID"))
identifier = Column(Unicode(32), nullable=False,
info=dict(description="A short identifier for the flag", format='identifier'))
- description = ProseColumn(markdown.MarkdownColumn(128), plural='descriptions', nullable=False,
- info=dict(description="A short description of the flag", format='markdown'))
- name = ProseColumn(Unicode(32), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
-class MoveFlavorText(TableBase, LanguageSpecific):
+create_translation_table('move_flag_type_prose', MoveFlagType, 'prose',
+ name = Column(Unicode(32), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+ description = Column(markdown.MarkdownColumn(128), nullable=False,
+ info=dict(description="A short description of the flag", format='markdown')),
+)
+
+class MoveFlavorText(TableBase):
u"""In-game description of a move
"""
__tablename__ = 'move_flavor_text'
info=dict(description="ID of the move"))
version_group_id = Column(Integer, ForeignKey('version_groups.id'), primary_key=True, nullable=False, autoincrement=False,
info=dict(description="ID of the version group this text appears in"))
+ language_id = Column(Integer, ForeignKey('languages.id'), primary_key=True, nullable=False,
+ info=dict(description="The language"))
flavor_text = Column(Unicode(255), nullable=False,
info=dict(description="The flavor text", official=True, format='gametext'))
info=dict(description="A numeric ID"))
identifier = Column(Unicode(24), nullable=False,
info=dict(description="An identifier", format='identifier'))
- name = TextColumn(Unicode(24), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('move_meta_ailment_texts', MoveMetaAilment, 'names',
+ name = Column(Unicode(24), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class MoveMetaCategory(TableBase):
u"""Very general categories that loosely group move effects."""
__singlename__ = 'move_meta_category'
id = Column(Integer, primary_key=True, nullable=False,
info=dict(description="A numeric ID"))
- description = ProseColumn(Unicode(64), plural='descriptions', nullable=False,
- info=dict(description="A description of the category"))
+
+create_translation_table('move_meta_category_prose', MoveMetaCategory, 'prose',
+ description = Column(Unicode(64), nullable=False,
+ info=dict(description="A description of the category")),
+)
class MoveMetaStatChange(TableBase):
u"""Stat changes moves (may) make."""
info=dict(description="A numeric ID"))
identifier = Column(Unicode(32), nullable=False,
info=dict(description="An identifier", format='identifier'))
- description = ProseColumn(Unicode(128), plural='descriptions', nullable=False,
- info=dict(description="A description", format='plaintext'))
- name = ProseColumn(Unicode(32), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('move_target_prose', MoveTarget, 'prose',
+ name = Column(Unicode(32), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+ description = Column(Unicode(128), nullable=False,
+ info=dict(description="A description", format='plaintext')),
+)
class Move(TableBase):
u"""A Move: technique or attack a Pokémon can learn to use
info=dict(description="ID of the move's Contest effect"))
super_contest_effect_id = Column(Integer, ForeignKey('super_contest_effects.id'), nullable=True,
info=dict(description="ID of the move's Super Contest effect"))
- name = TextColumn(Unicode(24), nullable=False, index=True, plural='names',
+
+create_translation_table('move_texts', Move, 'names',
+ name = Column(Unicode(24), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True))
+)
+
class MoveChangelog(TableBase):
"""History of changes to moves across main game versions."""
info=dict(description=u"ID of the Berry flavor the Pokémon hates (if likes_flavor_id is the same, the effects cancel out)"))
likes_flavor_id = Column(Integer, ForeignKey('contest_types.id'), nullable=False,
info=dict(description=u"ID of the Berry flavor the Pokémon likes (if hates_flavor_id is the same, the effects cancel out)"))
- name = TextColumn(Unicode(8), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
@property
def is_neutral(self):
"""
return self.increased_stat_id == self.decreased_stat_id
+create_translation_table('nature_texts', Nature, 'names',
+ name = Column(Unicode(8), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
+
class NatureBattleStylePreference(TableBase):
u"""Battle Palace move preference
info=dict(description="A numeric ID"))
identifier = Column(Unicode(8), nullable=False,
info=dict(description="An identifier", format='identifier'))
- name = TextColumn(Unicode(8), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('pokeathlon_stat_texts', PokeathlonStat, 'names',
+ name = Column(Unicode(8), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class Pokedex(TableBase):
u"""A collection of Pokémon species ordered in a particular way
info=dict(description=u"ID of the region this Pokédex is used in, or None if it's global"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
- description = ProseColumn(Unicode(512), plural='descriptions', nullable=False,
- info=dict(description=u"A longer description of the Pokédex", format='plaintext'))
- name = ProseColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('pokedex_prose', Pokedex, 'prose',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+ description = Column(Unicode(512), nullable=False,
+ info=dict(description=u"A longer description of the Pokédex", format='plaintext')),
+)
class Pokemon(TableBase):
u"""A species of Pokémon. The core to this whole mess.
info=dict(description=u"The height of the Pokémon, in decimeters (tenths of a meter)"))
weight = Column(Integer, nullable=False,
info=dict(description=u"The weight of the Pokémon, in tenths of a kilogram (decigrams)"))
- species = TextColumn(Unicode(16), nullable=False, plural='species_names',
- info=dict(description=u'The short flavor text, such as "Seed" or "Lizard"; usually affixed with the word "Pokémon"',
- official=True, format='plaintext'))
color_id = Column(Integer, ForeignKey('pokemon_colors.id'), nullable=False,
info=dict(description=u"ID of this Pokémon's Pokédex color, as used for a gimmick search function in the games."))
pokemon_shape_id = Column(Integer, ForeignKey('pokemon_shapes.id'), nullable=True,
info=dict(description=u"Set iff the species exhibits enough sexual dimorphism to have separate sets of sprites in Gen IV and beyond."))
order = Column(Integer, nullable=False, index=True,
info=dict(description=u"Order for sorting. Almost national order, except families and forms are grouped together."))
- name = TextColumn(Unicode(20), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
### Stuff to handle alternate Pokémon forms
else:
return None
+create_translation_table('pokemon_texts', Pokemon, 'names',
+ name = Column(Unicode(20), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+ species = Column(Unicode(16), nullable=False,
+ info=dict(description=u'The short flavor text, such as "Seed" or "Lizard"; usually affixed with the word "Pokémon"',
+ official=True, format='plaintext')),
+)
+
class PokemonAbility(TableBase):
u"""Maps an ability to a Pokémon that can have it
"""
info=dict(description=u"ID of the Pokémon"))
identifier = Column(Unicode(6), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
- name = TextColumn(Unicode(6), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('pokemon_color_texts', PokemonColor, 'names',
+ name = Column(Unicode(6), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class PokemonDexNumber(TableBase):
u"""The number of a Pokémon in a particular Pokédex (e.g. Jigglypuff is #138 in Hoenn's 'dex)
trade_pokemon_id = Column(Integer, ForeignKey('pokemon.id'), nullable=True,
info=dict(description=u"The ID of the Pokémon for which this Pokémon must be traded."))
-class PokemonFlavorText(TableBase, LanguageSpecific):
+class PokemonFlavorText(TableBase):
u"""In-game Pokédex descrption of a Pokémon.
"""
__tablename__ = 'pokemon_flavor_text'
info=dict(description=u"ID of the Pokémon"))
version_id = Column(Integer, ForeignKey('versions.id'), primary_key=True, nullable=False, autoincrement=False,
info=dict(description=u"ID of the version that has this flavor text"))
+ language_id = Column(Integer, ForeignKey('languages.id'), primary_key=True, nullable=False,
+ info=dict(description="The language"))
flavor_text = Column(Unicode(255), nullable=False,
info=dict(description=u"ID of the version that has this flavor text", official=True, format='gametext'))
info=dict(description=u'Set for exactly one form used as the default for each species.'))
order = Column(Integer, nullable=False, autoincrement=False,
info=dict(description=u'The order in which forms should be sorted. Multiple forms may have equal order, in which case they should fall back on sorting by name.'))
- name = TextColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
@property
def pokemon(self):
else:
return self.form_base_pokemon.name
+create_translation_table('pokemon_form_texts', PokemonForm, 'names',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
+
class PokemonFormGroup(TableBase):
u"""Information about a Pokémon's forms as a group."""
__tablename__ = 'pokemon_form_groups'
__singlename__ = 'pokemon_form_group'
pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, nullable=False, autoincrement=False,
info=dict(description=u"ID of the base form Pokémon"))
- term = ProseColumn(Unicode(16), plural='terms', nullable=True,
- info=dict(description=u"The term for this Pokémon's forms, e.g. \"Cloak\" for Burmy or \"Forme\" for Deoxys.", official=True, format='plaintext'))
is_battle_only = Column(Boolean, nullable=False,
info=dict(description=u"Set iff the forms only change in battle"))
- description = ProseColumn(markdown.MarkdownColumn(1024), plural='descriptions', nullable=False,
- info=dict(description=u"Description of how the forms work", format='markdown'))
+# FIXME remooove
PokemonFormGroup.id = PokemonFormGroup.pokemon_id
+create_translation_table('pokemon_form_group_prose', PokemonFormGroup, 'prose',
+ term = Column(Unicode(16), nullable=True,
+ info=dict(description=u"The term for this Pokémon's forms, e.g. \"Cloak\" for Burmy or \"Forme\" for Deoxys.", official=True, format='plaintext')),
+ description = Column(markdown.MarkdownColumn(1024), nullable=False,
+ info=dict(description=u"Description of how the forms work", format='markdown')),
+)
+
class PokemonFormPokeathlonStat(TableBase):
u"""A Pokémon form's performance in one Pokéathlon stat."""
__tablename__ = 'pokemon_form_pokeathlon_stats'
info=dict(description=u"A numeric ID"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
- name = TextColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('pokemon_habitat_texts', PokemonHabitat, 'names',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class PokemonInternalID(TableBase):
u"""The number of a Pokémon a game uses internally
info=dict(description=u"A numeric ID"))
identifier = Column(Unicode(64), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
- description = ProseColumn(Unicode(255), plural='descriptions', nullable=False,
- info=dict(description=u"A detailed description of how the method works", format='plaintext'))
- name = ProseColumn(Unicode(64), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('pokemon_move_method_prose', PokemonMoveMethod, 'prose',
+ name = Column(Unicode(64), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+ description = Column(Unicode(255), nullable=False,
+ info=dict(description=u"A detailed description of how the method works", format='plaintext')),
+)
+
class PokemonShape(TableBase):
u"""The shape of a Pokémon's body, as used in generation IV Pokédexes.
info=dict(description=u"A numeric ID"))
identifier = Column(Unicode(24), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
- awesome_name = ProseColumn(Unicode(16), plural='awesome_names', nullable=False,
- info=dict(description=u"A splendiferous name of the body shape", format='plaintext'))
- name = ProseColumn(Unicode(24), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=False))
+
+create_translation_table('pokemon_shape_prose', PokemonShape, 'prose',
+ name = Column(Unicode(24), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=False)),
+ awesome_name = Column(Unicode(16), nullable=False,
+ info=dict(description=u"A splendiferous name of the body shape", format='plaintext')),
+)
class PokemonStat(TableBase):
u"""A stat value of a Pokémon
info=dict(description=u"A numeric ID"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
- name = TextColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('region_texts', Region, 'names',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class Stat(TableBase):
u"""A Stat, such as Attack or Speed
info=dict(description=u"For offensive and defensive stats, the damage this stat relates to; otherwise None (the NULL value)"))
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
- name = TextColumn(Unicode(16), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('stat_texts', Stat, 'names',
+ name = Column(Unicode(16), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class StatHint(TableBase):
u"""Flavor text for genes that appears in a Pokémon's summary. Sometimes
info=dict(description=u"ID of the highest stat"))
gene_mod_5 = Column(Integer, nullable=False, index=True,
info=dict(description=u"Value of the highest stat modulo 5"))
- message = TextColumn(Unicode(24), plural='messages', nullable=False, index=True, unique=True,
- info=dict(description=u"The text displayed", official=True, format='plaintext'))
+
+create_translation_table('stat_hint_texts', StatHint, 'names',
+ message = Column(Unicode(24), nullable=False, index=True,
+ info=dict(description=u"The text displayed", official=True, format='plaintext')),
+)
class SuperContestCombo(TableBase):
u"""Combo of two moves in a Super Contest.
info=dict(description=u"This effect's unique ID."))
appeal = Column(SmallInteger, nullable=False,
info=dict(description=u"The number of hearts the user gains."))
- flavor_text = ProseColumn(Unicode(64), plural='flavor_texts', nullable=False,
- info=dict(description=u"A description of the effect.", format='plaintext'))
+
+create_translation_table('super_contest_effect_prose', SuperContestEffect, 'prose',
+ flavor_text = Column(Unicode(64), nullable=False,
+ info=dict(description=u"A description of the effect.", format='plaintext')),
+)
+
class TypeEfficacy(TableBase):
u"""The damage multiplier used when a move of a particular type damages a
info=dict(description=u"The ID of the generation this type first appeared in."))
damage_class_id = Column(Integer, ForeignKey('move_damage_classes.id'), nullable=True,
info=dict(description=u"The ID of the damage class this type's moves had before Generation IV, null if not applicable (e.g. ???)."))
- name = TextColumn(Unicode(12), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('type_texts', Type, 'names',
+ name = Column(Unicode(12), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
class VersionGroup(TableBase):
u"""A group of versions, containing either two paired versions (such as Red
info=dict(description=u"The ID of the version group this game belongs to."))
identifier = Column(Unicode(32), nullable=False,
info=dict(description=u'And identifier', format='identifier'))
- name = TextColumn(Unicode(32), nullable=False, index=True, plural='names',
- info=dict(description="The name", format='plaintext', official=True))
+
+create_translation_table('version_texts', Version, 'names',
+ name = Column(Unicode(32), nullable=False, index=True,
+ info=dict(description="The name", format='plaintext', official=True)),
+)
### Relations down here, to avoid ordering problems
Ability.all_pokemon = relation(Pokemon,
secondary=PokemonAbility.__table__,
order_by=Pokemon.order,
- back_populates='all_abilities',
+ #back_populates='all_abilities',
)
Ability.pokemon = relation(Pokemon,
secondary=PokemonAbility.__table__,
PokemonAbility.is_dream == False
),
order_by=Pokemon.order,
- back_populates='abilities',
+ #back_populates='abilities',
)
Ability.dream_pokemon = relation(Pokemon,
secondary=PokemonAbility.__table__,
PokemonAbility.is_dream == True
),
order_by=Pokemon.order,
- back_populates='dream_ability',
+ #back_populates='dream_ability',
)
AbilityChangelog.changed_in = relation(VersionGroup, backref='ability_changelog')
AbilityFlavorText.version_group = relation(VersionGroup)
+AbilityFlavorText.language = relation(Language)
Berry.berry_firmness = relation(BerryFirmness, backref='berries')
Berry.firmness = association_proxy('berry_firmness', 'name')
EvolutionChain.growth_rate = relation(GrowthRate, backref='evolution_chains')
EvolutionChain.baby_trigger_item = relation(Item, backref='evolution_chains')
-EvolutionChain.pokemon = relation(Pokemon, order_by=Pokemon.order, back_populates='evolution_chain')
+EvolutionChain.pokemon = relation(Pokemon, order_by=Pokemon.order)#, back_populates='evolution_chain')
Experience.growth_rate = relation(GrowthRate, backref='experience_table')
ItemCategory.pocket = relation(ItemPocket)
ItemFlavorText.version_group = relation(VersionGroup)
+ItemFlavorText.language = relation(Language)
ItemInternalID.item = relation(Item, backref='internal_ids')
ItemInternalID.generation = relation(Generation)
Move.super_contest_combo_next = association_proxy('super_contest_combo_first', 'second')
Move.super_contest_combo_prev = association_proxy('super_contest_combo_second', 'first')
Move.target = relation(MoveTarget, backref='moves')
-Move.type = relation(Type, back_populates='moves')
+Move.type = relation(Type)#, back_populates='moves')
MoveChangelog.changed_in = relation(VersionGroup, backref='move_changelog')
MoveChangelog.move_effect = relation(MoveEffect, backref='move_changelog')
MoveFlag.flag = relation(MoveFlagType)
MoveFlavorText.version_group = relation(VersionGroup)
+MoveFlavorText.language = relation(Language)
MoveMeta.category = relation(MoveMetaCategory, backref='move_meta')
MoveMeta.ailment = relation(MoveMetaAilment, backref='move_meta')
NaturePokeathlonStat.pokeathlon_stat = relation(PokeathlonStat, backref='nature_effects')
Pokedex.region = relation(Region, backref='pokedexes')
-Pokedex.version_groups = relation(VersionGroup, order_by=VersionGroup.id, back_populates='pokedex')
+Pokedex.version_groups = relation(VersionGroup, order_by=VersionGroup.id)#, back_populates='pokedex')
Pokemon.all_abilities = relation(Ability,
secondary=PokemonAbility.__table__,
Pokemon.egg_groups = relation(EggGroup, secondary=PokemonEggGroup.__table__,
order_by=PokemonEggGroup.egg_group_id,
backref=backref('pokemon', order_by=Pokemon.order))
-Pokemon.evolution_chain = relation(EvolutionChain, back_populates='pokemon')
+Pokemon.evolution_chain = relation(EvolutionChain)#, back_populates='pokemon')
Pokemon.child_pokemon = relation(Pokemon,
primaryjoin=Pokemon.id==PokemonEvolution.from_pokemon_id,
secondary=PokemonEvolution.__table__,
Pokemon.stats = relation(PokemonStat, backref='pokemon', order_by=PokemonStat.stat_id.asc())
Pokemon.types = relation(Type, secondary=PokemonType.__table__,
order_by=PokemonType.slot.asc(),
- back_populates='pokemon')
+ )#back_populates='pokemon')
PokemonDexNumber.pokedex = relation(Pokedex)
)
PokemonFlavorText.version = relation(Version)
+PokemonFlavorText.language = relation(Language)
PokemonForm.form_base_pokemon = relation(Pokemon, primaryjoin=PokemonForm.form_base_pokemon_id==Pokemon.id)
PokemonForm.unique_pokemon = relation(Pokemon, backref=backref('unique_form', uselist=False),
Type.damage_class = relation(MoveDamageClass, backref='types')
Type.pokemon = relation(Pokemon, secondary=PokemonType.__table__,
order_by=Pokemon.order,
- back_populates='types')
+ )#back_populates='types')
Type.moves = relation(Move, back_populates='type', order_by=Move.id)
-Version.version_group = relation(VersionGroup, back_populates='versions')
+Version.version_group = relation(VersionGroup)#, back_populates='versions')
Version.generation = association_proxy('version_group', 'generation')
VersionGroup.versions = relation(Version, order_by=Version.id, back_populates='version_group')
VersionGroup.regions = association_proxy('version_group_regions', 'region')
VersionGroup.pokedex = relation(Pokedex, back_populates='version_groups')
+Move.effect = markdown.MoveEffectProperty('effect')
+Move.effect_map = markdown.MoveEffectProperty('effect_map')
+Move.short_effect = markdown.MoveEffectProperty('short_effect')
+Move.short_effect_map = markdown.MoveEffectProperty('short_effect_map')
-### Add text/prose tables
-
-default_lang = u'en'
-
-def makeTextTable(object_table, name_plural, name_singular, columns, lazy):
- # With "Language", we'd have two language_id. So, rename one to 'lang'
- safe_name = object_table.__singlename__
- if safe_name == 'language':
- safe_name = 'lang'
-
- tablename = object_table.__singlename__ + '_' + name_plural
- singlename = object_table.__singlename__ + '_' + name_singular
-
- class Strings(object):
- __tablename__ = tablename
- __singlename__ = singlename
- _attrname = name_plural
- _language_identifier = association_proxy('language', 'identifier')
-
- for name, plural, column in columns:
- column.name = name
- if not column.nullable:
- # A Python side default value, so that the strings can be set
- # one by one without the DB complaining about missing values
- column.default = ColumnDefault(u'')
-
- table = Table(tablename, metadata,
- Column(safe_name + '_id', Integer, ForeignKey(object_table.id),
- primary_key=True, nullable=False),
- Column('language_id', Integer, ForeignKey(Language.id),
- primary_key=True, index=True, nullable=False),
- *(column for name, plural, column in columns)
- )
-
- mapper(Strings, table,
- properties={
- "object_id": synonym(safe_name + '_id'),
- "language": relation(Language,
- primaryjoin=table.c.language_id == Language.id,
- ),
- safe_name: relation(object_table,
- primaryjoin=(object_table.id == table.c[safe_name + "_id"]),
- backref=backref(name_plural,
- collection_class=attribute_mapped_collection('language'),
- lazy=lazy,
- ),
- ),
- },
- )
-
- # The relation to the object
- Strings.object = getattr(Strings, safe_name)
-
- # Link the tables themselves, so we can get them if needed
- Strings.object_table = object_table
- setattr(object_table, name_singular + '_table', Strings)
-
- for colname, pluralname, column in columns:
- # Provide a property with all the names, and an English accessor
- # for backwards compatibility
- setattr(object_table, pluralname, StringProperty(
- object_table, Strings, colname,
- ))
- setattr(object_table, colname, DefaultLangProperty(pluralname))
-
- if colname == 'name':
- object_table.name_table = Strings
-
- return Strings
-
-class StringProperty(object):
- def __init__(self, cls, stringclass, colname):
- self.cls = cls
- self.colname = colname
- self.stringclass = stringclass
-
- def __get__(self, instance, cls):
- if instance:
- return StringMapping(instance, self)
- else:
- return self
-
- def __getitem__(self, lang):
- return StringExpression(self, lang)
-
- def __str__(self):
- return '<StringDict %s.%s>' % (self.cls, self.colname)
-
-class StringMapping(collections.MutableMapping):
- def __init__(self, instance, prop):
- self.stringclass = prop.stringclass
- self.instance = instance
- self.strings = getattr(instance, prop.stringclass._attrname)
- self.colname = prop.colname
-
- def __len__(self):
- return len(self.strings)
-
- def __iter__(self):
- return iter(self.strings)
-
- def __contains__(self, lang):
- return lang in self.strings
-
- def __getitem__(self, lang):
- return getattr(self.strings[lang], self.colname)
-
- def __setitem__(self, lang, value):
- try:
- # Modifying an existing row
- row = self.strings[lang]
- except KeyError:
- # We need do add a whole row for the language
- row = self.stringclass()
- row.object_id = self.instance.id
- session = object_session(self.instance)
- if isinstance(lang, basestring):
- lang = session.query(Language).filter_by(
- identifier=lang).one()
- row.language = lang
- self.strings[lang] = row
- session.add(row)
- return setattr(row, self.colname, value)
-
- def __delitem__(self, lang):
- raise NotImplementedError('Cannot delete a single string. '
- 'Perhaps you wan to delete all of %s.%s?' %
- (self.instance, self.stringclass._attrname)
- )
-
-class StringExpression(ColumnOperators):
- def __init__(self, prop, lang):
- self.prop = prop
- self.column = getattr(prop.stringclass, prop.colname)
- self.lang_column = prop.stringclass._language_identifier
- if isinstance(lang, basestring):
- self.lang = lang
- else:
- self.lang = lang.identifier
-
- def operate(self, op, *values, **kwargs):
- return getattr(self.prop.cls, self.prop.stringclass._attrname).any(and_(
- self.lang_column == self.lang,
- op(self.column, *values, **kwargs),
- ))
-
-class DefaultLangProperty(object):
- def __init__(self, colname):
- self.colname = colname
-
- def __get__(self, instance, cls):
- if instance:
- return getattr(instance, self.colname)[default_lang]
- else:
- return getattr(cls, self.colname)[default_lang]
-
- def __set__(self, instance, value):
- getattr(instance, self.colname)[default_lang] = value
-
- def __delete__(self, instance):
- del getattr(instance, self.colname)[default_lang]
-
-for table in list(table_classes):
- # Find all the language-specific columns, keeping them in the order they
- # were defined
- all_columns = []
- for colname in dir(table):
- column = getattr(table, colname)
- if isinstance(column, LanguageSpecificColumn):
- all_columns.append((colname, column))
- delattr(table, colname)
- all_columns.sort(key=lambda pair: pair[1].order)
-
- # Break them into text and prose columns
- text_columns = []
- prose_columns = []
- for colname, column in all_columns:
- spec = colname, column.plural, column.makeSAColumn()
- if isinstance(column, TextColumn):
- text_columns.append(spec)
- elif isinstance(column, ProseColumn):
- prose_columns.append(spec)
-
- if (text_columns or prose_columns) and issubclass(table, LanguageSpecific):
- raise AssertionError("Language-specific table %s shouldn't have explicit language-specific columns" % table)
-
- if text_columns:
- string_table = makeTextTable(table, 'texts', 'text', text_columns, lazy=False)
- if prose_columns:
- string_table = makeTextTable(table, 'prose', 'prose', prose_columns, lazy=True)
-
-### Add language relations
-for table in list(table_classes):
- if issubclass(table, LanguageSpecific):
- table.language = relation(Language, primaryjoin=table.language_id == Language.id)
-
-Move.effect = DefaultLangProperty('effects')
-Move.effects = markdown.MoveEffectsProperty('effect')
-Move.short_effect = DefaultLangProperty('short_effects')
-Move.short_effects = markdown.MoveEffectsProperty('short_effect')
-
-MoveChangelog.effect = DefaultLangProperty('effects')
-MoveChangelog.effects = markdown.MoveEffectsProperty('effect')
-MoveChangelog.short_effect = DefaultLangProperty('short_effects')
-MoveChangelog.short_effects = markdown.MoveEffectsProperty('short_effect')
+MoveChangelog.effect = markdown.MoveEffectProperty('effect')
+MoveChangelog.effect_map = markdown.MoveEffectProperty('effect_map')
+MoveChangelog.short_effect = markdown.MoveEffectProperty('short_effect')
+MoveChangelog.short_effect_map = markdown.MoveEffectProperty('short_effect_map')
projects / zzz-pokedex.git / blobdiff