Moved the files into a folder structure

1 files changed, 32 insertions, 0 deletions

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)