Move to pydantic settings paradigm

telegram_downloader_bot/main.py

@@ -1,4 +1,3 @@
-import os
 import uvloop
 
 from pyrogram import Client, filters
@@ -7,20 +6,18 @@ from pyrogram.types import Message
 from telegram_downloader_bot.logger import log
 from telegram_downloader_bot.telemetry import init_telemetry
 from telegram_downloader_bot import utils, security
-
-API_ID = os.getenv("API_ID")  # Your API ID from my.telegram.org
-API_HASH = os.getenv("API_HASH")  # Your API Hash from my.telegram.org
-BOT_TOKEN = os.getenv("BOT_TOKEN")  # Your bot token from BotFather
-STORAGE = os.getenv("STORAGE")  # Your bot token from BotFather
+from telegram_downloader_bot.settings import settings
 
 uvloop.install()
-init_telemetry()
+
+if settings.app_env == "PROD":
+    init_telemetry()
 
 app = Client("downloader_bot",
-             api_id=API_ID,
-             api_hash=API_HASH,
-             bot_token=BOT_TOKEN,
-             workers=1)
+             api_id=settings.api_id,
+             api_hash=settings.api_hash,
+             bot_token=settings.bot_token,
+             workers=settings.workers)
 
 
 @app.on_message(filters.command("start"))
@@ -59,7 +56,7 @@ async def message_handler(_, message: Message):
         msg = f"Downloading video {i+1}/{len(urls)}..."
         log.info(msg)
         await message.reply_text(msg)
-        utils.download_tt_video(STORAGE, url)
+        utils.download_tt_video(settings.storage, url)
 
     await message.reply_text("Done.")
 
@@ -70,9 +67,9 @@ async def media_handler(client, message: Message):
 
     await message.reply_text("Downloading media...")
 
-    utils.handle_media_message_contents(STORAGE, client, message)
+    utils.handle_media_message_contents(settings.storage, client, message)
 
 
 if __name__ == "__main__":
-    utils.make_fs(STORAGE)
+    utils.make_fs(settings.storage)
     app.run()