Cleand up the application code a bit

1 files changed, 28 insertions, 0 deletions

diff --git a/tools/generate_dummy_data.py b/tools/generate_dummy_data.py
new file mode 100644
index 0000000..3d6d116
--- /dev/null
+++ b/tools/generate_dummy_data.py
@@ -0,0 +1,28 @@
+from csv import writer
+from model import Player
+from names import get_full_name
+from random import randint, getrandbits
+
+
+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 main():
+    dummy_list = get_test_players()
+    with open("test_data_large.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])
+
+
+if __name__ == "__main__":
+    main()