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)