lenaverse-bot/lenaverse_bot/core/config.py

import os
import tomllib
from typing import Annotated, Self

import discord
from pydantic import BaseModel, StringConstraints

StrippedStr = Annotated[str, StringConstraints(strip_whitespace=True)]


class Post(BaseModel):
    # users authorized to posts
    users: list[int]
    # where to put posts
    channel: int

    def get_channel(
        self,
        client: discord.Client,
    ) -> discord.Thread | discord.TextChannel:
        """
        Zielkanal für Posts finden
        """

        channel = client.get_channel(self.channel)
        assert isinstance(channel, discord.Thread | discord.TextChannel)

        return channel


class InfoCommand(BaseModel):
    description: StrippedStr = "..."
    content: StrippedStr = ""


class FileCommand(InfoCommand):
    filename: str = ""


class ClubInfo(BaseModel):
    channels: list[int]

    linktree: InfoCommand = InfoCommand()
    join: FileCommand = FileCommand(filename="Aufnahmeantrag.pdf")


class Config(BaseModel):
    discord_token: str
    command_prefix: str

    post: Post
    ev_info: ClubInfo

    @classmethod
    def get(cls) -> Self:
        cfg_path = os.getenv(
            key="CONFIG_PATH",
            default="/usr/local/etc/lenaverse-bot/lenaverse-bot.toml",
        )

        with open(cfg_path, "rb") as cfg_file:
            return cls.model_validate(tomllib.load(cfg_file))


CONFIG = Config.get()