Skip to content

Commit 59e7ea7

Browse files
committed
fix: add error handling for lack of data
1 parent d3b0986 commit 59e7ea7

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

database/statistics.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,11 @@
1313
def set_game_won(user_id):
1414
if user_id > 0:
1515
db = Database()
16-
games_won = int(db.get_user(user_id)[6]) + 1
16+
user = db.get_user(user_id)
17+
if user is None:
18+
logger.warning("User '{}' is None - can't set won games!".format(user))
19+
return
20+
games_won = int(user[6]) + 1
1721
logger.debug("Add game won for user: {}".format(user_id))
1822
db.set_games_won(games_won, user_id)
1923

@@ -45,6 +49,11 @@ def get_user_stats(user_id):
4549
:return:
4650
"""
4751
user = Database().get_user(user_id)
52+
53+
if user is None:
54+
logger.warning("User '{}' is not stored in the database!".format(user_id))
55+
return "No statistics found!"
56+
4857
lang_id = Database().get_lang_id(user_id)
4958

5059
try:

0 commit comments

Comments
 (0)