config.py - develop - dmsav/ydvs

from dataclasses import dataclass

from environs import Env


@dataclass
class TgBot:
    token: str      # Токен для доступа к телеграм-боту
    bot_name: str   # Название бота
    img_path: str   # путь для сохранения файла статистики
    img_www_path: str   # путь для загрузки файлов в сообщение

@dataclass
class Config:
    tg_bot: TgBot
    botenv: str      # Код окружения DEV, TEST, PROD

# Создаем функцию, которая будет читать файл .env и возвращать экземпляр класса Config с заполненными полями token и admin_ids
def load_config(path: str | None = None) -> Config:
    env = Env()
    env.read_env(path)
    return Config(tg_bot=TgBot(token=env('BOT_TOKEN'),
                               img_path=env('IMG_PATH'),
                               img_www_path=env('IMG_WWW_PATH'),
                               bot_name=env('BOT_NAME')),
                    botenv=env('BOT_ENV'))

if __name__ == '__main__':
    config = load_config()
    print(config)