blob: bf116538e75377f71d999c7fe42d4de1c525a94e (
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
|
from csv import writer
from random import randint
from names import get_full_name
from ..model import Player, MIN_LEVEL, MAX_LEVEL
PLAYER_COUNT = 80
def _generate_players(player_count: int) -> list[Player]:
players = []
for _ in range(player_count):
random_name = get_full_name()
random_skill = randint(MIN_LEVEL, MAX_LEVEL)
players.append(Player(name=random_name, skill=random_skill))
return players
def _write_players_to_csv(player_list: list[Player]) -> None:
with open("database_test.csv", "w", newline="") as file:
data_writer = writer(file)
data_writer.writerow(["name", "skill"])
for player in player_list:
data_writer.writerow([player.name, player.skill])
def main(player_count: int):
players = _generate_players(player_count)
_write_players_to_csv(players)
if __name__ == "__main__":
main(PLAYER_COUNT)
|
