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] info: InfoCommand linktree: InfoCommand join: FileCommand fest: InfoCommand aktion: InfoCommand 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()