aboutsummaryrefslogtreecommitdiff
path: root/src/match_up/gen_test_data.py
diff options
context:
space:
mode:
authorKaran Jayachandra <mail@karanjayachandra.com>2024-11-09 18:49:40 +0100
committerKaran Jayachandra <mail@karanjayachandra.com>2024-11-09 18:49:40 +0100
commit14df28d5b8d3d9d80f24c532e1c096f8ac997e5b (patch)
treeb0b41dd9a9f3d3cd94cb07535410aa99bdfe8507 /src/match_up/gen_test_data.py
parentbc587d676dca709c8ff8a21470a68938359a7ec8 (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.py32
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)