68 lines
1.4 KiB
Python
68 lines
1.4 KiB
Python
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()
|