diff options
Diffstat (limited to 'model.py')
| -rw-r--r-- | model.py | 11 |
1 files changed, 5 insertions, 6 deletions
@@ -18,7 +18,6 @@ class CourtList: def __post_init__(self): self.courts = [Court(i + 1, True) for i in range(self.total)] - self.active = self.total def get_court(self, court_number: int): return [court for court in self.courts if court.number == court_number] @@ -35,7 +34,6 @@ class CourtList: ) for court in self.courts ] - self.active = sum([court.status for court in self.courts]) @dataclass @@ -45,16 +43,17 @@ class Player: status: bool = True +def create_player_from_row(row: list[str]) -> Player: + return Player(name=row[0], skill=row[1], status=(row[2] == "True")) + + class PlayerList: def __init__(self, location: Path = "players.csv"): - players = [] with open(location, newline="") as file: data_reader = reader(file) next(data_reader, None) - for row in data_reader: - player = Player(name=row[0], skill=row[1], status=(row[2] == "True")) - players.append(player) + players = [create_player_from_row(row) for row in data_reader] players.sort(key=attrgetter("name")) self.players = players |