From b9967b20dddaed6673d2333edc7cdee735ed1395 Mon Sep 17 00:00:00 2001 From: Karan Jayachandra Date: Mon, 2 Dec 2024 19:43:33 +0100 Subject: Added a random selection even for teams --- src/match_up/utilities.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'src/match_up/utilities.py') diff --git a/src/match_up/utilities.py b/src/match_up/utilities.py index dd9568c..7d7de08 100644 --- a/src/match_up/utilities.py +++ b/src/match_up/utilities.py @@ -2,7 +2,7 @@ from math import floor from typing import Tuple from random import shuffle from pandas import read_csv -from random import randint, getrandbits +from random import randint, getrandbits, sample from names import get_first_name, get_last_name from match_up.data import Player, DisplayPlayer, Team, Game from match_up.data import ( @@ -71,7 +71,7 @@ def _create_groups_of_four_players( return court_players -def _create_games(courts: dict, players: list[Player], levels: int) -> list[Game]: +def _create_shuffle_games(courts: dict, players: list[Player], levels: int) -> list[Game]: court_players = _create_groups_of_four_players(players, levels) games = [] for court, players in zip(courts, court_players): @@ -88,7 +88,9 @@ def _create_team_games(courts: dict, teams: dict) -> tuple[list[Game], list[Play games = [] selected_players = [] for court, (_, players) in zip(courts, teams.items()): - selected_players.append(players[:3]) + players = sample(players, PLAYER_PER_COURT) + selected_players.append(players) + players.sort(key=lambda x : x.level, reverse=True) team1 = Team( player1=DisplayPlayer(players[0]), player2=DisplayPlayer(players[3]) ) -- cgit v1.3.1