PamGram/config.py

import os
from typing import Any

import ujson
from dotenv import load_dotenv

from utils.storage import Storage

# take environment variables from .env.
load_dotenv()

env = os.getenv


def str_to_bool(value: Any) -> bool:
    """Return whether the provided string (or any value really) represents true. Otherwise false.
    Just like plugin server stringToBoolean.
    """
    if not value:
        return False
    return str(value).lower() in ("y", "yes", "t", "true", "on", "1")


_config = {
    "debug": str_to_bool(os.getenv('DEBUG', 'False')),

    "mysql": {
        "host": env("DB_HOST", "127.0.0.1"),
        "port": int(env("DB_PORT", "3306")),
        "user": env("DB_USERNAME"),
        "password": env("DB_PASSWORD"),
        "database": env("DB_DATABASE"),
    },

    "redis": {
        "host": env("REDIS_HOST", "127.0.0.1"),
        "port": int(env("REDIS_PORT", "6369")),
        "database": int(env("REDIS_DB", "0")),
    },

    # 联系 https://t.me/BotFather 使用 /newbot 命令创建机器人并获取 token
    "bot_token": env("BOT_TOKEN"),

    # 记录错误并发送消息通知开发人员
    "error_notification_chat_id": env("ERROR_NOTIFICATION_CHAT_ID"),

    # 文章推送群组
    "channels": [
        # {"name": "", "chat_id": 1},
        # 在环境变量里的格式是 json: [{"name": "", "chat_id": 1}]
        *ujson.loads(env('CHANNELS', '[]'))
    ],

    # bot 管理员
    "admins": [
        # {"username": "", "user_id": 123},
        # 在环境变量里的格式是 json: [{"username": "", "user_id": 1}]
        *ujson.loads(env('ADMINS', '[]'))
    ],

    "joining_verification_groups": ujson.loads(env('JOINING_VERIFICATION_GROUPS', '[]')),
}

config = Storage(_config)