aboutsummaryrefslogtreecommitdiff
path: root/dummy_data_generation.py
diff options
context:
space:
mode:
Diffstat (limited to 'dummy_data_generation.py')
-rw-r--r--dummy_data_generation.py28
1 files changed, 28 insertions, 0 deletions
diff --git a/dummy_data_generation.py b/dummy_data_generation.py
new file mode 100644
index 0000000..3d6d116
--- /dev/null
+++ b/dummy_data_generation.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()