mirror of
https://github.com/PokeAPI/pokeapi
synced 2024-11-26 21:30:19 +00:00
1733 lines
52 KiB
Python
1733 lines
52 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/"
|
|
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(":")[1]
|
|
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]),
|
|
move_damage_class_id=int(info[9]),
|
|
move_effect_id=int(info[10]),
|
|
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]),
|
|
evolves_from_species=None,
|
|
evolution_chain_id=int(info[4]),
|
|
pokemon_color_id=int(info[5]),
|
|
pokemon_shape_id=int(info[6]),
|
|
pokemon_habitat_id=int(info[7]) if info[7] != "" else None,
|
|
gender_rate=int(info[8]),
|
|
capture_rate=int(info[9]),
|
|
base_happiness=int(info[10]),
|
|
is_baby=bool(int(info[11])),
|
|
hatch_counter=int(info[12]),
|
|
has_gender_differences=bool(int(info[13])),
|
|
growth_rate_id=int(info[14]),
|
|
forms_switchable=bool(int(info[15])),
|
|
order=int(info[16]),
|
|
)
|
|
|
|
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]),
|
|
height=int(info[3]),
|
|
weight=int(info[4]),
|
|
base_experience=int(info[5]),
|
|
order=int(info[6]),
|
|
is_default=bool(int(info[7])),
|
|
)
|
|
|
|
build_generic((Pokemon,), "pokemon.csv", csv_record_to_objects)
|
|
|
|
def csv_record_to_objects(info):
|
|
file_name = "%s.png" % info[0]
|
|
poke_sprites = "pokemon/{0}"
|
|
sprites = {
|
|
"front_default": file_path_or_none(poke_sprites.format(file_name)),
|
|
"front_female": file_path_or_none(
|
|
poke_sprites.format("female/" + file_name)
|
|
),
|
|
"front_shiny": file_path_or_none(poke_sprites.format("shiny/" + file_name)),
|
|
"front_shiny_female": file_path_or_none(
|
|
poke_sprites.format("shiny/female/" + file_name)
|
|
),
|
|
"back_default": file_path_or_none(poke_sprites.format("back/" + file_name)),
|
|
"back_female": file_path_or_none(
|
|
poke_sprites.format("back/female/" + file_name)
|
|
),
|
|
"back_shiny": file_path_or_none(
|
|
poke_sprites.format("back/shiny/" + file_name)
|
|
),
|
|
"back_shiny_female": file_path_or_none(
|
|
poke_sprites.format("back/shiny/female/" + file_name)
|
|
),
|
|
}
|
|
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)
|
|
|
|
def csv_record_to_objects(info):
|
|
yield PokemonTypePast(
|
|
pokemon_id=int(info[0]),
|
|
generation_id=int(info[1]),
|
|
type_id=int(info[2]),
|
|
slot=int(info[3])
|
|
)
|
|
|
|
build_generic((PokemonTypePast,), "pokemon_types_past.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()
|