diff options
| author | Karan Jayachandra <mail@karanjayachandra.com> | 2024-11-09 18:49:40 +0100 |
|---|---|---|
| committer | Karan Jayachandra <mail@karanjayachandra.com> | 2024-11-09 18:49:40 +0100 |
| commit | 14df28d5b8d3d9d80f24c532e1c096f8ac997e5b (patch) | |
| tree | b0b41dd9a9f3d3cd94cb07535410aa99bdfe8507 /src/match_up/gen_test_data.py | |
| parent | bc587d676dca709c8ff8a21470a68938359a7ec8 (diff) | |
Added the random name generation to be part of the package
Diffstat (limited to 'src/match_up/gen_test_data.py')
| -rw-r--r-- | src/match_up/gen_test_data.py | 32 |
1 files changed, 0 insertions, 32 deletions
diff --git a/src/match_up/gen_test_data.py b/src/match_up/gen_test_data.py deleted file mode 100644 index a49594f..0000000 --- a/src/match_up/gen_test_data.py +++ /dev/null @@ -1,32 +0,0 @@ -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) |
