Merge pull request #66 from encukou/internal-ids

Add internal IDs to forms

pokedex/db/tables.py

@@ -643,7 +643,7 @@ class LocationArea(TableBase):
     location_id = Column(Integer, ForeignKey('locations.id'), nullable=False,
         info=dict(description="ID of the location this area is part of"))
     game_index = Column(Integer, nullable=False,
-        info=dict(description="ID the games ude for this area"))
+        info=dict(description="ID the games use for this area"))
     identifier = Column(Unicode(64), nullable=True,
         info=dict(description="An identifier", format='identifier'))
 
@@ -1249,6 +1249,16 @@ create_translation_table('pokemon_form_names', PokemonForm, 'names',
         info=dict(description=u"The full pokémon name, e.g. 'Sky Shaymin', for pokémon with different forms", format='plaintext', official=True)),
 )
 
+class PokemonFormGeneration(TableBase):
+    u"""Links Pokémon forms to the generations they exist in"""
+    __tablename__ = 'pokemon_form_generations'
+    pokemon_form_id = Column(Integer, ForeignKey('pokemon_forms.id'), primary_key=True, nullable=False, autoincrement=False,
+        info=dict(description=u'The ID of the Pokémon form.'))
+    generation_id = Column(Integer, ForeignKey('generations.id'), primary_key=True, nullable=False, autoincrement=False,
+        info=dict(description=u'The ID of the generation.'))
+    game_index = Column(Integer, nullable=False,
+        info=dict(description=u'The internal ID the games use for this form.'))
+
 class PokemonFormPokeathlonStat(TableBase):
     u"""A Pokémon form's performance in one Pokéathlon stat."""
     __tablename__ = 'pokemon_form_pokeathlon_stats'
@@ -1990,6 +2000,13 @@ PokemonForm.pokeathlon_stats = relationship(PokemonFormPokeathlonStat,
 PokemonFormPokeathlonStat.pokeathlon_stat = relationship(PokeathlonStat,
     innerjoin=True, lazy='joined')
 
+PokemonFormGeneration.form = relationship(PokemonForm,
+    backref=backref('pokemon_form_generations',
+        order_by=PokemonFormGeneration.generation_id))
+PokemonFormGeneration.generation = relationship(Generation,
+    backref=backref('pokemon_form_generations',
+        order_by=PokemonFormGeneration.game_index))
+
 PokemonItem.item = relationship(Item,
     innerjoin=True, lazy='joined',
     backref='pokemon')

scripts/fill-pokemon_form_generations.py

@@ -0,0 +1,143 @@
+# Encoding: UTF-8
+"""Fill the pokemon_form_generations table
+
+This is an unmaintained one-shot script, only included in the repo for reference.
+
+
+"""
+
+
+from sqlalchemy.sql import exists, func
+from sqlalchemy.orm import lazyload, eagerload, eagerload_all
+from sqlalchemy import and_, or_, not_
+
+from pokedex.db import connect, tables, load
+
+session = connect()
+
+session.query(tables.PokemonFormGeneration).delete()
+
+generations = list(session.query(tables.Generation).order_by(
+        tables.Generation.id))
+
+q = session.query(tables.PokemonForm)
+q = q.options(eagerload_all('pokemon', 'species'))
+q = q.order_by(tables.PokemonForm.order)
+
+form_orders = dict(
+    unown=list('abcdefghijklmnopqrstuvwxyz') + ['exclamation', 'question'],
+
+    deoxys=['normal', 'attack', 'defense', 'speed'],
+
+    burmy=['plant', 'sandy', 'trash'],
+    wormadam=['plant', 'sandy', 'trash'],
+
+    shellos=['west', 'east'],
+    gastrodon=['west', 'east'],
+
+    rotom=[None, 'heat', 'wash', 'frost', 'fan', 'mow'],
+
+    giratina=['altered', 'origin'],
+
+    shaymin=['land', 'sky'],
+
+    castform=[None, 'sunny', 'rainy', 'snowy'],
+    basculin=['red-striped', 'blue-striped'],
+    darmanitan=['standard', 'zen'],
+    deerling=['spring', 'summer', 'autumn', 'winter'],
+    sawsbuck=['spring', 'summer', 'autumn', 'winter'],
+    meloetta=['aria', 'pirouette'],
+    genesect=[None, 'douse', 'shock', 'burn', 'chill'],
+    cherrim=['overcast', 'sunshine'],
+)
+
+arceus = {4: '''normal fighting flying poison ground rock bug ghost steel
+    unknown fire water grass electric psychic ice dragon dark'''.split()}
+arceus[5] = list(arceus[4])
+arceus[5].remove('unknown')
+
+for form in q:
+    species_ident = form.species.identifier
+    form_ident = form.form_identifier
+    is_default = form.is_default and form.pokemon.is_default
+    print form_ident, species_ident
+    for gen in generations:
+        game_index = None
+        if gen.id >= form.version_group.generation_id:
+            if gen.id < 4:
+                # forms not really implemented yet
+                if species_ident == 'pichu':
+                    if is_default:
+                        game_index = 0
+                    else:
+                        continue
+                elif species_ident in ('unown', 'castform'):
+                    lst = form_orders[species_ident]
+                    game_index = lst.index(form_ident)
+                elif species_ident == 'deoxys':
+                    game_index = 0
+                elif is_default:
+                    game_index = 0
+            else:
+                try:
+                    lst = form_orders[species_ident]
+                except KeyError:
+                    if species_ident == 'pichu' and form_ident == 'spiky-eared':
+                        if gen.id == 4:
+                            game_index = 1
+                        else:
+                            continue
+                    elif species_ident == 'cherrim':
+                        if gen.id < 5:
+                            if is_default:
+                                game_index = 0
+                            else:
+                                continue
+                        else:
+                            lst = ['overcast', 'sunshine']
+                            game_index = lst.index(form_ident)
+                    elif species_ident == 'castform':
+                        if gen.id < 5:
+                            if is_default:
+                                game_index = 0
+                            else:
+                                continue
+                        else:
+                            lst = [None, 'sunny', 'rainy', 'snowy']
+                            game_index = lst.index(form_ident)
+                    elif species_ident == 'arceus':
+                        if gen.id >= 5 and form_ident == 'unknown':
+                            continue
+                        else:
+                            lst = arceus[gen.id]
+                            game_index = lst.index(form_ident)
+                    elif form.is_default and form.pokemon.is_default:
+                        game_index = 0
+                    else:
+                        raise AssertionError()
+                else:
+                    game_index = lst.index(form_ident)
+            obj = tables.PokemonFormGeneration(form=form, generation=gen,
+                    game_index=game_index)
+            session.add(obj)
+
+q = session.query(tables.PokemonFormGeneration)
+for species in session.query(tables.PokemonSpecies).options(
+        eagerload_all('forms', 'pokemon_form_generations')):
+    if len(species.forms) > 1:
+        print species.name
+    for gen in generations:
+        if len(species.forms) == 1:
+            pfg = q.get((species.forms[0].id, gen.id))
+            assert pfg is None or pfg.game_index == 0
+            continue
+        forms = [(q.get((f.id, gen.id)), f) for f in species.forms if q.get((f.id, gen.id))]
+        forms = [(pfg.game_index, f) for pfg, f in forms if pfg]
+        if forms:
+            forms.sort()
+            pl = ["%s=%s" % (gi, f.form_identifier) for gi, f in forms]
+            print '   ', gen.id, ' '.join(pl)
+
+load.dump(session, tables=['pokemon_form_generations'])
+print "Dumped to CSV, rolling back transaction"
+session.rollback()