ban: Wrapped DB queries in try-except clause to avoid tracebacks when…

… leaver ban is on.
MinoMino · Jan 14, 2016 · 849a2a1 · 849a2a1
1 parent 753499e
commit 849a2a1
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/ban.py b/ban.py
@@ -359,13 +359,14 @@ def leave_status(self, steam_id):
         if not self.get_cvar("qlx_leaverBan", bool):
             return None
 
-        completed = self.db[PLAYER_KEY.format(steam_id) + ":games_completed"]
-        left = self.db[PLAYER_KEY.format(steam_id) + ":games_left"]
-        if completed is None or left is None:
+        try:
+            completed = self.db[PLAYER_KEY.format(steam_id) + ":games_completed"]
+            left = self.db[PLAYER_KEY.format(steam_id) + ":games_left"]
+        except KeyError:
             return None
-        else:
-            completed = int(completed)
-            left = int(left)
+
+        completed = int(completed)
+        left = int(left)
 
         min_games_completed = self.get_cvar("qlx_leaverBanMinimumGames", int)
         warn_threshold = self.get_cvar("qlx_leaverBanWarnThreshold", float)