diff --git a/.env.example b/.env.example index 32551cb..d8d5d99 100644 --- a/.env.example +++ b/.env.example @@ -4,6 +4,12 @@ API_ID=2040 API_HASH=b18441a1ff607e10a989891a5462e627 +# Токен для бота уведомлений, создать новый бот: @BotFather (https://t.me/BotFather) +BOT_TOKEN= +ADMIN_ID= +NOTIFY_ADMIN=true + +# Токен для Lolz, требуются 3 права: basic, read, post (другие не имеет смысла выдавать) LOLZ_TOKEN= LOLZ_THREAD_ID= diff --git a/config.py b/config.py index 0b5d5db..27ee8b1 100644 --- a/config.py +++ b/config.py @@ -3,6 +3,7 @@ import logging from typing import Any from dotenv import load_dotenv +from curses.ascii import isdigit logger = logging.getLogger(__name__) @@ -41,6 +42,18 @@ class Config: key="API_HASH", default="b18441a1ff607e10a989891a5462e627" ) + # Notification bot (aiogram) + bot_token = os.getenv(key="BOT_TOKEN", default=None) + admin_id = ( + os.getenv(key="ADMIN_ID", default=None) + if not os.getenv(key="ADMIN_ID", default=None) + or not isdigit(os.getenv(key="ADMIN_ID", default=None)) + else int(os.getenv(key="ADMIN_ID", default=None)) + ) + notify_admin = ( + os.getenv(key="NOTIFY_ADMIN", default="true") + ).lower() == "true" + # Lolz lolz_token = os.getenv(key="LOLZ_TOKEN", default=None) lolz_thread_id = int(os.getenv(key="LOLZ_THREAD_ID", default=None)) @@ -71,6 +84,10 @@ class Config: # Pyrogram "api_id": api_id, "api_hash": api_hash, + # Notification bot (aiogram) + "bot_token": bot_token, + "admin_id": admin_id, + "notify_admin": notify_admin, # Lolz "lolz_token": lolz_token, "lolz_thread_id": lolz_thread_id,