from os import remove from os.path import isfile from random import randint from pathlib import PurePath from names import get_first_name, get_last_name from match_up.model import Player, PlayerDatabase, MIN_SKILL, MAX_SKILL TEST_DATABASE_LOCATION = PurePath("test_storage.json") def generate_test_database(total_players: int = 20) -> None: try: if isfile(TEST_DATABASE_LOCATION): remove(TEST_DATABASE_LOCATION) db = PlayerDatabase(TEST_DATABASE_LOCATION) for _ in range(total_players): db.add_player( Player( first_name=get_first_name(), last_name=get_last_name(), skill=randint(MIN_SKILL, MAX_SKILL), ) ) finally: db.close() def activate_random_players(activate_players: int = 17) -> None: try: db = PlayerDatabase(TEST_DATABASE_LOCATION) player_list = db.get_players() for index, player in enumerate(player_list): if index < activate_players: db.activate_player(player) finally: db.close() def main(): pass if __name__ == "__main__": main()