From d18e5880b47fcdf6542bcb9502e8cf35a4276f05 Mon Sep 17 00:00:00 2001 From: Karan Jayachandra Date: Mon, 14 Oct 2024 11:10:35 +0200 Subject: Cleand up the application code a bit --- model.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) (limited to 'model.py') diff --git a/model.py b/model.py index a623d0a..c83bcfd 100644 --- a/model.py +++ b/model.py @@ -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 -- cgit v1.3.1