File size: 2,460 Bytes
93fe4f5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 | import json
import os
import logging
from dotenv import load_dotenv
# Load environment variables
load_dotenv()
BOT_OWNER_ID = int(os.getenv("BOT_OWNER_ID", "7014665654"))
# Setup log directory and logging
LOG_DIR = os.path.abspath("logs")
os.makedirs(LOG_DIR, exist_ok=True) # Ensure logs directory exists
LOG_FILE = os.path.join(LOG_DIR, "auth.log")
file_handler = logging.FileHandler(LOG_FILE)
file_handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
logger = logging.getLogger("terabox_bot")
logger.setLevel(logging.INFO)
logger.addHandler(file_handler)
# Path to authorized users file
AUTHORIZED_USERS_FILE = os.path.abspath("authorized_users.json")
logger.info(f"Authorized users file path: {AUTHORIZED_USERS_FILE}")
# Load authorized users
if os.path.exists(AUTHORIZED_USERS_FILE):
try:
with open(AUTHORIZED_USERS_FILE, "r") as f:
AUTHORIZED_USERS = json.load(f)
logger.info(f"Loaded authorized users: {AUTHORIZED_USERS}")
except Exception as e:
AUTHORIZED_USERS = []
logger.warning(f"Failed to load users: {str(e)}. Initializing empty list.")
else:
AUTHORIZED_USERS = []
logger.info("No user file found, initializing empty list")
# Add user
def add_authorized_user(user_id: int) -> bool:
user_id = int(user_id)
if user_id in AUTHORIZED_USERS:
logger.info(f"User {user_id} already authorized")
return False
AUTHORIZED_USERS.append(user_id)
return save_users()
# Remove user
def remove_authorized_user(user_id: int) -> bool:
if user_id not in AUTHORIZED_USERS:
logger.info(f"User {user_id} not found in list")
return False
AUTHORIZED_USERS.remove(user_id)
return save_users()
# Save user list to file
def save_users() -> bool:
try:
with open(AUTHORIZED_USERS_FILE, "w") as f:
json.dump(AUTHORIZED_USERS, f)
logger.info("✅ User list saved successfully")
logger.info(f"✅ Saved to path: {AUTHORIZED_USERS_FILE}")
return True
except Exception as e:
logger.error(f"❌ Error saving user list: {str(e)}")
return False
# Check if user is authorized
def is_authorized(user_id: int) -> bool:
return user_id == BOT_OWNER_ID or user_id in AUTHORIZED_USERS
# Get all authorized users
def get_authorized_users() -> list:
logger.info(f"Returning authorized users: {AUTHORIZED_USERS}")
return AUTHORIZED_USERS
|