mirror of
https://github.com/veekun/pokedex.git
synced 2024-08-20 18:16:34 +00:00
parent
216e4c8462
commit
5b7e13892a
3 changed files with 2290 additions and 0 deletions
2130
pokedex/data/csv/pokemon_form_generations.csv
Normal file
2130
pokedex/data/csv/pokemon_form_generations.csv
Normal file
File diff suppressed because it is too large
Load diff
|
@ -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)),
|
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):
|
class PokemonFormPokeathlonStat(TableBase):
|
||||||
u"""A Pokémon form's performance in one Pokéathlon stat."""
|
u"""A Pokémon form's performance in one Pokéathlon stat."""
|
||||||
__tablename__ = 'pokemon_form_pokeathlon_stats'
|
__tablename__ = 'pokemon_form_pokeathlon_stats'
|
||||||
|
@ -1978,6 +1988,13 @@ PokemonForm.pokeathlon_stats = relationship(PokemonFormPokeathlonStat,
|
||||||
PokemonFormPokeathlonStat.pokeathlon_stat = relationship(PokeathlonStat,
|
PokemonFormPokeathlonStat.pokeathlon_stat = relationship(PokeathlonStat,
|
||||||
innerjoin=True, lazy='joined')
|
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,
|
PokemonItem.item = relationship(Item,
|
||||||
innerjoin=True, lazy='joined',
|
innerjoin=True, lazy='joined',
|
||||||
backref='pokemon')
|
backref='pokemon')
|
||||||
|
|
143
scripts/fill-pokemon_form_generations.py
Normal file
143
scripts/fill-pokemon_form_generations.py
Normal file
|
@ -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()
|
Loading…
Reference in a new issue