tfjm-discord-bot/src/constants.py

import os
from pathlib import Path

__all__ = [
    "TOKEN",
    "Role",
    "PROBLEMS",
    "MAX_REFUSE",
    "ROUND_NAMES",
    "TIRAGES_FILE",
    "TEAMS_FILE",
    "JOKES_FILE",
    "TEAMS_CHANNEL_CATEGORY",
    "DIEGO",
    "TOURNOIS",
]

TOKEN = os.environ.get("TFJM_DISCORD_TOKEN")

if TOKEN is None:
    print("No token for the bot were found.")
    print("You need to set the TFJM_DISCORD_TOKEN variable in your environement")
    print("Or just run:")
    print()
    print(f'    TFJM_DISCORD_TOKEN="your token here" python tfjm-discord-bot.py')
    print()
    quit(1)

GUILD = "690934836696973404"
DIEGO = "Diego"  # Mon display name

TOURNOIS = [
    "Lille",
    "Lyon",
    "Paris-Saclay",
    "Avignon",
    "Paris-Est",
    "Tours",
    "Bordeaux",
    "Nancy",
    "Rennes",
]


class Role:
    CNO = "CNO"
    DEV = "dev"
    ORGA = "Orga"
    ORGAS = tuple(f"Orga {t}" for t in TOURNOIS)
    BENEVOLE = "Bénévole"
    CAPTAIN = "Capitaine"
    PARTICIPANT = "Participant"


TEAMS_CHANNEL_CATEGORY = "Channels d'équipes"

ROUND_NAMES = ["premier tour", "deuxième tour"]

TOP_LEVEL_DIR = Path(__file__).parent.parent
TIRAGES_FILE = TOP_LEVEL_DIR / "data" / "tirages.yaml"
TEAMS_FILE = TOP_LEVEL_DIR / "data" / "teams"
JOKES_FILE = TOP_LEVEL_DIR / "data" / "jokes"

with open(TOP_LEVEL_DIR / "data" / "problems") as f:
    PROBLEMS = f.read().splitlines()
MAX_REFUSE = len(PROBLEMS) - 4  # -5 usually but not in 2020 because of covid-19