diff --git a/flaskr/templates/games.html b/flaskr/templates/games.html index 9966393..4122206 100644 --- a/flaskr/templates/games.html +++ b/flaskr/templates/games.html @@ -8,6 +8,8 @@ table, tr, td { } +

There is {{count}} games.

+ {% if games_exist %} diff --git a/flaskr/utils/games.py b/flaskr/utils/games.py index 0514f92..27631a5 100644 --- a/flaskr/utils/games.py +++ b/flaskr/utils/games.py @@ -4,7 +4,9 @@ from flaskr.database import db def query_games(): sql = "SELECT name, id FROM games" result = db.session.execute(sql).fetchall() - return result + sql = "SELECT count(games.id) FROM games" + count = db.session.execute(sql).fetchone() + return result, count[0] def query_game_by_name(game_name: str): diff --git a/flaskr/views/games.py b/flaskr/views/games.py index f01a6ec..fcb101f 100644 --- a/flaskr/views/games.py +++ b/flaskr/views/games.py @@ -10,10 +10,10 @@ from flaskr.utils.games import ( @app.route("/games") def get_games(): - games = query_games() + games, count = query_games() if games is None: - return render_template("games.html", games_exist=False) - return render_template("games.html", games=games, games_exist=True) + return render_template("games.html", games_exist=False, count=count) + return render_template("games.html", games=games, games_exist=True, count=count) @app.route("/games/")