diff options
| -rw-r--r-- | app.py | 7 | ||||
| -rw-r--r-- | config.py | 9 | ||||
| -rw-r--r-- | tools/generate_dummy_data.py | 40 |
3 files changed, 32 insertions, 24 deletions
@@ -8,11 +8,9 @@ from flask import Flask, render_template, request load_dotenv() -rg = RoundGenerator(courts=CourtList(12), players=PlayerList("test_data_large.csv")) - - app = Flask("Match Up! Backend") app.config.from_object(getenv("PROD_APP_SETTINGS", "config.DevelopmentConfig")) +rg = RoundGenerator(courts=CourtList(12), players=PlayerList("test_data_large.csv")) @app.route("/", methods=["GET"]) @@ -28,7 +26,8 @@ def get_random_games(): @app.route("/confirm", methods=["POST"]) def confirm_games(): - return render_template("games.j2", title=f"Round: {rg.round}", games=rg.confirm()) + games = rg.confirm() + return render_template("games.j2", title=f"Round: {rg.round}", games=games) @app.route("/clear", methods=["POST"]) @@ -5,8 +5,12 @@ class Config: ASSETS_DEBUG = False -class ProductionConfig(Config): - pass +class YouthConfig(Config): + DATABASE = "test_data_large.csv" + + +class SeniorConfig(Config): + DATABASE = "test_data_large.csv" class DevelopmentConfig(Config): @@ -14,3 +18,4 @@ class DevelopmentConfig(Config): DEVELOPMENT = True TEMPLATES_AUTO_RELOAD = True ASSETS_DEBUG = True + DATABASE = "test_data_large.csv" 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) |