def edit_shelter(shelter_id): shelter = Shelter.query.filter_by(id=shelter_id).one() form = ShelterForm(obj=shelter) form.address['address'].data = shelter.address form.address['city'].data = shelter.city form.address['state'].data = shelter.state form.address['zip_code'].data = shelter.zipCode if request.method == 'POST' and form.validate_on_submit(): if request.form['name'] and request.form['name'] != shelter.name: shelter.name = request.form['name'] if request.form['address'] and request.form['address'] != shelter.address: shelter.address = request.form['address'] if request.form['city'] and request.form['city'] != shelter.city: shelter.city = request.form['city'] if request.form['state'] and request.form['state'] != shelter.state: shelter.state = request.form['state'] if request.form['zipcode'] and request.form['zipcode'] != shelter.zipCode: shelter.zipCode = request.form['zipcode'] if request.form['website'] and request.form['website'] != shelter.website: shelter.website = request.form['website'] if request.form['maxcap'] and request.form['maxcap'] != shelter.maximum_capacity: shelter.maximum_capacity = request.form['maxcap'] db.session.add(shelter) db.session.commit() print("Shelter {} edited!".format(shelter_id)) flash("Shelter {} edited!".format(shelter_id)) return redirect(url_for('shelters')) else: return render_template('editshelter.html', shelter=shelter, form=form)
def edit_shelter(shelter_id): shelter = Shelter.query.filter_by(id=shelter_id).one() form = ShelterForm(obj=shelter) form.address['address'].data = shelter.address form.address['city'].data = shelter.city form.address['state'].data = shelter.state form.address['zip_code'].data = shelter.zipCode if request.method == 'POST' and form.validate_on_submit(): if request.form['name'] and request.form['name'] != shelter.name: shelter.name = request.form['name'] if request.form[ 'address'] and request.form['address'] != shelter.address: shelter.address = request.form['address'] if request.form['city'] and request.form['city'] != shelter.city: shelter.city = request.form['city'] if request.form['state'] and request.form['state'] != shelter.state: shelter.state = request.form['state'] if request.form[ 'zipcode'] and request.form['zipcode'] != shelter.zipCode: shelter.zipCode = request.form['zipcode'] if request.form[ 'website'] and request.form['website'] != shelter.website: shelter.website = request.form['website'] if request.form['maxcap'] and request.form[ 'maxcap'] != shelter.maximum_capacity: shelter.maximum_capacity = request.form['maxcap'] db.session.add(shelter) db.session.commit() print("Shelter {} edited!".format(shelter_id)) flash("Shelter {} edited!".format(shelter_id)) return redirect(url_for('shelters')) else: return render_template('editshelter.html', shelter=shelter, form=form)
def new_shelter(): form = ShelterForm(request.form) if request.method == 'POST' and form.validate_on_submit(): # if request.form['name']: # name = request.form['name'] # else: # raise AttributeError("name must not be null!") # if request.form['address']: # address = request.form['address'] # else: # address = None # if request.form['city']: # city = request.form['city'] # else: # city = None # if request.form['state']: # state = request.form['state'] # else: # state = None # if request.form['zipcode']: # zipCode = request.form['zipcode'] # else: # zipCode = None # if request.form['website']: # website = request.form['website'] # else: # website = None # if request.form['maxcap']: # maxcap = request.form['maxcap'] # else: # maxcap = 0 new_shelter = Shelter(name=form.name.data, address=form.address['address'].data, city=form.address['city'].data, state=form.address['state'].data, zipCode=form.address['zip_code'].data, website=form.website.data, maximum_capacity=form.maximum_capacity.data) db.session.add(new_shelter) db.session.commit() print("Shelter created!") flash("Shelter created!") return redirect(url_for('shelters')) else: return render_template('newshelter.html', form=form)