From aafd477d554f8ee8ded323656a2d015e5e39e4f9 Mon Sep 17 00:00:00 2001 From: Xevion Date: Mon, 20 Jan 2020 16:12:30 -0600 Subject: [PATCH] start adding HTML viewing options --- app/routes.py | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/app/routes.py b/app/routes.py index 91130d5..be7e855 100644 --- a/app/routes.py +++ b/app/routes.py @@ -1,6 +1,22 @@ +from flask import send_from_directory, redirect, url_for, render_template, request +from app.models import Season, Episode from app import app -from flask import send_from_directory, redirect, url_for, render_template @app.route('/') def index(): - return 'WIP' \ No newline at end of file + return 'WIP' + +@app.route('/view') +def view(): + seasonID = request.args.get('season') + episodeNUM = request.args.get('episode') + + season = Season.query.get(int(seasonID)) + episode = Episode.query.filter_by(season_id=season.id, number=int(episodeNUM)) + + if season: + if episode: + return render_template('episode.html') + else: + return render_template('season.html', season=Season.query.get()) + return redirect(url_for('index')) \ No newline at end of file