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/")