add setting of winning points to startup UI

Author: Janfiderheld

model/movie_handler.py

@@ -54,7 +54,6 @@ def add_new_movie(self, new_m: Movie):
             return
 
         self.__movies.append(new_m)
-        print(f"Movie {str(new_m)} is added with img: {new_m.get_img()}")
 
     def return_rand_movie(self) -> Movie:
         if len(self.__current_movies) == 0:
@@ -65,3 +64,6 @@ def return_rand_movie(self) -> Movie:
 
     def reset_current_movie_list(self):
         self.__current_movies = self.__movies.copy()
+
+    def get_movie_amount(self) -> int:
+        return len(self.__movies)

model/player.py

@@ -4,29 +4,23 @@
 
 
 class Player:
-    MAX_TOKENS = 2
-    POINTS_TO_WIN = 8
+    max_points = 8
 
     def __init__(self, name: str):
         self.__name = name
-        self.__tokens = Player.MAX_TOKENS
         self.__timeline: List[Movie] = []
         self.get_starting_movie()
 
     def get_name(self) -> str:
         return self.__name
 
-    def get_current_token(self) -> int:
-        return self.__tokens
-
     def get_timeline(self) -> List[Movie]:
         return self.__timeline
 
     def check_win(self) -> bool:
-        return len(self.__timeline) >= self.POINTS_TO_WIN
+        return len(self.__timeline) >= self.max_points
 
     def reset(self):
-        self.__tokens = Player.MAX_TOKENS
         self.get_starting_movie()
 
     def get_starting_movie(self):
@@ -60,4 +54,4 @@ def add_to_timeline(self, m: Movie, pos: int):
         self.__timeline.insert(pos, m)
 
     def __str__(self) -> str:
-        return f"{self.__name} ({self.__tokens}/{Player.MAX_TOKENS} Tokens) - Timeline: {len(self.__timeline)}/{self.POINTS_TO_WIN}"
+        return f"{self.__name} - Timeline: {len(self.__timeline)}/{self.max_points}"

ui/startup.py

@@ -2,7 +2,7 @@
 
 import ui
 from data import write_top250_to_csv, add_movies_from_top250
-from model import PlayerHandler, Player
+from model import PlayerHandler, Player, MovieHandler
 
 
 class StartupUI(QtWidgets.QWidget):
@@ -11,6 +11,7 @@ def __init__(self):
         self.__game_window = QtWidgets.QWidget()
         self.__add_window = QtWidgets.QWidget()
         self.__player_handler = PlayerHandler()
+        self.__movie_handler = MovieHandler()
 
         self.__btn_start = QtWidgets.QPushButton("Start Game")
         self.__btn_start.clicked.connect(self.start_game)
@@ -28,13 +29,21 @@ def __init__(self):
         self.__txt_header = QtWidgets.QLabel("Current Players:")
         self.__ed_player_name = QtWidgets.QLineEdit()
 
+        self.__txt_points = QtWidgets.QLabel("Timeline length for Winning:")
+        self.__ed_points = QtWidgets.QLineEdit(str(Player.max_points))
+
         self.layout = QtWidgets.QVBoxLayout(self)
         self.layout_btn = QtWidgets.QHBoxLayout(self)
+        self.layout_points = QtWidgets.QHBoxLayout(self)
 
         self.layout_btn.addWidget(self.__ed_player_name)
         self.layout_btn.addWidget(self.__btn_add_player)
         self.layout_btn.addWidget(self.__btn_rem_player)
 
+        self.layout_points.addWidget(self.__txt_points)
+        self.layout_points.addWidget(self.__ed_points)
+
+        self.layout.addLayout(self.layout_points)
         self.layout.addWidget(self.__txt_header)
         self.layout.addWidget(self.__player_list)
         self.layout.addLayout(self.layout_btn)
@@ -54,11 +63,33 @@ def start_game(self):
             self.__btn_start = dlg.exec()
             return
 
+        if not self.__ed_points.text():
+            dlg = QtWidgets.QMessageBox(self)
+            dlg.setWindowTitle("Error")
+            dlg.setText("There are no points specified")
+            self.__btn_start = dlg.exec()
+            return
+
+        if not self.__ed_points.text().isdigit():
+            dlg = QtWidgets.QMessageBox(self)
+            dlg.setWindowTitle("Error")
+            dlg.setText("The points are not a number")
+            self.__btn_start = dlg.exec()
+            return
+
+        if int(self.__ed_points.text())-1 * self.__player_list.count() > self.__movie_handler.get_movie_amount():
+            dlg = QtWidgets.QMessageBox(self)
+            dlg.setWindowTitle("Error")
+            dlg.setText("There are not enough movies for that many points")
+            self.__btn_start = dlg.exec()
+            return
+
         for idx in range(self.__player_list.count()):
             i = self.__player_list.item(idx)
             p = Player(i.text())
             self.__player_handler.add_new_player(p)
 
+        Player.max_points = int(self.__ed_points.text())
         self.__game_window = ui.GameUI()
         self.close()