From 60a489cfb3fb7a12a06212c2c7ed3c05cd0757b5 Mon Sep 17 00:00:00 2001 From: Karan Jayachandra Date: Fri, 25 Oct 2024 18:03:18 +0200 Subject: Moved the files into a folder structure --- src/match_up/gen_test_data.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 src/match_up/gen_test_data.py (limited to 'src/match_up/gen_test_data.py') diff --git a/src/match_up/gen_test_data.py b/src/match_up/gen_test_data.py new file mode 100644 index 0000000..a49594f --- /dev/null +++ b/src/match_up/gen_test_data.py @@ -0,0 +1,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) -- cgit v1.3.1