def addspot():
form = SpotForm(request.form)
if request.method == "POST" and form.validate_on_submit():
spot_name = form.spot_name.data
logo = request.files.get("spot_logo")
spot_loc = form.spot_loc.data
better_season = form.better_season.data
tickets = form.tickets.data
content = form.content.data
try:
spot_logo = photos.save(logo) ## 保存图片
except UploadNotAllowed:
flash("The upload was not allowed")
spot = Spot(
spot_name=spot_name,
spot_loc=spot_loc,
spot_logo=spot_logo,
better_season=better_season,
tickets=tickets,
content=content,
)
try:
spot._store_to_db()
flash(u"数据保存成功!")
except:
flash(u"保存失败")
return redirect(url_for("user.index"))
return render_template("user/addspot.html", form=form)
def spot_edit(id):
spot = Spot.query.filter_by(id=id).first()
if not spot:
abort(404)
form = SpotForm(
spot_name=spot.spot_name,
spot_loc=spot.spot_loc,
better_season=spot.better_season,
tickets=spot.tickets,
content=spot.content,
)
if request.method == "POST" and form.validate_on_submit():
logo = request.files.get("spot_logo")
try:
spot_logo = photos.save(logo)
except UploadNotAllowed:
flash("The upload was not allowed")
Spot.query.filter_by(id=id).update(
{
Spot.spot_name: request.form["spot_name"],
Spot.spot_logo: spot_logo,
Spot.spot_loc: request.form["spot_loc"],
Spot.better_season: request.form["better_season"],
Spot.tickets: request.form["tickets"],
Spot.content: request.form["content"],
}
)
db.session.commit()
return redirect(url_for("user.spot_manager"))
return render_template("user/spotedit.html", form=form)
