seed_database.py

"""
    Script to seed database.
    You'll have to enter the password to your server twice to have your database perform the drop and create db
    After doing that and no errors pop up, you're good to run 
"""

import os
import requests

import crud
import model
import server

os.system("dropdb specs-capstone")

os.system('createdb specs-capstone')

model.connect_to_db(server.app)

def seed_moves():

    # Fetch data from PokeAPI for moves
    response = requests.get('https://pokeapi.co/api/v2/move/?limit=919')
    data = response.json()

    # Iterate over each move and populate the database
    for move in data['results']:
        move_data = requests.get(move['url']).json()
        move_instance = crud.create_move(
            name=move_data['name'],
            power=move_data['power'],
            accuracy=move_data['accuracy'],
            pp=move_data['pp']
        )
        model.db.session.add(move_instance)
    
    model.db.session.commit()

def seed_poke_types():
    response = requests.get('https://pokeapi.co/api/v2/type/?limit=19')
    data = response.json()

    for type_data in data['results']:
        type_name = type_data['name']
        
        # Create PokeType instance
        poke_type = crud.create_poke_type(poketype=type_name)
        
        model.db.session.add(poke_type)
    
    model.db.session.commit()

def seed_pokemon():
    response = requests.get('https://pokeapi.co/api/v2/pokemon/?limit=1025') # grabs all pokemon
    data = response.json()

    # Iterate over each Pokemon and populate the database
    for pokemon in data['results']:
        pokemon_data = requests.get(pokemon['url']).json()

        # Get or create PokeType instances for each type
        poke_types = []
        for type_data in pokemon_data['types']:
            poke_type_name = type_data['type']['name']
            poke_type = crud.get_poke_type_by_type(poke_type_name)
            if not poke_type:
                poke_type = crud.create_poke_type(poketype=poke_type_name)
            poke_types.append(poke_type.id)
            
        # Create Pokemon instance
        pokemon_instance = crud.create_pokemon(
            name=pokemon_data['name'],
            # poketype will store the id of the foreign key from poketypes
            # so create a crud get by id for poketype
            poketype1=poke_types[0],
            poketype2=poke_types[1] if len(poke_types) > 1 else None
        )
        model.db.session.add(pokemon_instance)
    
    model.db.session.commit()

def seed_pokemon_moves():

    pokemon = crud.get_pokemon() # all pokemon in db

    # iterate over all pokemon and assign connections with pokemon to all moves that the api listed that pokemon has
    for poke in pokemon:
        moves = requests.get(f'https://pokeapi.co/api/v2/pokemon/{poke.id}').json()['moves']
        for move in moves:
            poke_move = crud.get_move_by_name(move['move']['name'])
            if poke_move:
                pokemon_move_instance = crud.create_poke_move(
                pokemon_id=poke.id,
                move_id=poke_move.id
                )
                model.db.session.add(pokemon_move_instance)
    
    model.db.session.commit()

# Call the seeding functions

with server.app.app_context():
    model.db.create_all()

    seed_moves()
    seed_poke_types()
    seed_pokemon()
    seed_pokemon_moves()