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)