Update config.py

config.py

@@ -1,62 +1,43 @@
-# filename: config.py (Corrected Version)
+# filename: config.py (Corrected Version 2)
 
 import os
 import re
 from dotenv import load_dotenv
 
-# Load environment variables from a .env file for local development
 load_dotenv()
 
-# --- Helper function to cast environment variables (CORRECTED) ---
 def get_env(key, default=None, cast_to=str):
-    """
-    Fetches an environment variable, casts it to a specified type,
-    and correctly handles non-string default values.
-    """
     value = os.getenv(key, default)
-    
-    # If no value is found at all, return None
-    if value is None:
-        return None
-        
-    # --- The Fix ---
-    # If the retrieved value is already the type we want (e.g., default=True was used),
-    # return it directly without trying to process it as a string.
-    if isinstance(value, cast_to):
-        return value
+    if value is None: return None
+    if isinstance(value, cast_to): return value
+    if cast_to == bool: return value.lower() in ('true', '1', 't', 'y', 'yes')
+    try: return cast_to(value)
+    except (ValueError, TypeError): return default
 
-    # If we have a string from the environment, process it.
-    if cast_to == bool:
-        return value.lower() in ('true', '1', 't', 'y', 'yes')
-    
-    try:
-        return cast_to(value)
-    except (ValueError, TypeError):
-        return default
-
-
-# --- Telegram Credentials (Mandatory) ---
+# --- Telegram Credentials ---
 API_ID = get_env("API_ID", cast_to=int)
 API_HASH = get_env("API_HASH")
 BOT_TOKEN = get_env("BOT_TOKEN")
 OWNER_ID = get_env("OWNER_ID", cast_to=int)
-
 if not all([API_ID, API_HASH, BOT_TOKEN, OWNER_ID]):
     raise ValueError("Error: One or more mandatory Telegram credentials are not set.")
 
-# --- Database Configuration (Mandatory) ---
+# --- Database Configuration ---
 mongo_vars = {}
 for key, value in os.environ.items():
     if match := re.match(r"MONGODB_URL_(\d+)", key):
-        if value:
-            mongo_vars[int(match.group(1))] = value
-
+        if value: mongo_vars[int(match.group(1))] = value
 MONGODB_URIS = [mongo_vars[key] for key in sorted(mongo_vars.keys())]
-
 if not MONGODB_URIS:
-    raise ValueError("Error: No MONGODB_URL_<n> environment variables found. Please set at least MONGODB_URL_1.")
+    raise ValueError("Error: No MONGODB_URL_<n> variables found. Set at least MONGODB_URL_1.")
 
-# --- Bot Functionality (Mandatory) ---
+# ** NEW LINE ADDED HERE **
+# Get the specific database name to use within the cluster.
+DATABASE_NAME = get_env("DATABASE_NAME", "terabox_bot_db")
+if not DATABASE_NAME:
+    raise ValueError("Error: DATABASE_NAME environment variable is not set.")
+
+# --- Bot Functionality ---
 TERABOX_WORKER_URL = get_env("TERABOX_WORKER_URL")
 if not TERABOX_WORKER_URL:
     raise ValueError("Error: TERABOX_WORKER_URL is not set.")
@@ -68,10 +49,6 @@ BACKUP_CHANNEL_ID = get_env("BACKUP_CHANNEL_ID", cast_to=int)
 # --- Performance, Limits & Toggles ---
 PREMIUM_WORKERS = get_env("PREMIUM_WORKERS", 1, cast_to=int)
 FREE_WORKERS = get_env("FREE_WORKERS", 1, cast_to=int)
-
 _free_limit_mb = get_env("FREE_USER_FILE_SIZE_LIMIT", 50, cast_to=int)
 FREE_USER_FILE_SIZE_LIMIT_BYTES = _free_limit_mb * 1024 * 1024
-
-# This line now works correctly with the fixed helper function.
 ENABLE_FFMPEG = get_env("ENABLE_FFMPEG", True, cast_to=bool)
-