tfjm-discord-bot/src/constants.py

import os
from pathlib import Path
from time import time

__all__ = [
    "DISCORD_TOKEN",
    "TFJM_TOKEN",
    "Role",
    "PROBLEMS",
    "MAX_REFUSE",
    "ROUND_NAMES",
    "TEAMS_CHANNEL_CATEGORY",
    "DIEGO",
    "ANANAS",
    "YOHANN",
    "FAN_CLUBS",
    "BOT",
    "TOURNOIS",
    "EMBED_COLOR",
    "FRACTAL_URL",
    "FRACTAL_COOLDOWN",
    "File",
    "Emoji",
]


DISCORD_TOKEN = os.environ.get("TFJM_DISCORD_TOKEN")
TFJM_TOKEN = os.environ.get("TFJM_ORG_TOKEN")


if DISCORD_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 = 430566197868625920  # Mon id
ANANAS = 619132180408303616
YOHANN = 157252601119899648

FAN_CLUBS = {
    DIEGO: 706586020841259078,
    ANANAS: 706586027535368223,
    YOHANN: 711707591847444610,
}

BOT = 703305132300959754
TEAMS_CHANNEL_CATEGORY = "Channels d'équipes 2"
EMBED_COLOR = 0xFFA500
FRACTAL_URL = "https://thefractal.space/img/{seed}.png?size=1000"
FRACTAL_COOLDOWN = 30  # seconds

ROUND_NAMES = ["premier tour", "deuxième tour"]
TOURNOIS = [
    "Lille",
    "Lyon",
    "Paris-Saclay",
    "Paris-Avignon-Est",
    "Tours",
    "Bordeaux-Nancy",
    "Rennes",
    "Finale",
]


class Role:
    CNO = "CNO"
    DEV = "dev"
    ORGA = "Orga"
    ORGAS = tuple(f"Orga {t}" for t in TOURNOIS)
    JURY = tuple(f"Jury {t}" for t in TOURNOIS)
    BENEVOLE = "Bénévole"
    CAPTAIN = "Capitaine"
    FINALISTE = "Finaliste"
    PARTICIPANT = "Participant"
    TOURIST = "Touriste"
    PRETRESSE_CALINS = "Grande prêtresse des câlins"


class Emoji:
    HEART = "❤️"
    JOY = "😂"
    SOB = "😭"
    BIN = "🗑️"
    DICE = "🎲"
    CHECK = "✅"
    CROSS = "❌"
    PLUS_1 = "👍"
    MINUS_1 = "👎"
    RAINBOW_HEART = "<:rainbow_heart:714172834632564818>"


class File:
    TOP_LEVEL = Path(__file__).parent.parent
    TIRAGES = TOP_LEVEL / "data" / "tirages.yaml"
    TEAMS = TOP_LEVEL / "data" / "teams"
    JOKES = TOP_LEVEL / "data" / "jokes"
    JOKES_V2 = TOP_LEVEL / "data" / "jokesv2"
    MEMES = TOP_LEVEL / "data" / "memes"
    HUGS = TOP_LEVEL / "data" / "hugs"


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


def setup(bot):
    # Just so we can reload the constants
    pass