blob: 3ee6e4208b33427f288b060ecd04d8872430edac (
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
|
from csv import writer
from model import Player
from names import get_full_name
from random import randint, getrandbits
def get_test_players(total_player: int = 20, min_skill: int = 1, max_skill: int = 10):
return [
Player(
name=get_full_name(),
skill=randint(min_skill, max_skill),
status=bool(getrandbits(1)),
)
for _ in range(total_player)
]
def main():
dummy_list = get_test_players()
with open("test_data.csv", "w", newline="") as file:
data_writer = writer(file)
data_writer.writerow(["name", "skill", "status"])
for player in dummy_list:
data_writer.writerow([player.name, player.skill, player.status])
if __name__ == "__main__":
main()
|
