def addShelter():
if 'username' not in login_session:
flash ("Please login for more complete access to our puppies.")
return redirect ('/puppies')
if login_session['admin'] < 2:
flash ("You do not have the authority to access that page.")
return redirect ('/puppies')
form=NewShelterForm()
if form.validate_on_submit():
newShelter = Shelter(name=form.name.data, address=form.address.data,\
city=form.city.data, state=form.state.data, \
zipCode=form.zipcode.data, website=form.website.data, \
occupancy=form.occupancy.data, capacity=form.capacity.data, \
user_id=login_session['user_id'])
session.add(newShelter)
session.commit()
flash("You successfully added a shelter.")
return redirect(url_for('shelters'))
else:
return render_template('addShelter.html', form=form)
def editShelter(shelter_id):
if 'username' not in login_session:
flash ("Please login for more complete access to our puppies.")
return redirect ('/puppies')
if login_session['admin'] < 2:
flash ("You do not have the authority to access that page.")
return redirect ('/puppies')
form=NewShelterForm()
editedShelter = session.query(Shelter).filter_by(id=shelter_id).one()
if form.validate_on_submit():
session.query(Shelter).filter(Shelter.id==shelter_id).update({'name' :\
form.name.data, 'address' : form.address.data, 'city' : \
form.city.data, 'state' : form.state.data, 'zipCode' : \
form.zipcode.data, 'website' : form.website.data, 'capacity' :\
form.capacity.data, 'user_id' : login_session['user_id']})
session.commit()
flash("Shelter edited successfully!")
return redirect(url_for('shelters'))
else:
return render_template('editShelter.html', form=form,\
shelter=editedShelter)
