diff --git a/flaskr/templates/courses.html b/flaskr/templates/courses.html index 302dd99..365eaf8 100644 --- a/flaskr/templates/courses.html +++ b/flaskr/templates/courses.html @@ -10,6 +10,8 @@ } +

There is {{count}} courses.

+ {% if courses_exist %} diff --git a/flaskr/utils/courses.py b/flaskr/utils/courses.py index d2be7f4..30ad86a 100644 --- a/flaskr/utils/courses.py +++ b/flaskr/utils/courses.py @@ -2,9 +2,15 @@ from flaskr.database import db def query_courses(): - sql = "SELECT courses.name, games.name, cups.name FROM courses JOIN games ON games.id=courses.game_id JOIN cups ON cups.id=courses.cup_id" + sql = """ + SELECT courses.name, games.name, cups.name + FROM courses + JOIN games ON games.id=courses.game_id + JOIN cups ON cups.id=courses.cup_id""" result = db.session.execute(sql).fetchall() - return result + sql = "SELECT count(courses.id) FROM courses" + count = db.session.execute(sql).fetchone() + return result, count[0] def query_course_by_name(course_name: str): diff --git a/flaskr/views/courses.py b/flaskr/views/courses.py index 507c2ab..d75b94c 100644 --- a/flaskr/views/courses.py +++ b/flaskr/views/courses.py @@ -6,10 +6,12 @@ from flaskr.utils.courses import query_courses, query_course_by_name @app.route("/courses") def get_courses(): - courses = query_courses() + courses, count = query_courses() if courses is None: - return render_template("courses.html", courses_exist=False) - return render_template("courses.html", courses=courses, courses_exist=True) + return render_template("courses.html", courses_exist=False, count=count) + return render_template( + "courses.html", courses=courses, courses_exist=True, count=count + ) @app.route("/courses/")