blob: 53119eb4be6802e8ffc939bb1a95e9a3f2de09c1 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
from csv import reader
from pathlib import Path
from dataclasses import dataclass
@dataclass
class Player:
name: str
skill: int
status: bool
class PlayerList:
def __init__(self, location: Path = "players.csv"):
with open(location, newline="") as file:
data_reader = reader(file)
next(data_reader, None)
self.players = [
(
Player(
name=row[0],
skill=row[1],
status=(row[2] == "True"),
)
)
for row in data_reader
]
def get_player(self, player_request: str):
return [player for player in self.players if player.name == player_request]
def toggle_player(self, player_request: Player):
self.players = [
(
Player(
name=player.name,
skill=player.skill,
status=not player.status,
)
if player.name == player_request
else player
)
for player in self.players
]
def __str__(self):
display_string = ""
for player in self.players:
display_string += player.__str__() + "\n"
return display_string
|
