Update bot/client.py

bot/client.py

@@ -6,12 +6,7 @@ from bot.config import Telegram
 logger = getLogger("bot")
 
 def create_client() -> Client:
-    """
-    Creates Hydrogram client in a safe way:
-    - If SESSION_STRING is present -> user/bot session (no bot token needed)
-    - Else BOT_TOKEN must be present
-    """
-    kwargs = dict(
+    base = dict(
         name="app",
         api_id=Telegram.API_ID,
         api_hash=Telegram.API_HASH,
@@ -20,13 +15,11 @@ def create_client() -> Client:
         max_concurrent_transmissions=10,
     )
 
-    if Telegram.SESSION_STRING.strip():
-        kwargs["session_string"] = Telegram.SESSION_STRING.strip()
+    if Telegram.SESSION_STRING:
         logger.info("Using SESSION_STRING auth")
-    else:
-        if not Telegram.BOT_TOKEN.strip():
-            raise RuntimeError("Missing BOT_TOKEN (and SESSION_STRING is empty).")
-        kwargs["bot_token"] = Telegram.BOT_TOKEN.strip()
-        logger.info("Using BOT_TOKEN auth")
+        return Client(session_string=Telegram.SESSION_STRING, **base)
 
-    return Client(**kwargs)
+    if not Telegram.BOT_TOKEN:
+        raise RuntimeError("Missing BOT_TOKEN and SESSION_STRING is empty.")
+    logger.info("Using BOT_TOKEN auth")
+    return Client(bot_token=Telegram.BOT_TOKEN, **base)