def map_pool(id):
season = Season(id)
season.load()
form = MapPoolForm(request.form)
form.map_id.choices = season.get_maps_not_in_pool()
if request.method == 'POST' and form.validate_on_submit():
query = """
INSERT INTO season_maps(season_id, map_id)
VALUES(?, ?)
"""
values = (form.season_id.data, form.map_id.data)
cur = g.db.execute(query, values)
if cur.rowcount:
flash("Map added to the pool")
g.db.commit()
else:
flash("Failed to add map")
g.db.rollback()
form.season_id.data = id
form.map_id.choices = season.get_maps_not_in_pool()
return render_template("admin/map_pool.html", form=form, season=season)
