diff options
Diffstat (limited to 'src/match_up/utilities.py')
| -rw-r--r-- | src/match_up/utilities.py | 8 |
1 files changed, 5 insertions, 3 deletions
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]) ) |