From d18e5880b47fcdf6542bcb9502e8cf35a4276f05 Mon Sep 17 00:00:00 2001 From: Karan Jayachandra Date: Mon, 14 Oct 2024 11:10:35 +0200 Subject: Cleand up the application code a bit --- tools/generate_dummy_data.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 tools/generate_dummy_data.py (limited to 'tools/generate_dummy_data.py') 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() -- cgit v1.3.1