aboutsummaryrefslogtreecommitdiff
path: root/gen_test_data.py
diff options
context:
space:
mode:
authorKaran Jayachandra <mail@karanjayachandra.com>2024-10-25 18:03:18 +0200
committerKaran Jayachandra <mail@karanjayachandra.com>2024-10-25 18:03:18 +0200
commit60a489cfb3fb7a12a06212c2c7ed3c05cd0757b5 (patch)
tree8bf7b9cc50d8b133fe0666b0bff7fa143cd095a5 /gen_test_data.py
parent2b18e0929a0b3e083cbe9e5e8ffd421ed8d17e52 (diff)
Moved the files into a folder structure
Diffstat (limited to 'gen_test_data.py')
-rw-r--r--gen_test_data.py32
1 files changed, 0 insertions, 32 deletions
diff --git a/gen_test_data.py b/gen_test_data.py
deleted file mode 100644
index a49594f..0000000
--- a/gen_test_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)