def update_nurse(id):
nurse = nurse_repository.select(id)
name = request.form["name"]
nurse = Nurse(name)
nurse.id = id
nurse_repository.update(nurse)
return redirect("/nurses/index")
def assign_nurse():
nurse_id = request.form['nurse_id']
pokemon_id = request.form['pokemon_id']
nurse = nurse_repository.select(nurse_id)
pokemon = pokemon_repository.select(pokemon_id)
nurse.assign_pokemon(pokemon)
# pdb.set_trace()
pokemon_repository.update(pokemon)
return redirect('/nurses/index')
def select_all():
pokemons = []
sql = "SELECT * FROM pokemons"
results = run_sql(sql)
for row in results:
nurse = nurse_repository.select(row['nurse_id'])
pokemon = Pokemon(row['name'], row['trainer'], row['species'], row['hatched'], nurse, row['notes'], row['id'])
pokemons.append(pokemon)
return pokemons
def create_pokemon():
name = request.form["name"]
trainer = request.form["trainer"]
species = request.form["species"]
hatched = request.form["hatched"]
nurse_id = request.form["nurse_id"]
nurse = nurse_repository.select(nurse_id)
notes = request.form["notes"]
new_pokemon = Pokemon(name, trainer, species, hatched, nurse, notes)
pokemon_repository.save(new_pokemon)
return redirect('/pokemons/index')
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 update_pokemon(id):
pokemon = pokemon_repository.select(id)
name = request.form["name"]
trainer = request.form['trainer']
species = request.form['species']
hatched = request.form['hatched']
nurse_id = request.form['nurse_id']
nurse = nurse_repository.select(nurse_id)
notes = request.form['notes']
pokemon = Pokemon(name, trainer, species, hatched, nurse, notes)
pokemon.id = id
pokemon_repository.update(pokemon)
return redirect("/pokemons/index")
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
def show_nurses_pokemon(id):
pokemons = pokemon_repository.select_all()
nurse = nurse_repository.select(id)
return render_template("/staff/pokemon.html",
nurse=nurse,
pokemons=pokemons)
def edit_nurse_form(id):
nurse = nurse_repository.select(id)
return render_template("/staff/edit.html", nurse=nurse)
def edit_nurse(id):
nurse = nurse_repository.select(id)
return render_template('nurses/edit.html', nurse=nurse)
