From 0a20788e82e86f3c826acf6b11c470b63c83ed8b Mon Sep 17 00:00:00 2001
From: emkael <emkael@tlen.pl>
Date: Sat, 17 Feb 2018 12:31:50 +0100
Subject: Refactoring board count fetching

---
 jfr_playoff/matchinfo.py | 39 ++++++++++++++++++++++++++-------------
 1 file changed, 26 insertions(+), 13 deletions(-)

(limited to 'jfr_playoff')

diff --git a/jfr_playoff/matchinfo.py b/jfr_playoff/matchinfo.py
index a7e5324..819cdd7 100644
--- a/jfr_playoff/matchinfo.py
+++ b/jfr_playoff/matchinfo.py
@@ -105,23 +105,36 @@ class MatchInfo:
                 self.info.teams[i].score = self.config['score'][i]
             self.info.running = -1
 
+    def __get_db_board_count(self):
+        towels = self.database.fetch(
+            self.config['database'], p_sql.TOWEL_COUNT,
+            (self.config['table'], self.config['round']))
+        row = [0 if r is None
+               else r for r in
+               self.database.fetch(
+                   self.config['database'], p_sql.BOARD_COUNT,
+                   (self.config['table'], self.config['round']))]
+        boards_to_play = row[0]
+        if row[1] > 0:
+            boards_played = int(row[1])
+        if boards_to_play > 0:
+            boards_played += int(towels[0])
+        return boards_played, boards_to_play
+
     def __fetch_board_count(self):
+        boards_played = 0
+        boards_to_play = 0
         try:
-            towels = self.database.fetch(
-                self.config['database'], p_sql.TOWEL_COUNT,
-                (self.config['table'], self.config['round']))
-            row = [0 if r is None
-                   else r for r in
-                   self.database.fetch(
-                       self.config['database'], p_sql.BOARD_COUNT,
-                       (self.config['table'], self.config['round']))]
-            if row[1] > 0:
-                self.info.running = int(row[1])
-            if row[0] > 0:
-                if row[1] >= row[0] - towels[0]:
-                    self.info.running = -1
+            boards_played, boards_to_play = self.__get_db_board_count()
         except (mysql.connector.Error, TypeError, KeyError):
             pass
+        if boards_played > 0:
+            self.info.running = -1 \
+                                if boards_played >= boards_to_play \
+                                   else boards_played
+        else:
+            self.info.running = 0
+
 
     def __determine_outcome(self):
         if (self.info.running == -1):
-- 
cgit v1.2.3