Create bot/init.py

bot/init.py

@@ -0,0 +1,39 @@
+from hydrogram import Client
+from logging import getLogger
+from logging.config import dictConfig
+from .config import Telegram
+
+LOGGER_CONFIG_JSON = {
+    "version": 1,
+    "formatters": {
+        "default": {
+            "format": "[%(asctime)s][%(name)s][%(levelname)s] -> %(message)s",
+            "datefmt": "%d/%m/%Y %H:%M:%S",
+        },
+    },
+    "handlers": {
+        "stream_handler": {"class": "logging.StreamHandler", "formatter": "default"},
+    },
+    "loggers": {
+        "uvicorn": {"level": "INFO", "handlers": ["stream_handler"]},
+        "uvicorn.error": {"level": "WARNING", "handlers": ["stream_handler"]},
+        "bot": {"level": "INFO", "handlers": ["stream_handler"]},
+        "hydrogram": {"level": "INFO", "handlers": ["stream_handler"]},
+    },
+}
+
+dictConfig(LOGGER_CONFIG_JSON)
+logger = getLogger("bot")
+
+if not Telegram.SESSION_STRING:
+    raise RuntimeError("SESSION_STRING missing. Put it in HF Secrets/Variables.")
+
+TelegramBot = Client(
+    name="bot",
+    api_id=Telegram.API_ID,
+    api_hash=Telegram.API_HASH,
+    session_string=Telegram.SESSION_STRING,
+    plugins={"root": "bot/plugins"},
+    sleep_threshold=-1,
+    max_concurrent_transmissions=10,
+)