Pokemon

from project.pokemon import Pokemon


class Trainer:
    def __init__(self, name):
        self.name = name
        self.pokemons = []

    def add_pokemon(self, pokemon):
        if pokemon not in self.pokemons:
            self.pokemons.append(pokemon)
            return f"Caught {pokemon.name} with health {pokemon.health}"
        else:
            return "This pokemon is already caught"

    def release_pokemon(self, pokemon_name):
        for pok in self.pokemons:
            if pok == pokemon_name:
                self.pokemons.remove(pok)
                return f"You have released {pokemon_name}"
            return "Pokemon is not caught"

    def trainer_data(self):
        to_print = ""
        to_print += f"Pokemon trainer {self.name}\n"
        to_print += f"Pokemon count {len(self.pokemons)}\n"
        for pok in self.pokemons:
            to_print += "- " + pokemon.pokemon_details()

        return to_print


pokemon = Pokemon("Pikachu", 90)
print(pokemon.pokemon_details())
trainer = Trainer("Ash")
print(trainer.add_pokemon(pokemon))
second_pokemon = Pokemon("Charizard", 110)
print(trainer.add_pokemon(second_pokemon))
print(trainer.add_pokemon(second_pokemon))
print(trainer.release_pokemon("Pikachu"))
print(trainer.release_pokemon("Pikachu"))
print(trainer.trainer_data())