From 67e6c6344e2828bcda8700e44da3ff5c96bc035b Mon Sep 17 00:00:00 2001 From: Karan Jayachandra Date: Wed, 16 Oct 2024 11:25:08 +0200 Subject: Cleaned up the code to generate test data --- tools/generate_dummy_data.py | 40 ++++++++++++++++++++++------------------ 1 file changed, 22 insertions(+), 18 deletions(-) (limited to 'tools/generate_dummy_data.py') 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) -- cgit v1.3.1 From fbed8387a39e42cee7235384d7f8380ee158a3b3 Mon Sep 17 00:00:00 2001 From: Karan Jayachandra Date: Wed, 16 Oct 2024 11:25:55 +0200 Subject: Removed one folder --- gen_test_data.py | 32 ++++++++++++++++++++++++++++++++ tools/generate_dummy_data.py | 32 -------------------------------- 2 files changed, 32 insertions(+), 32 deletions(-) create mode 100644 gen_test_data.py delete mode 100644 tools/generate_dummy_data.py (limited to 'tools/generate_dummy_data.py') diff --git a/gen_test_data.py b/gen_test_data.py new file mode 100644 index 0000000..bf11653 --- /dev/null +++ b/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) diff --git a/tools/generate_dummy_data.py b/tools/generate_dummy_data.py deleted file mode 100644 index bf11653..0000000 --- a/tools/generate_dummy_data.py +++ /dev/null @@ -1,32 +0,0 @@ -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) -- cgit v1.3.1