def select_all():
pokemons = []
sql = "SELECT * FROM pokemons"
results = run_sql(sql)
for row in results:
nurse = nurse_repository.select(row['nurse_id'])
trainer = trainer_repository.select(row['trainer_id'])
pokemon = Pokemon(row['nickname'], row['species'], row['type'],
row['dob'], trainer, row['status'], row['id'])
pokemon.assign_nurse(nurse)
pokemons.append(pokemon)
return (pokemons)
def select(id):
pokemon = None
sql = "SELECT * FROM pokemons WHERE id = %s"
values = [id]
result = run_sql(sql, values)[0]
if result is not None:
nurse = nurse_repository.select(result['nurse_id'])
trainer = trainer_repository.select(result['trainer_id'])
pokemon = Pokemon(result['nickname'], result['species'],
result['type'], result['dob'], trainer,
result['status'], result['id'])
pokemon.assign_nurse(nurse)
return pokemon
def edit_pokemon(id):
nickname = request.form['nickname']
species = request.form['species']
type = request.form['type']
dob = request.form['dob']
trainer_id = request.form['trainer_id']
status = request.form['status']
nurse_id = request.form['nurse_id']
trainer = trainer_repository.select(trainer_id)
nurse = nurse_repository.select(nurse_id)
pokemon = Pokemon(nickname, species, type, dob, trainer, status, id)
pokemon.assign_nurse(nurse)
pokemon_repository.update(pokemon)
return redirect('/pokemon')
def search(category, search):
pokemons = []
if category == "nickname":
sql = "SELECT * FROM pokemons WHERE nickname = %s"
elif category == "type":
sql = "SELECT * FROM pokemons WHERE type = %s"
elif category == "species":
sql = "SELECT * FROM pokemons WHERE species = %s"
elif category == "status":
sql = "SELECT * FROM pokemons WHERE status = %s"
values = [search]
results = run_sql(sql, values)
for row in results:
nurse = nurse_repository.select(row['nurse_id'])
trainer = trainer_repository.select(row['trainer_id'])
pokemon = Pokemon(row['nickname'], row['species'], row['type'],
row['dob'], trainer, row['status'], row['id'])
pokemon.assign_nurse(nurse)
pokemons.append(pokemon)
return pokemons
class TestPokemon(unittest.TestCase):
def setUp(self):
trainer = Trainer("Ash", 259)
self.pokemon = Pokemon("Seed Backman", "Bulbasaur", "Grass",
"25/09/1990", trainer, 'burned')
def test_pokemon_has_nickname(self):
self.assertEqual("Seed Backman", self.pokemon.nickname)
def test_pokemon_has_species(self):
self.assertEqual("Bulbasaur", self.pokemon.species)
def test_pokemon_has_type(self):
self.assertEqual("Grass", self.pokemon.type)
def test_pokemon_has_dob(self):
self.assertEqual("25/09/1990", self.pokemon.dob)
def test_pokemon_has_trainer_name(self):
self.assertEqual("Ash", self.pokemon.trainer.name)
def test_pokemon_has_trainer_number(self):
self.assertEqual(259, self.pokemon.trainer.number)
def test_pokemon_has_status(self):
self.assertEqual('burned', self.pokemon.status)
def test_pokemon_before_nurse(self):
self.assertEqual(None, self.pokemon.nurse)
def test_assign_nurse(self):
nurse = Nurse('Joy', 'fire')
self.pokemon.assign_nurse(nurse)
self.assertEqual(nurse, self.pokemon.nurse)
def test_get_nurse_name(self):
nurse = Nurse('Joy', 'fire')
self.pokemon.assign_nurse(nurse)
self.assertEqual('Joy', self.pokemon.nurse.name)
from models.trainer import Trainer
import repositories.nurse_repository as nurse_repository
import repositories.pokemon_repository as pokemon_repository
import repositories.trainer_repository as trainer_repository
# pokemon_repository.delete_all()
# nurse_repository.delete_all()
# trainer_repository.delete_all()
nurse_1 = Nurse("Joy", "Fire")
nurse_repository.save(nurse_1)
trainer = Trainer("Simon", 12345)
trainer_repository.save(trainer)
pokemon = Pokemon('Seed Backman', 'Bulbasaur', 'Grass', '25/09/1990', trainer , 'burned')
pokemon.assign_nurse(nurse_1)
pokemon_repository.save(pokemon)
trainer.name = 'Simondovich'
trainer_repository.update(trainer)
nurse_1.name = "Betsy"
nurse_repository.update(nurse_1)
pokemon_repository.delete_all()
nurse_repository.delete_all()
# trainer_repository.delete_all()
# trainer_1 = Trainer('Ash', '2 Pallet Town', 32356)
# trainer_2 = Trainer('Misty', 'Cerulean Gym', 78232)
nurse_1 = Nurse('Joy')
nurse_2 = Nurse('Joye')
pokemon_1 = Pokemon('Sparky', 'trainer_1', 'Pikachu', '15/6/2017', nurse_1,
"Crushed by Snorlax")
pokemon_2 = Pokemon('Frankie', 'trainer_2', 'Psyduck', '12/1/2016', nurse_1,
"Burnt by Charizard")
pokemon_3 = Pokemon('Sally', 'trainer_2', 'Seel', '15/8/2015', nurse_2,
"Shocked by Pikachu")
#assign nurse to pokemon before running console.py
# trainer_repository.save(trainer_1)
# trainer_repository.save(trainer_2)
nurse_repository.save(nurse_1)
nurse_repository.save(nurse_2)
pokemon_repository.save(pokemon_1)
pokemon_repository.save(pokemon_2)
pokemon_repository.save(pokemon_3)
pokemon_1.assign_nurse(nurse_1)
pokemon_2.assign_nurse(nurse_2)
pokemon_3.assign_nurse(nurse_2)