1 files changed, 22 insertions, 18 deletions
diff --git a/tools/generate_dummy_data.py b/tools/generate_dummy_data.py
index 3d6d116..bf11653 100644
--- a/tools/generate_dummy_data.py
+++ b/tools/generate_dummy_data.py
@@ -1,28 +1,32 @@
 from csv import writer
-from model import Player
+from random import randint
 from names import get_full_name
-from random import randint, getrandbits
+from ..model import Player, MIN_LEVEL, MAX_LEVEL
 
+PLAYER_COUNT = 80
 
-def get_test_players(total_player: int = 80, min_skill: int = 1, max_skill: int = 10):
-    return [
-        Player(
-            name=get_full_name(),
-            skill=randint(min_skill, max_skill),
-            status=bool(getrandbits(1)),
-        )
-        for _ in range(total_player)
-    ]
 
+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 main():
-    dummy_list = get_test_players()
-    with open("test_data_large.csv", "w", newline="") as file:
+
+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", "status"])
-        for player in dummy_list:
-            data_writer.writerow([player.name, player.skill, player.status])
+        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()
+    main(PLAYER_COUNT)