from functools import partial
from sqlalchemy import Column, ForeignKey, MetaData, PrimaryKeyConstraint, Table, UniqueConstraint
-from sqlalchemy.ext.declarative import (
- declarative_base, declared_attr, DeclarativeMeta,
- )
+from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.ext.associationproxy import association_proxy
-from sqlalchemy.orm import (
- backref, compile_mappers, eagerload_all, relation, class_mapper, synonym, mapper,
- )
-from sqlalchemy.orm.session import Session, object_session
+from sqlalchemy.orm import backref, relation
+from sqlalchemy.orm.session import Session
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, bindparam
from sqlalchemy.schema import ColumnDefault
from sqlalchemy.types import *
-from inspect import isclass
from pokedex.db import markdown, multilang
-# 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)
-
class TableSuperclass(object):
"""Superclass for declarative tables, to give them some generic niceties
like stringification.
return unicode(self).encode('utf8')
metadata = MetaData()
-TableBase = declarative_base(metadata=metadata, cls=TableSuperclass, metaclass=TableMetaclass)
-
-### 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"))
+TableBase = declarative_base(metadata=metadata, cls=TableSuperclass)
### Need Language first, to create the partial() below
create_translation_table = partial(multilang.create_translation_table, language_class=Language)
-create_translation_table('language_texts', Language, 'names',
+create_translation_table('language_names', Language, 'names',
name = Column(Unicode(16), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
generation_id = Column(Integer, ForeignKey('generations.id'), nullable=False,
info=dict(description="The ID of the generation this ability was introduced in", detail=True))
-create_translation_table('ability_texts', Ability, 'names',
+create_translation_table('ability_names', Ability, 'names',
name = Column(Unicode(24), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
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'))
identifier = Column(Unicode(10), nullable=False,
info=dict(description="An identifier", format='identifier'))
-create_translation_table('berry_firmness_texts', BerryFirmness, 'names',
+create_translation_table('berry_firmness_names', BerryFirmness, 'names',
name = Column(Unicode(10), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
info=dict(description="The base number of hearts the user's opponent loses"))
create_translation_table('contest_effect_prose', ContestEffect, 'prose',
- effect = Column(Unicode(255), nullable=False,
- info=dict(description="A detailed description of the effect", format='plaintext')),
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):
identifier = Column(Unicode(6), nullable=False,
info=dict(description="An identifier", format='identifier'))
-create_translation_table('contest_type_texts', ContestType, 'names',
+create_translation_table('contest_type_names', 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')),
- name = Column(Unicode(6), nullable=False, index=True,
- info=dict(description="The name", format='plaintext', official=True)),
)
class EggGroup(TableBase):
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u'An identifier', format='identifier'))
-create_translation_table('generation_texts', Generation, 'names',
+create_translation_table('generation_names', Generation, 'names',
name = Column(Unicode(16), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
"""
return any(flag.identifier == u'underground' for flag in self.flags)
-create_translation_table('item_texts', Item, 'names',
+create_translation_table('item_names', Item, 'names',
name = Column(Unicode(20), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
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'))
info=dict(description="Description of the effect", format='plaintext')),
)
-class ItemInternalID(TableBase):
+class ItemGameIndex(TableBase):
u"""The internal ID number a game uses for an item
"""
- __tablename__ = 'item_internal_ids'
+ __tablename__ = 'item_game_indices'
item_id = Column(Integer, ForeignKey('items.id'), primary_key=True, autoincrement=False, nullable=False,
info=dict(description="The database ID of the item"))
generation_id = Column(Integer, ForeignKey('generations.id'), primary_key=True, autoincrement=False, nullable=False,
info=dict(description="ID of the generation of games"))
- internal_id = Column(Integer, nullable=False,
+ game_index = Column(Integer, nullable=False,
info=dict(description="Internal ID of the item in the generation"))
class ItemPocket(TableBase):
identifier = Column(Unicode(16), nullable=False,
info=dict(description="An identifier of this pocket", format='identifier'))
-create_translation_table('item_pocket_texts', ItemPocket, 'names',
+create_translation_table('item_pocket_names', ItemPocket, 'names',
name = Column(Unicode(16), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
identifier = Column(Unicode(64), nullable=False,
info=dict(description="An identifier", format='identifier'))
-create_translation_table('location_texts', Location, 'names',
+create_translation_table('location_names', Location, 'names',
name = Column(Unicode(64), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
info=dict(description="A numeric ID"))
location_id = Column(Integer, ForeignKey('locations.id'), nullable=False,
info=dict(description="ID of the location this area is part of"))
- internal_id = Column(Integer, nullable=False,
+ game_index = Column(Integer, nullable=False,
info=dict(description="ID the games ude for this area"))
identifier = Column(Unicode(64), nullable=True,
info=dict(description="An identifier", format='identifier'))
rate = Column(Integer, nullable=True,
info=dict(description="The encounter rate")) # units?
-class LocationInternalID(TableBase):
+class LocationGameIndex(TableBase):
u"""IDs the games use internally for locations
"""
- __tablename__ = 'location_internal_ids'
+ __tablename__ = 'location_game_indices'
location_id = Column(Integer, ForeignKey('locations.id'), nullable=False, primary_key=True,
info=dict(description="Database ID of the locaion"))
generation_id = Column(Integer, ForeignKey('generations.id'), nullable=False, primary_key=True,
info=dict(description="ID of the generation this entry to"))
- internal_id = Column(Integer, nullable=False,
+ game_index = Column(Integer, nullable=False,
info=dict(description="Internal game ID of the location"))
class Machine(TableBase):
info=dict(description="A short description of the flag", format='markdown')),
)
-class MoveFlavorText(TableBase, LanguageSpecific):
+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'))
identifier = Column(Unicode(24), nullable=False,
info=dict(description="An identifier", format='identifier'))
-create_translation_table('move_meta_ailment_texts', MoveMetaAilment, 'names',
+create_translation_table('move_meta_ailment_names', MoveMetaAilment, 'names',
name = Column(Unicode(24), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
super_contest_effect_id = Column(Integer, ForeignKey('super_contest_effects.id'), nullable=True,
info=dict(description="ID of the move's Super Contest effect"))
-create_translation_table('move_texts', Move, 'names',
+create_translation_table('move_names', Move, 'names',
name = Column(Unicode(24), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True))
)
"""
return self.increased_stat_id == self.decreased_stat_id
-create_translation_table('nature_texts', Nature, 'names',
+create_translation_table('nature_names', Nature, 'names',
name = Column(Unicode(8), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
identifier = Column(Unicode(8), nullable=False,
info=dict(description="An identifier", format='identifier'))
-create_translation_table('pokeathlon_stat_texts', PokeathlonStat, 'names',
+create_translation_table('pokeathlon_stat_names', PokeathlonStat, 'names',
name = Column(Unicode(8), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
else:
return None
-create_translation_table('pokemon_texts', Pokemon, 'names',
+create_translation_table('pokemon_names', 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,
identifier = Column(Unicode(6), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
-create_translation_table('pokemon_color_texts', PokemonColor, 'names',
+create_translation_table('pokemon_color_names', PokemonColor, 'names',
name = Column(Unicode(6), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
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'))
else:
return self.form_base_pokemon.name
-create_translation_table('pokemon_form_texts', PokemonForm, 'names',
+create_translation_table('pokemon_form_names', PokemonForm, 'names',
name = Column(Unicode(16), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
-create_translation_table('pokemon_habitat_texts', PokemonHabitat, 'names',
+create_translation_table('pokemon_habitat_names', PokemonHabitat, 'names',
name = Column(Unicode(16), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
-class PokemonInternalID(TableBase):
+class PokemonGameIndex(TableBase):
u"""The number of a Pokémon a game uses internally
"""
- __tablename__ = 'pokemon_internal_ids'
+ __tablename__ = 'pokemon_game_indices'
pokemon_id = Column(Integer, ForeignKey('pokemon.id'), primary_key=True, autoincrement=False, nullable=False,
info=dict(description=u"Database ID of the Pokémon"))
generation_id = Column(Integer, ForeignKey('generations.id'), primary_key=True, autoincrement=False, nullable=False,
info=dict(description=u"Database ID of the generation"))
- internal_id = Column(Integer, nullable=False,
+ game_index = Column(Integer, nullable=False,
info=dict(description=u"Internal ID the generation's games use for the Pokémon"))
class PokemonItem(TableBase):
info=dict(description=u"An identifier", format='identifier'))
create_translation_table('pokemon_shape_prose', PokemonShape, 'prose',
- awesome_name = Column(Unicode(16), nullable=False,
- info=dict(description=u"A splendiferous name of the body shape", format='plaintext')),
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):
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
-create_translation_table('region_texts', Region, 'names',
+create_translation_table('region_names', Region, 'names',
name = Column(Unicode(16), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
identifier = Column(Unicode(16), nullable=False,
info=dict(description=u"An identifier", format='identifier'))
-create_translation_table('stat_texts', Stat, 'names',
+create_translation_table('stat_names', Stat, 'names',
name = Column(Unicode(16), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
gene_mod_5 = Column(Integer, nullable=False, index=True,
info=dict(description=u"Value of the highest stat modulo 5"))
-create_translation_table('stat_hint_texts', StatHint, 'names',
+create_translation_table('stat_hint_names', StatHint, 'names',
message = Column(Unicode(24), nullable=False, index=True,
info=dict(description=u"The text displayed", official=True, format='plaintext')),
)
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. ???)."))
-create_translation_table('type_texts', Type, 'names',
+create_translation_table('type_names', Type, 'names',
name = Column(Unicode(12), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
identifier = Column(Unicode(32), nullable=False,
info=dict(description=u'And identifier', format='identifier'))
-create_translation_table('version_texts', Version, 'names',
+create_translation_table('version_names', Version, 'names',
name = Column(Unicode(32), nullable=False, index=True,
info=dict(description="The name", format='plaintext', official=True)),
)
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')
ItemCategory.pocket = relation(ItemPocket)
ItemFlavorText.version_group = relation(VersionGroup)
+ItemFlavorText.language = relation(Language)
-ItemInternalID.item = relation(Item, backref='internal_ids')
-ItemInternalID.generation = relation(Generation)
+ItemGameIndex.item = relation(Item, backref='game_indices')
+ItemGameIndex.generation = relation(Generation)
ItemPocket.categories = relation(ItemCategory, order_by=ItemCategory.identifier)
LocationArea.location = relation(Location, backref='areas')
-LocationInternalID.location = relation(Location, backref='internal_ids')
-LocationInternalID.generation = relation(Generation)
+LocationGameIndex.location = relation(Location, backref='game_indices')
+LocationGameIndex.generation = relation(Generation)
Machine.item = relation(Item)
Machine.version_group = relation(VersionGroup)
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')
)
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),
VersionGroup.regions = association_proxy('version_group_regions', 'region')
VersionGroup.pokedex = relation(Pokedex, back_populates='version_groups')
-
-### 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 = markdown.MoveEffectProperty('effect')
Move.effect_map = markdown.MoveEffectProperty('effect_map')
Move.short_effect = markdown.MoveEffectProperty('short_effect')
projects / zzz-pokedex.git / blobdiff