def add_track():
"""
Add a department to the database
"""
check_admin()
add_track = True
form = TrackForm()
if form.validate_on_submit():
track = Track(name=form.name.data)
try:
# add track to the database
db.session.add(track)
db.session.commit()
flash('You have successfully added a new track.')
except:
# in case department name already exists
db.session.rollback()
flash('Error: track name already exists.')
# redirect to departments page
return redirect(url_for('admin.list_tracks'))
# load department template
return render_template('admin/tracks/track.html',
action="Add",
add_track=add_track,
form=form,
title="Add track")
def edit_track(id):
"""
Edit a department
"""
check_admin()
add_track = False
track = Track.query.get_or_404(id)
form = TrackForm(obj=track)
if form.validate_on_submit():
track.track_id = form.track_id.data
track.name = form.name.data
track.description = form.description.data
db.session.commit()
flash('You have successfully edited the track.')
# redirect to the departments page
return redirect(url_for('admin.list_tracks'))
form.description.data = track.description
form.name.data = track.name
return render_template('admin/tracks/track.html',
action="Edit",
add_track=add_track,
form=form,
track=track,
title="Edit track")
def index():
form = TrackForm()
if form.validate_on_submit():
tracking = Track (
weight = form.weight.data,
happy = form.happy.data,
diet = form.diet.data,
exercise = form.exercise.data,
floss = form.floss.data,
meditation = form.meditation.data,
note = form.note.data,
timestamp = datetime.utcnow(),
author = g.user.to_dbref())
tracking.save()
flash('Your post is now live!')
calculate_weightAvg_async(g.user.to_dbref())
return redirect(url_for('index'))
posts = Track.objects(author=g.user)
analysis = Analysis.objects(author=g.user).first()
return render_template("index.html",
title = 'Home',
form = form,
posts = posts,
analysis = analysis)
