diff options
Diffstat (limited to 'tools/generate_dummy_data.py')
| -rw-r--r-- | tools/generate_dummy_data.py | 28 |
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() |
