def update(bar_id):
if request.method == 'POST':
image = request.files['image']
secure_image_name = secure_filename(image.filename)
db_bar = Bar.query.get(request.form['id'])
db_bar.name = request.form['name']
db_bar.description = request.form['description']
db_bar.speciality = request.form['speciality']
db_bar.phone = request.form['phone']
db_bar.address = request.form['address']
if secure_image_name:
db_bar.image = secure_image_name
db_session.commit()
return redirect(url_for("index"))
else:
bar = Bar.query.get(bar_id)
return render_template("update_bar.html", bar=bar)
def add():
if request.method == 'POST':
print request.host_url
print request.values.get('next')
print request.referrer
validate_bar_registering()
image = request.files['image']
secure_image_name = secure_filename(image.filename)
bar = Bar(request.form['name'], request.form['description'],
request.form['speciality'], request.form['phone'], request.form['address'], secure_image_name)
if secure_image_name:
path = os.path.join(app.config['UPLOAD_FOLDER'], secure_image_name)
image.save(path)
db_session.add(bar)
db_session.commit()
return redirect(url_for("index"))
else:
return render_template("add_bar.html")
def remove(bar_id):
bar_to_delete = Bar.query.get(bar_id)
db_session.delete(bar_to_delete)
db_session.commit()
return redirect(url_for("index"))
