pokeapi/data/v2/build.py
2020-09-02 19:57:00 +02:00

2312 lines
79 KiB
Python

# To build out the data you'll need to jump into the Django shell
#
# $ python manage.py shell
#
# and run the build script with
#
# $ from data.v2.build import build_all
# $ build_all()
#
# Each time the build script is run it will iterate over each table in the database,
# wipe it and rewrite each row using the data found in data/v2/csv.
import csv
import os
import os.path
import re
import json
from django.db import connection
from pokemon_v2.models import *
# why this way? how about use `__file__`
DATA_LOCATION = "data/v2/csv/"
DATA_LOCATION2 = os.path.join(os.path.dirname(__file__), "csv")
GROUP_RGX = r"\[(.*?)\]\{(.*?)\}"
SUB_RGX = r"\[.*?\]\{.*?\}"
DB_CURSOR = connection.cursor()
DB_VENDOR = connection.vendor
MEDIA_DIR = "/media/sprites/{0}"
IMAGE_DIR = os.getcwd() + "/data/v2/sprites/sprites/"
RESOURCE_IMAGES = []
for root, dirs, files in os.walk(IMAGE_DIR):
for file in files:
image_path = os.path.join(root.replace(IMAGE_DIR, ""), file)
image_path = image_path.replace("\\", "/") # convert Windows-style path to Unix
RESOURCE_IMAGES.append(image_path)
def file_path_or_none(file_name):
return MEDIA_DIR.format(file_name) if file_name in RESOURCE_IMAGES else None
def with_iter(context, iterable=None):
if iterable is None:
iterable = context
with context:
for value in iterable:
yield value
def load_data(file_name):
# with_iter closes the file when it has finished
return csv.reader(
with_iter(open(DATA_LOCATION + file_name, "rt", encoding="utf8")), delimiter=","
)
def clear_table(model):
table_name = model._meta.db_table
model.objects.all().delete()
print("building " + table_name)
# Reset DB auto increments to start at 1
if DB_VENDOR == "sqlite":
DB_CURSOR.execute(
"DELETE FROM sqlite_sequence WHERE name = " + "'" + table_name + "'"
)
else:
DB_CURSOR.execute(
"SELECT setval(pg_get_serial_sequence("
+ "'"
+ table_name
+ "'"
+ ",'id'), 1, false);"
)
def build_generic(model_classes, file_name, csv_record_to_objects):
batches = {}
for model_class in model_classes:
clear_table(model_class)
batches[model_class] = [] # one batch per model class
csv_data = load_data(file_name)
next(csv_data, None) # skip header
for csv_record in csv_data:
for obj in csv_record_to_objects(csv_record):
model_class = type(obj)
batches[model_class].append(obj)
# Limit the batch size
if len(batches[model_class]) > 200:
model_class.objects.bulk_create(batches[model_class])
batches[model_class] = []
for model_class, batch in batches.items():
model_class.objects.bulk_create(batch)
def scrub_str(string):
"""
The purpose of this function is to scrub the weird template mark-up out of strings
that Veekun is using for their pokedex.
Example:
[]{move:dragon-tail} will effect the opponents [HP]{mechanic:hp}.
Becomes:
dragon tail will effect the opponents HP.
If you find this results in weird strings please take a stab at improving or re-writing.
"""
groups = re.findall(GROUP_RGX, string)
for group in groups:
if group[0]:
sub = group[0]
else:
sub = group[1].split(":")
if len(sub) >= 2:
sub = sub[1]
else:
sub = sub[0]
sub = sub.replace("-", " ")
string = re.sub(SUB_RGX, sub, string, 1)
return string
##############
# LANGUAGE #
##############
def _build_languages():
def csv_record_to_objects(info):
yield Language(
id=int(info[0]),
iso639=info[1],
iso3166=info[2],
name=info[3],
official=bool(int(info[4])),
order=info[5],
)
build_generic((Language,), "languages.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield LanguageName(
language_id=int(info[0]), local_language_id=int(info[1]), name=info[2]
)
build_generic((LanguageName,), "language_names.csv", csv_record_to_objects)
############
# REGION #
############
def _build_regions():
def csv_record_to_objects(info):
yield Region(id=int(info[0]), name=info[1])
build_generic((Region,), "regions.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield RegionName(region_id=int(info[0]), language_id=int(info[1]), name=info[2])
build_generic((RegionName,), "region_names.csv", csv_record_to_objects)
################
# GENERATION #
################
def _build_generations():
def csv_record_to_objects(info):
yield Generation(id=int(info[0]), region_id=int(info[1]), name=info[2])
build_generic((Generation,), "generations.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield GenerationName(
generation_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((GenerationName,), "generation_names.csv", csv_record_to_objects)
#############
# VERSION #
#############
def _build_versions():
def csv_record_to_objects(info):
yield VersionGroup(
id=int(info[0]),
name=info[1],
generation_id=int(info[2]),
order=int(info[3]),
)
build_generic((VersionGroup,), "version_groups.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield VersionGroupRegion(version_group_id=int(info[0]), region_id=int(info[1]))
build_generic(
(VersionGroupRegion,), "version_group_regions.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield Version(id=int(info[0]), version_group_id=int(info[1]), name=info[2])
build_generic((Version,), "versions.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield VersionName(
version_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((VersionName,), "version_names.csv", csv_record_to_objects)
##################
# DAMAGE CLASS #
##################
def _build_damage_classes():
def csv_record_to_objects(info):
yield MoveDamageClass(id=int(info[0]), name=info[1])
build_generic((MoveDamageClass,), "move_damage_classes.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveDamageClassName(
move_damage_class_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
yield MoveDamageClassDescription(
move_damage_class_id=int(info[0]),
language_id=int(info[1]),
description=info[3],
)
build_generic(
(MoveDamageClassName, MoveDamageClassDescription),
"move_damage_class_prose.csv",
csv_record_to_objects,
)
###########
# STATS #
###########
def _build_stats():
def csv_record_to_objects(info):
yield Stat(
id=int(info[0]),
move_damage_class_id=int(info[1]) if info[1] != "" else None,
name=info[2],
is_battle_only=bool(int(info[3])),
game_index=int(info[4]) if info[4] else 0,
)
build_generic((Stat,), "stats.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield StatName(stat_id=int(info[0]), language_id=int(info[1]), name=info[2])
build_generic((StatName,), "stat_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokeathlonStat(id=int(info[0]), name=info[1])
build_generic((PokeathlonStat,), "pokeathlon_stats.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokeathlonStatName(
pokeathlon_stat_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic(
(PokeathlonStatName,), "pokeathlon_stat_names.csv", csv_record_to_objects
)
# ###############
# # ABILITIES #
# ###############
def _build_abilities():
def csv_record_to_objects(info):
yield Ability(
id=int(info[0]),
name=info[1],
generation_id=int(info[2]),
is_main_series=bool(int(info[3])),
)
build_generic((Ability,), "abilities.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield AbilityName(
ability_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((AbilityName,), "ability_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield AbilityChange(
id=int(info[0]), ability_id=int(info[1]), version_group_id=int(info[2])
)
build_generic((AbilityChange,), "ability_changelog.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield AbilityEffectText(
ability_id=int(info[0]),
language_id=int(info[1]),
short_effect=scrub_str(info[2]),
effect=scrub_str(info[3]),
)
build_generic((AbilityEffectText,), "ability_prose.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield AbilityChangeEffectText(
ability_change_id=int(info[0]),
language_id=int(info[1]),
effect=scrub_str(info[2]),
)
build_generic(
(AbilityChangeEffectText,), "ability_changelog_prose.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield AbilityFlavorText(
ability_id=int(info[0]),
version_group_id=int(info[1]),
language_id=int(info[2]),
flavor_text=info[3],
)
build_generic(
(AbilityFlavorText,), "ability_flavor_text.csv", csv_record_to_objects
)
####################
# CHARACTERISTIC #
####################
def _build_characteristics():
def csv_record_to_objects(info):
yield Characteristic(
id=int(info[0]), stat_id=int(info[1]), gene_mod_5=int(info[2])
)
build_generic((Characteristic,), "characteristics.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield CharacteristicDescription(
characteristic_id=int(info[0]),
language_id=int(info[1]),
description=info[2],
)
build_generic(
(CharacteristicDescription,), "characteristic_text.csv", csv_record_to_objects
)
###############
# EGG GROUP #
###############
def _build_egg_groups():
def csv_record_to_objects(info):
yield EggGroup(id=int(info[0]), name=info[1])
build_generic((EggGroup,), "egg_groups.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield EggGroupName(
egg_group_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((EggGroupName,), "egg_group_prose.csv", csv_record_to_objects)
#################
# GROWTH RATE #
#################
def _build_growth_rates():
def csv_record_to_objects(info):
yield GrowthRate(id=int(info[0]), name=info[1], formula=info[2])
build_generic((GrowthRate,), "growth_rates.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield GrowthRateDescription(
growth_rate_id=int(info[0]), language_id=int(info[1]), description=info[2]
)
build_generic(
(GrowthRateDescription,), "growth_rate_prose.csv", csv_record_to_objects
)
# ###########
# # ITEMS #
# ###########
def _build_items():
def csv_record_to_objects(info):
yield ItemPocket(id=int(info[0]), name=info[1])
build_generic((ItemPocket,), "item_pockets.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemPocketName(
item_pocket_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((ItemPocketName,), "item_pocket_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemFlingEffect(id=int(info[0]), name=info[1])
build_generic((ItemFlingEffect,), "item_fling_effects.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemFlingEffectEffectText(
item_fling_effect_id=int(info[0]),
language_id=int(info[1]),
effect=scrub_str(info[2]),
)
build_generic(
(ItemFlingEffectEffectText,),
"item_fling_effect_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield ItemCategory(id=int(info[0]), item_pocket_id=int(info[1]), name=info[2])
build_generic((ItemCategory,), "item_categories.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemCategoryName(
item_category_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((ItemCategoryName,), "item_category_prose.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield Item(
id=int(info[0]),
name=info[1],
item_category_id=int(info[2]),
cost=int(info[3]),
fling_power=int(info[4]) if info[4] != "" else None,
item_fling_effect_id=int(info[5]) if info[5] != "" else None,
)
build_generic((Item,), "items.csv", csv_record_to_objects)
def csv_record_to_objects(info):
if re.search(r"^data-card", info[1]):
file_name = "data-card.png"
elif re.search(r"^tm[0-9]", info[1]):
file_name = "tm-normal.png"
elif re.search(r"^hm[0-9]", info[1]):
file_name = "hm-normal.png"
else:
file_name = "%s.png" % info[1]
item_sprites = "items/{0}"
sprites = {"default": file_path_or_none(item_sprites.format(file_name))}
yield ItemSprites(
id=int(info[0]), item_id=int(info[0]), sprites=json.dumps(sprites)
)
build_generic((ItemSprites,), "items.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemName(item_id=int(info[0]), language_id=int(info[1]), name=info[2])
build_generic((ItemName,), "item_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemEffectText(
item_id=int(info[0]),
language_id=int(info[1]),
short_effect=scrub_str(info[2]),
effect=scrub_str(info[3]),
)
build_generic((ItemEffectText,), "item_prose.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemGameIndex(
item_id=int(info[0]), generation_id=int(info[1]), game_index=int(info[2])
)
build_generic((ItemGameIndex,), "item_game_indices.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemFlavorText(
item_id=int(info[0]),
version_group_id=int(info[1]),
language_id=int(info[2]),
flavor_text=info[3],
)
build_generic((ItemFlavorText,), "item_flavor_text.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemAttribute(id=int(info[0]), name=info[1])
build_generic((ItemAttribute,), "item_flags.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ItemAttributeName(
item_attribute_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
yield ItemAttributeDescription(
item_attribute_id=int(info[0]),
language_id=int(info[1]),
description=info[3],
)
build_generic(
(ItemAttributeName, ItemAttributeDescription),
"item_flag_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield ItemAttributeMap(item_id=int(info[0]), item_attribute_id=int(info[1]))
build_generic((ItemAttributeMap,), "item_flag_map.csv", csv_record_to_objects)
###########
# TYPES #
###########
def _build_types():
def csv_record_to_objects(info):
yield Type(
id=int(info[0]),
name=info[1],
generation_id=int(info[2]),
move_damage_class_id=int(info[3]) if info[3] != "" else None,
)
build_generic((Type,), "types.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield TypeName(type_id=int(info[0]), language_id=int(info[1]), name=info[2])
build_generic((TypeName,), "type_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield TypeGameIndex(
type_id=int(info[0]), generation_id=int(info[1]), game_index=int(info[2])
)
build_generic((TypeGameIndex,), "type_game_indices.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield TypeEfficacy(
damage_type_id=int(info[0]),
target_type_id=int(info[1]),
damage_factor=int(info[2]),
)
build_generic((TypeEfficacy,), "type_efficacy.csv", csv_record_to_objects)
#############
# CONTEST #
#############
def _build_contests():
def csv_record_to_objects(info):
yield ContestType(id=int(info[0]), name=info[1])
build_generic((ContestType,), "contest_types.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ContestTypeName(
contest_type_id=int(info[0]),
language_id=int(info[1]),
name=info[2],
flavor=info[3],
color=info[4],
)
build_generic((ContestTypeName,), "contest_type_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ContestEffect(id=int(info[0]), appeal=int(info[1]), jam=int(info[2]))
build_generic((ContestEffect,), "contest_effects.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield ContestEffectEffectText(
contest_effect_id=int(info[0]), language_id=int(info[1]), effect=info[3]
)
yield ContestEffectFlavorText(
contest_effect_id=int(info[0]),
language_id=int(info[1]),
flavor_text=info[2],
)
build_generic(
(ContestEffectEffectText, ContestEffectFlavorText),
"contest_effect_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield SuperContestEffect(id=int(info[0]), appeal=int(info[1]))
build_generic(
(SuperContestEffect,), "super_contest_effects.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield SuperContestEffectFlavorText(
super_contest_effect_id=int(info[0]),
language_id=int(info[1]),
flavor_text=info[2],
)
build_generic(
(SuperContestEffectFlavorText,),
"super_contest_effect_prose.csv",
csv_record_to_objects,
)
###########
# MOVES #
###########
def _build_moves():
def csv_record_to_objects(info):
yield MoveEffect(id=int(info[0]))
build_generic((MoveEffect,), "move_effects.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveEffectEffectText(
move_effect_id=int(info[0]),
language_id=int(info[1]),
short_effect=scrub_str(info[2]),
effect=scrub_str(info[3]),
)
build_generic(
(MoveEffectEffectText,), "move_effect_prose.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield MoveEffectChange(
id=int(info[0]), move_effect_id=int(info[1]), version_group_id=int(info[2])
)
build_generic(
(MoveEffectChange,), "move_effect_changelog.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield MoveEffectChangeEffectText(
move_effect_change_id=int(info[0]),
language_id=int(info[1]),
effect=scrub_str(info[2]),
)
build_generic(
(MoveEffectChangeEffectText,),
"move_effect_changelog_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield MoveLearnMethod(id=int(info[0]), name=info[1])
build_generic((MoveLearnMethod,), "pokemon_move_methods.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield VersionGroupMoveLearnMethod(
version_group_id=int(info[0]), move_learn_method_id=int(info[1])
)
build_generic(
(VersionGroupMoveLearnMethod,),
"version_group_pokemon_move_methods.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield MoveLearnMethodName(
move_learn_method_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
yield MoveLearnMethodDescription(
move_learn_method_id=int(info[0]),
language_id=int(info[1]),
description=info[3],
)
build_generic(
(MoveLearnMethodName, MoveLearnMethodDescription),
"pokemon_move_method_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield MoveTarget(id=int(info[0]), name=info[1])
build_generic((MoveTarget,), "move_targets.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveTargetName(
move_target_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
yield MoveTargetDescription(
move_target_id=int(info[0]), language_id=int(info[1]), description=info[3]
)
build_generic(
(MoveTargetName, MoveTargetDescription),
"move_target_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield Move(
id=int(info[0]),
name=info[1],
generation_id=int(info[2]),
type_id=int(info[3]),
power=int(info[4]) if info[4] != "" else None,
pp=int(info[5]) if info[5] != "" else None,
accuracy=int(info[6]) if info[6] != "" else None,
priority=int(info[7]) if info[7] != "" else None,
move_target_id=int(info[8]) if info[8] != "" else None,
move_damage_class_id=int(info[9]) if info[9] != "" else None,
move_effect_id=int(info[10]) if info[10] != "" else None,
move_effect_chance=int(info[11]) if info[11] != "" else None,
contest_type_id=int(info[12]) if info[12] != "" else None,
contest_effect_id=int(info[13]) if info[13] != "" else None,
super_contest_effect_id=int(info[14]) if info[14] != "" else None,
)
build_generic((Move,), "moves.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveName(move_id=int(info[0]), language_id=int(info[1]), name=info[2])
build_generic((MoveName,), "move_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveFlavorText(
move_id=int(info[0]),
version_group_id=int(info[1]),
language_id=int(info[2]),
flavor_text=info[3],
)
build_generic((MoveFlavorText,), "move_flavor_text.csv", csv_record_to_objects)
def csv_record_to_objects(info):
_move_effect = None
try:
_move_effect = (
MoveEffect.objects.get(pk=int(info[6])) if info[6] != "" else None
)
except:
pass
yield MoveChange(
move_id=int(info[0]),
version_group_id=int(info[1]),
type_id=int(info[2]) if info[2] != "" else None,
power=int(info[3]) if info[3] != "" else None,
pp=int(info[4]) if info[4] != "" else None,
accuracy=int(info[5]) if info[5] != "" else None,
move_effect_id=_move_effect.pk if _move_effect else None,
move_effect_chance=int(info[7]) if info[7] != "" else None,
)
build_generic((MoveChange,), "move_changelog.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveBattleStyle(id=int(info[0]), name=info[1])
build_generic((MoveBattleStyle,), "move_battle_styles.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveBattleStyleName(
move_battle_style_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic(
(MoveBattleStyleName,), "move_battle_style_prose.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield MoveAttribute(id=int(info[0]), name=info[1])
build_generic((MoveAttribute,), "move_flags.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveAttributeMap(move_id=int(info[0]), move_attribute_id=int(info[1]))
build_generic((MoveAttributeMap,), "move_flag_map.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveAttributeName(
move_attribute_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
yield MoveAttributeDescription(
move_attribute_id=int(info[0]),
language_id=int(info[1]),
description=scrub_str(info[3]),
)
build_generic(
(MoveAttributeName, MoveAttributeDescription),
"move_flag_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield MoveMetaAilment(id=int(info[0]), name=info[1])
build_generic((MoveMetaAilment,), "move_meta_ailments.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveMetaAilmentName(
move_meta_ailment_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic(
(MoveMetaAilmentName,), "move_meta_ailment_names.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield MoveMetaCategory(id=int(info[0]), name=info[1])
build_generic(
(MoveMetaCategory,), "move_meta_categories.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield MoveMetaCategoryDescription(
move_meta_category_id=int(info[0]),
language_id=int(info[1]),
description=info[2],
)
build_generic(
(MoveMetaCategoryDescription,),
"move_meta_category_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield MoveMeta(
move_id=int(info[0]),
move_meta_category_id=int(info[1]),
move_meta_ailment_id=int(info[2]),
min_hits=int(info[3]) if info[3] != "" else None,
max_hits=int(info[4]) if info[4] != "" else None,
min_turns=int(info[5]) if info[5] != "" else None,
max_turns=int(info[6]) if info[6] != "" else None,
drain=int(info[7]) if info[7] != "" else None,
healing=int(info[8]) if info[8] != "" else None,
crit_rate=int(info[9]) if info[9] != "" else None,
ailment_chance=int(info[10]) if info[10] != "" else None,
flinch_chance=int(info[11]) if info[11] != "" else None,
stat_chance=int(info[12]) if info[12] != "" else None,
)
build_generic((MoveMeta,), "move_meta.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield MoveMetaStatChange(
move_id=int(info[0]), stat_id=int(info[1]), change=int(info[2])
)
build_generic(
(MoveMetaStatChange,), "move_meta_stat_changes.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield ContestCombo(first_move_id=int(info[0]), second_move_id=int(info[1]))
build_generic((ContestCombo,), "contest_combos.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield SuperContestCombo(first_move_id=int(info[0]), second_move_id=int(info[1]))
build_generic(
(SuperContestCombo,), "super_contest_combos.csv", csv_record_to_objects
)
#############
# BERRIES #
#############
def _build_berries():
def csv_record_to_objects(info):
yield BerryFirmness(id=int(info[0]), name=info[1])
build_generic((BerryFirmness,), "berry_firmness.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield BerryFirmnessName(
berry_firmness_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic(
(BerryFirmnessName,), "berry_firmness_names.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
item = Item.objects.get(pk=int(info[1]))
yield Berry(
id=int(info[0]),
item_id=int(info[1]),
name=item.name[: item.name.index("-")],
berry_firmness_id=int(info[2]),
natural_gift_power=int(info[3]),
natural_gift_type_id=int(info[4]),
size=int(info[5]),
max_harvest=int(info[6]),
growth_time=int(info[7]),
soil_dryness=int(info[8]),
smoothness=int(info[9]),
)
build_generic((Berry,), "berries.csv", csv_record_to_objects)
def csv_record_to_objects(info):
# Get the english name for this contest type
contest_type_name = ContestTypeName.objects.get(
contest_type_id=int(info[0]), language_id=9
)
yield BerryFlavor(
id=int(info[0]),
name=contest_type_name.flavor.lower(),
contest_type=ContestType.objects.get(pk=int(info[0])),
)
# This is not an error
build_generic((BerryFlavor,), "contest_types.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield BerryFlavorName(
berry_flavor_id=int(info[0]), language_id=int(info[1]), name=info[3]
)
# This is not an error
build_generic((BerryFlavorName,), "contest_type_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield BerryFlavorMap(
berry_id=int(info[0]), berry_flavor_id=int(info[1]), potency=int(info[2])
)
# This is not an error
build_generic((BerryFlavorMap,), "berry_flavors.csv", csv_record_to_objects)
############
# NATURE #
############
def _build_natures():
def csv_record_to_objects(info):
decreased_stat = None
increased_stat = None
hates_flavor = None
likes_flavor = None
if info[2] != info[3]:
decreased_stat = Stat.objects.get(pk=int(info[2]))
increased_stat = Stat.objects.get(pk=int(info[3]))
if info[4] != info[5]:
hates_flavor = BerryFlavor.objects.get(pk=int(info[4]))
likes_flavor = BerryFlavor.objects.get(pk=int(info[5]))
yield Nature(
id=int(info[0]),
name=info[1],
decreased_stat=decreased_stat,
increased_stat=increased_stat,
hates_flavor=hates_flavor,
likes_flavor=likes_flavor,
game_index=info[6],
)
build_generic((Nature,), "natures.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield NatureName(nature_id=int(info[0]), language_id=int(info[1]), name=info[2])
build_generic((NatureName,), "nature_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield NaturePokeathlonStat(
nature_id=(info[0]), pokeathlon_stat_id=(info[1]), max_change=info[2]
)
build_generic(
(NaturePokeathlonStat,), "nature_pokeathlon_stats.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield NatureBattleStylePreference(
nature_id=int(info[0]),
move_battle_style_id=int(info[1]),
low_hp_preference=info[2],
high_hp_preference=info[3],
)
build_generic(
(NatureBattleStylePreference,),
"nature_battle_style_preferences.csv",
csv_record_to_objects,
)
###########
# GENDER #
###########
def _build_genders():
def csv_record_to_objects(info):
yield Gender(id=int(info[0]), name=info[1])
build_generic((Gender,), "genders.csv", csv_record_to_objects)
################
# EXPERIENCE #
################
def _build_experiences():
def csv_record_to_objects(info):
yield Experience(
growth_rate_id=int(info[0]), level=int(info[1]), experience=int(info[2])
)
build_generic((Experience,), "experience.csv", csv_record_to_objects)
##############
# MACHINES #
##############
def _build_machines():
def csv_record_to_objects(info):
yield Machine(
machine_number=int(info[0]),
version_group_id=int(info[1]),
item_id=int(info[2]),
move_id=int(info[3]),
)
build_generic((Machine,), "machines.csv", csv_record_to_objects)
###############
# EVOLUTION #
###############
def _build_evolutions():
def csv_record_to_objects(info):
yield EvolutionChain(
id=int(info[0]),
baby_trigger_item_id=int(info[1]) if info[1] != "" else None,
)
build_generic((EvolutionChain,), "evolution_chains.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield EvolutionTrigger(id=int(info[0]), name=info[1])
build_generic((EvolutionTrigger,), "evolution_triggers.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield EvolutionTriggerName(
evolution_trigger_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic(
(EvolutionTriggerName,), "evolution_trigger_prose.csv", csv_record_to_objects
)
#############
# POKEDEX #
#############
def _build_pokedexes():
def csv_record_to_objects(info):
yield Pokedex(
id=int(info[0]),
region_id=int(info[1]) if info[1] != "" else None,
name=info[2],
is_main_series=bool(int(info[3])),
)
build_generic((Pokedex,), "pokedexes.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokedexName(
pokedex_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
yield PokedexDescription(
pokedex_id=int(info[0]), language_id=int(info[1]), description=info[3]
)
build_generic(
(PokedexName, PokedexDescription), "pokedex_prose.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield PokedexVersionGroup(
pokedex_id=int(info[0]), version_group_id=int(info[1])
)
build_generic(
(PokedexVersionGroup,), "pokedex_version_groups.csv", csv_record_to_objects
)
##############
# LOCATION #
##############
def _build_locations():
def csv_record_to_objects(info):
yield Location(
id=int(info[0]),
region_id=int(info[1]) if info[1] != "" else None,
name=info[2],
)
build_generic((Location,), "locations.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield LocationName(
location_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((LocationName,), "location_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield LocationGameIndex(
location_id=int(info[0]),
generation_id=int(info[1]),
game_index=int(info[2]),
)
build_generic(
(LocationGameIndex,), "location_game_indices.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
location = Location.objects.get(pk=int(info[1]))
yield LocationArea(
id=int(info[0]),
location_id=int(info[1]),
game_index=int(info[2]),
name="{}-{}".format(location.name, info[3])
if info[3]
else "{}-{}".format(location.name, "area"),
)
build_generic((LocationArea,), "location_areas.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield LocationAreaName(
location_area_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((LocationAreaName,), "location_area_prose.csv", csv_record_to_objects)
#############
# POKEMON #
#############
def _build_pokemons():
def csv_record_to_objects(info):
yield PokemonColor(id=int(info[0]), name=info[1])
build_generic((PokemonColor,), "pokemon_colors.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonColorName(
pokemon_color_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((PokemonColorName,), "pokemon_color_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonShape(id=int(info[0]), name=info[1])
build_generic((PokemonShape,), "pokemon_shapes.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonShapeName(
pokemon_shape_id=int(info[0]),
language_id=int(info[1]),
name=info[2],
awesome_name=info[3],
)
build_generic((PokemonShapeName,), "pokemon_shape_prose.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonHabitat(id=int(info[0]), name=info[1])
build_generic((PokemonHabitat,), "pokemon_habitats.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonSpecies(
id=int(info[0]),
name=info[1],
generation_id=int(info[2]) if info[2] != "" else None,
evolves_from_species=None,
evolution_chain_id=int(info[4]) if info[4] != "" else None,
pokemon_color_id=int(info[5]) if info[5] != "" else None,
pokemon_shape_id=int(info[6]) if info[6] != "" else None,
pokemon_habitat_id=int(info[7]) if info[7] != "" else None,
gender_rate=int(info[8]) if info[8] != "" else None,
capture_rate=int(info[9]) if info[9] != "" else None,
base_happiness=int(info[10]) if info[10] != "" else None,
is_baby=bool(int(info[11])) if info[11] != "" else None,
hatch_counter=int(info[12]) if info[12] != "" else None,
has_gender_differences=bool(int(info[13])) if info[13] != "" else False,
growth_rate_id=int(info[14]) if info[14] != "" else None,
forms_switchable=bool(int(info[15])) if info[15] != "" else None,
is_legendary=bool(int(info[16])) if info[16] != "" else None,
is_mythical=bool(int(info[17])) if info[17] != "" else None,
order=int(info[18]) if info[18] != "" else None,
)
build_generic((PokemonSpecies,), "pokemon_species.csv", csv_record_to_objects)
# PokemonSpecies.evolves_from_species can't be set until all the species are created
data = load_data("pokemon_species.csv")
for index, info in enumerate(data):
if index > 0:
evolves = (
PokemonSpecies.objects.get(pk=int(info[3])) if info[3] != "" else None
)
if evolves:
species = PokemonSpecies.objects.get(pk=int(info[0]))
species.evolves_from_species = evolves
species.save()
def csv_record_to_objects(info):
yield PokemonSpeciesName(
pokemon_species_id=int(info[0]),
language_id=int(info[1]),
name=info[2],
genus=info[3],
)
build_generic(
(PokemonSpeciesName,), "pokemon_species_names.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield PokemonSpeciesDescription(
pokemon_species_id=int(info[0]),
language_id=int(info[1]),
description=scrub_str(info[2]),
)
build_generic(
(PokemonSpeciesDescription,), "pokemon_species_prose.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield PokemonSpeciesFlavorText(
pokemon_species_id=int(info[0]),
version_id=int(info[1]),
language_id=int(info[2]),
flavor_text=info[3],
)
build_generic(
(PokemonSpeciesFlavorText,),
"pokemon_species_flavor_text.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield Pokemon(
id=int(info[0]),
name=info[1],
pokemon_species_id=int(info[2]) if info[2] != "" else None,
height=int(info[3]) if info[3] != "" else None,
weight=int(info[4]) if info[4] != "" else None,
base_experience=int(info[5]) if info[5] != "" else None,
order=int(info[6]) if info[6] != "" else -1,
is_default=bool(int(info[7])) if info[7] != "" else None,
)
build_generic((Pokemon,), "pokemon.csv", csv_record_to_objects)
def csv_record_to_objects(info):
file_name_png = "%s.png" % info[0]
file_name_gif = "%s.gif" % info[0]
file_name_svg = "%s.svg" % info[0]
poke_sprites = "pokemon/{0}"
dream_world = "other/dream-world/{0}"
official_art = "other/official-artwork/{0}"
gen_i = "versions/generation-i/{0}"
gen_ii = "versions/generation-ii/{0}"
gen_iii = "versions/generation-iii/{0}"
gen_iv = "versions/generation-iv/{0}"
gen_v = "versions/generation-v/{0}"
gen_vi = "versions/generation-vi/{0}"
gen_vii = "versions/generation-vii/{0}"
gen_viii = "versions/generation-viii/{0}"
sprites = {
"front_default": file_path_or_none(poke_sprites.format(file_name_png)),
"front_female": file_path_or_none(
poke_sprites.format("female/" + file_name_png)
),
"front_shiny": file_path_or_none(
poke_sprites.format("shiny/" + file_name_png)
),
"front_shiny_female": file_path_or_none(
poke_sprites.format("shiny/female/" + file_name_png)
),
"back_default": file_path_or_none(
poke_sprites.format("back/" + file_name_png)
),
"back_female": file_path_or_none(
poke_sprites.format("back/female/" + file_name_png)
),
"back_shiny": file_path_or_none(
poke_sprites.format("back/shiny/" + file_name_png)
),
"back_shiny_female": file_path_or_none(
poke_sprites.format("back/shiny/female/" + file_name_png)
),
"other": {
"dream_world": {
"front_default": file_path_or_none(
poke_sprites.format(dream_world.format(file_name_svg))
),
"front_female": file_path_or_none(
poke_sprites.format(
dream_world.format("female/" + file_name_svg)
)
),
},
"official-artwork": {
"front_default": file_path_or_none(
poke_sprites.format(official_art.format(file_name_png))
)
},
},
"versions": {
"generation-i": {
"red-blue": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_i.format("red-blue/" + file_name_png)
)
),
"front_gray": file_path_or_none(
poke_sprites.format(
gen_i.format("red-blue/gray/" + file_name_png)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_i.format("red-blue/back/" + file_name_png)
)
),
"back_gray": file_path_or_none(
poke_sprites.format(
gen_i.format("red-blue/back/gray/" + file_name_png)
)
),
},
"yellow": {
"front_default": file_path_or_none(
poke_sprites.format(gen_i.format("yellow/" + file_name_png))
),
"front_gray": file_path_or_none(
poke_sprites.format(
gen_i.format("yellow/gray/" + file_name_png)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_i.format("yellow/back/" + file_name_png)
)
),
"back_gray": file_path_or_none(
poke_sprites.format(
gen_i.format("yellow/back/gray/" + file_name_png)
)
),
},
},
"generation-ii": {
"crystal": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_ii.format("crystal/" + file_name_png)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_ii.format("crystal/shiny/" + file_name_png)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_ii.format("crystal/back/" + file_name_png)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_ii.format("crystal/back/shiny/" + file_name_png)
)
),
},
"gold": {
"front_default": file_path_or_none(
poke_sprites.format(gen_ii.format("gold/" + file_name_png))
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_ii.format("gold/shiny/" + file_name_png)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_ii.format("gold/back/" + file_name_png)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_ii.format("gold/back/shiny/" + file_name_png)
)
),
},
"silver": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_ii.format("silver/" + file_name_png)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_ii.format("silver/shiny/" + file_name_png)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_ii.format("silver/back/" + file_name_png)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_ii.format("silver/back/shiny/" + file_name_png)
)
),
},
},
"generation-iii": {
"emerald": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_iii.format("emerald/" + file_name_png)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_iii.format("emerald/shiny/" + file_name_png)
)
),
},
"firered-leafgreen": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_iii.format("firered-leafgreen/" + file_name_png)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_iii.format(
"firered-leafgreen/shiny/" + file_name_png
)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_iii.format(
"firered-leafgreen/back/" + file_name_png
)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_iii.format(
"firered-leafgreen/back/shiny/" + file_name_png
)
)
),
},
"ruby-sapphire": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_iii.format("ruby-sapphire/" + file_name_png)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_iii.format("ruby-sapphire/shiny/" + file_name_png)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_iii.format("ruby-sapphire/back/" + file_name_png)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_iii.format(
"ruby-sapphire/back/shiny/" + file_name_png
)
)
),
},
},
"generation-iv": {
"diamond-pearl": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_iv.format("diamond-pearl/" + file_name_png)
)
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_iv.format("diamond-pearl/female/" + file_name_png)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_iv.format("diamond-pearl/shiny/" + file_name_png)
)
),
"front_shiny_female": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"diamond-pearl/shiny/female/" + file_name_png
)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_iv.format("diamond-pearl/back/" + file_name_png)
)
),
"back_female": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"diamond-pearl/back/female/" + file_name_png
)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"diamond-pearl/back/shiny/" + file_name_png
)
)
),
"back_shiny_female": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"diamond-pearl/back/shiny/female/" + file_name_png
)
)
),
},
"heartgold-soulsilver": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_iv.format("heartgold-soulsilver/" + file_name_png)
)
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"heartgold-soulsilver/female/" + file_name_png
)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"heartgold-soulsilver/shiny/" + file_name_png
)
)
),
"front_shiny_female": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"heartgold-soulsilver/shiny/female/" + file_name_png
)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"heartgold-soulsilver/back/" + file_name_png
)
)
),
"back_female": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"heartgold-soulsilver/back/female/" + file_name_png
)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"heartgold-soulsilver/back/shiny/" + file_name_png
)
)
),
"back_shiny_female": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"heartgold-soulsilver/back/shiny/female/"
+ file_name_png
)
)
),
},
"platinum": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_iv.format("platinum/" + file_name_png)
)
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_iv.format("platinum/female/" + file_name_png)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_iv.format("platinum/shiny/" + file_name_png)
)
),
"front_shiny_female": file_path_or_none(
poke_sprites.format(
gen_iv.format("platinum/shiny/female/" + file_name_png)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_iv.format("platinum/back/" + file_name_png)
)
),
"back_female": file_path_or_none(
poke_sprites.format(
gen_iv.format("platinum/back/female/" + file_name_png)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_iv.format("platinum/back/shiny/" + file_name_png)
)
),
"back_shiny_female": file_path_or_none(
poke_sprites.format(
gen_iv.format(
"platinum/back/shiny/female/" + file_name_png
)
)
),
},
},
"generation-v": {
"black-white": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_v.format("black-white/" + file_name_png)
)
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_v.format("black-white/female/" + file_name_png)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_v.format("black-white/shiny/" + file_name_png)
)
),
"front_shiny_female": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/shiny/female/" + file_name_png
)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_v.format("black-white/back/" + file_name_png)
)
),
"back_female": file_path_or_none(
poke_sprites.format(
gen_v.format("black-white/back/female/" + file_name_png)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_v.format("black-white/back/shiny/" + file_name_png)
)
),
"back_shiny_female": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/back/shiny/female/" + file_name_png
)
)
),
"animated": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/animated/" + file_name_gif
)
)
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/animated/female/" + file_name_gif
)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/animated/shiny/" + file_name_gif
)
)
),
"front_shiny_female": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/animated/shiny/female/"
+ file_name_gif
)
)
),
"back_default": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/animated/back/" + file_name_gif
)
)
),
"back_female": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/animated/back/female/"
+ file_name_gif
)
)
),
"back_shiny": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/animated/back/shiny/"
+ file_name_gif
)
)
),
"back_shiny_female": file_path_or_none(
poke_sprites.format(
gen_v.format(
"black-white/animated/back/shiny/female/"
+ file_name_gif
)
)
),
},
}
},
"generation-vi": {
"omegaruby-alphasapphire": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_vi.format(
"omegaruby-alphasapphire/" + file_name_png
)
)
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_vi.format(
"omegaruby-alphasapphire/female/" + file_name_png
)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_vi.format(
"omegaruby-alphasapphire/shiny/" + file_name_png
)
)
),
"front_shiny_female": file_path_or_none(
poke_sprites.format(
gen_vi.format(
"omegaruby-alphasapphire/shiny/female/"
+ file_name_png
)
)
),
},
"x-y": {
"front_default": file_path_or_none(
poke_sprites.format(gen_vi.format("x-y/" + file_name_png))
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_vi.format("x-y/female/" + file_name_png)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_vi.format("x-y/shiny/" + file_name_png)
)
),
"front_shiny_female": file_path_or_none(
poke_sprites.format(
gen_vi.format("x-y/shiny/female/" + file_name_png)
)
),
},
},
"generation-vii": {
"ultra-sun-ultra-moon": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_vii.format("ultra-sun-ultra-moon/" + file_name_png)
)
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_vii.format(
"ultra-sun-ultra-moon/female/" + file_name_png
)
)
),
"front_shiny": file_path_or_none(
poke_sprites.format(
gen_vii.format(
"ultra-sun-ultra-moon/shiny/" + file_name_png
)
)
),
"front_shiny_female": file_path_or_none(
poke_sprites.format(
gen_vii.format(
"ultra-sun-ultra-moon/shiny/female/" + file_name_png
)
)
),
},
"icons": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_vii.format("icons/" + file_name_png)
)
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_vii.format("icons/female/" + file_name_png)
)
),
},
},
"generation-viii": {
"icons": {
"front_default": file_path_or_none(
poke_sprites.format(
gen_viii.format("icons/" + file_name_png)
)
),
"front_female": file_path_or_none(
poke_sprites.format(
gen_viii.format("icons/female/" + file_name_png)
)
),
}
},
},
}
yield PokemonSprites(
id=int(info[0]),
pokemon=Pokemon.objects.get(pk=int(info[0])),
sprites=json.dumps(sprites),
)
build_generic((PokemonSprites,), "pokemon.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonAbility(
pokemon_id=int(info[0]),
ability_id=int(info[1]),
is_hidden=bool(int(info[2])),
slot=int(info[3]),
)
build_generic((PokemonAbility,), "pokemon_abilities.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonDexNumber(
pokemon_species_id=int(info[0]),
pokedex_id=int(info[1]),
pokedex_number=int(info[2]),
)
build_generic((PokemonDexNumber,), "pokemon_dex_numbers.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonEggGroup(
pokemon_species_id=int(info[0]), egg_group_id=int(info[1])
)
build_generic((PokemonEggGroup,), "pokemon_egg_groups.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonEvolution(
id=int(info[0]),
evolved_species_id=int(info[1]),
evolution_trigger_id=int(info[2]),
evolution_item_id=int(info[3]) if info[3] != "" else None,
min_level=int(info[4]) if info[4] != "" else None,
gender_id=int(info[5]) if info[5] != "" else None,
location_id=int(info[6]) if info[6] != "" else None,
held_item_id=int(info[7]) if info[7] != "" else None,
time_of_day=info[8],
known_move_id=int(info[9]) if info[9] != "" else None,
known_move_type_id=int(info[10]) if info[10] != "" else None,
min_happiness=int(info[11]) if info[11] != "" else None,
min_beauty=int(info[12]) if info[12] != "" else None,
min_affection=int(info[13]) if info[13] != "" else None,
relative_physical_stats=int(info[14]) if info[14] != "" else None,
party_species_id=int(info[15]) if info[15] != "" else None,
party_type_id=int(info[16]) if info[16] != "" else None,
trade_species_id=int(info[17]) if info[17] != "" else None,
needs_overworld_rain=bool(int(info[18])),
turn_upside_down=bool(int(info[19])),
)
build_generic((PokemonEvolution,), "pokemon_evolution.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonForm(
id=int(info[0]),
name=info[1],
form_name=info[2],
pokemon_id=int(info[3]),
version_group_id=int(info[4]),
is_default=bool(int(info[5])),
is_battle_only=bool(int(info[6])),
is_mega=bool(int(info[7])),
form_order=int(info[8]),
order=int(info[9]),
)
build_generic((PokemonForm,), "pokemon_forms.csv", csv_record_to_objects)
def csv_record_to_objects(info):
pokemon = Pokemon.objects.get(pk=int(info[3]))
if info[2]:
if re.search(r"^mega", info[2]):
file_name = "%s.png" % info[3]
else:
file_name = "%s-%s.png" % (
getattr(pokemon, "pokemon_species_id"),
info[2],
)
else:
file_name = "%s.png" % getattr(pokemon, "pokemon_species_id")
poke_sprites = "pokemon/{0}"
sprites = {
"front_default": file_path_or_none(poke_sprites.format(file_name)),
"front_shiny": file_path_or_none(poke_sprites.format("shiny/" + file_name)),
"back_default": file_path_or_none(poke_sprites.format("back/" + file_name)),
"back_shiny": file_path_or_none(
poke_sprites.format("back/shiny/" + file_name)
),
}
yield PokemonFormSprites(
id=int(info[0]), pokemon_form_id=int(info[0]), sprites=json.dumps(sprites)
)
build_generic((PokemonFormSprites,), "pokemon_forms.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonFormName(
pokemon_form=PokemonForm.objects.get(pk=int(info[0])),
language=Language.objects.get(pk=int(info[1])),
name=info[2],
pokemon_name=info[3],
)
build_generic((PokemonFormName,), "pokemon_form_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonFormGeneration(
pokemon_form_id=int(info[0]),
generation_id=int(info[1]),
game_index=int(info[2]),
)
build_generic(
(PokemonFormGeneration,), "pokemon_form_generations.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield PokemonGameIndex(
pokemon_id=int(info[0]), version_id=int(info[1]), game_index=int(info[2])
)
build_generic(
(PokemonGameIndex,), "pokemon_game_indices.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield PokemonHabitatName(
pokemon_habitat_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic(
(PokemonHabitatName,), "pokemon_habitat_names.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield PokemonItem(
pokemon_id=int(info[0]),
version_id=int(info[1]),
item_id=int(info[2]),
rarity=int(info[3]),
)
build_generic((PokemonItem,), "pokemon_items.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonMove(
pokemon_id=int(info[0]),
version_group_id=int(info[1]),
move_id=int(info[2]),
move_learn_method_id=int(info[3]),
level=int(info[4]),
order=int(info[5]) if info[5] != "" else None,
)
build_generic((PokemonMove,), "pokemon_moves.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonStat(
pokemon_id=int(info[0]),
stat_id=int(info[1]),
base_stat=int(info[2]),
effort=int(info[3]),
)
build_generic((PokemonStat,), "pokemon_stats.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PokemonType(
pokemon_id=int(info[0]), type_id=int(info[1]), slot=int(info[2])
)
build_generic((PokemonType,), "pokemon_types.csv", csv_record_to_objects)
###############
# ENCOUNTER #
###############
def _build_encounters():
def csv_record_to_objects(info):
yield EncounterMethod(id=int(info[0]), name=info[1], order=int(info[2]))
build_generic((EncounterMethod,), "encounter_methods.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield LocationAreaEncounterRate(
location_area_id=int(info[0]),
encounter_method_id=int(info[1]),
version_id=int(info[2]),
rate=int(info[3]),
)
build_generic(
(LocationAreaEncounterRate,),
"location_area_encounter_rates.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield EncounterMethodName(
encounter_method_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic(
(EncounterMethodName,), "encounter_method_prose.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield EncounterCondition(id=int(info[0]), name=info[1])
build_generic(
(EncounterCondition,), "encounter_conditions.csv", csv_record_to_objects
)
def csv_record_to_objects(info):
yield EncounterConditionName(
encounter_condition_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic(
(EncounterConditionName,),
"encounter_condition_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield EncounterSlot(
id=int(info[0]),
version_group_id=int(info[1]),
encounter_method_id=int(info[2]),
slot=int(info[3]) if info[3] != "" else None,
rarity=int(info[4]),
)
build_generic((EncounterSlot,), "encounter_slots.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield Encounter(
id=int(info[0]),
version_id=int(info[1]),
location_area_id=int(info[2]),
encounter_slot_id=int(info[3]),
pokemon_id=int(info[4]),
min_level=int(info[5]),
max_level=int(info[6]),
)
build_generic((Encounter,), "encounters.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield EncounterConditionValue(
id=int(info[0]),
encounter_condition_id=int(info[1]),
name=info[2],
is_default=bool(int(info[3])),
)
build_generic(
(EncounterConditionValue,),
"encounter_condition_values.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield EncounterConditionValueName(
encounter_condition_value_id=int(info[0]),
language_id=int(info[1]),
name=info[2],
)
build_generic(
(EncounterConditionValueName,),
"encounter_condition_value_prose.csv",
csv_record_to_objects,
)
def csv_record_to_objects(info):
yield EncounterConditionValueMap(
encounter_id=int(info[0]), encounter_condition_value_id=int(info[1])
)
build_generic(
(EncounterConditionValueMap,),
"encounter_condition_value_map.csv",
csv_record_to_objects,
)
##############
# PAL PARK #
##############
def _build_pal_parks():
def csv_record_to_objects(info):
yield PalParkArea(id=int(info[0]), name=info[1])
build_generic((PalParkArea,), "pal_park_areas.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PalParkAreaName(
pal_park_area_id=int(info[0]), language_id=int(info[1]), name=info[2]
)
build_generic((PalParkAreaName,), "pal_park_area_names.csv", csv_record_to_objects)
def csv_record_to_objects(info):
yield PalPark(
pokemon_species_id=int(info[0]),
pal_park_area_id=int(info[1]),
base_score=int(info[2]),
rate=int(info[3]),
)
build_generic((PalPark,), "pal_park.csv", csv_record_to_objects)
def build_all():
_build_languages()
_build_regions()
_build_generations()
_build_versions()
_build_damage_classes()
_build_stats()
_build_abilities()
_build_characteristics()
_build_egg_groups()
_build_growth_rates()
_build_items()
_build_types()
_build_contests()
_build_moves()
_build_berries()
_build_natures()
_build_genders()
_build_experiences()
_build_machines()
_build_evolutions()
_build_pokedexes()
_build_locations()
_build_pokemons()
_build_encounters()
_build_pal_parks()
if __name__ == "__main__":
build_all()