From 5260b6eba817bd3fefd5be83fe9ed4a497473294 Mon Sep 17 00:00:00 2001 From: legonzaur Date: Sun, 12 May 2024 11:07:40 +0200 Subject: [PATCH] Add game.kill_team; attemp to fix #11 --- src/model/game.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/src/model/game.py b/src/model/game.py index e68f22c..322b84d 100644 --- a/src/model/game.py +++ b/src/model/game.py @@ -190,3 +190,19 @@ class Game(Serializable): def end_game(self): self.lobby.remove(self) self.ended = True + + def kill_team(self, team: Team): + + if team == self.current_turn: + self.end_turn() + + for p in self.players: + if p.team != team: + continue + self.players.remove(p) + + if team in self.teams: + self.teams.remove(team) + + if len(self.teams) <= 1: + self.end_game()