envbot-v5.py

import os
import logging
from dotenv import load_dotenv
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext import Updater, CommandHandler, CallbackContext, CallbackQueryHandler

# Set up basic logging
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)

# Add a StreamHandler to also output logs to the console
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.DEBUG)
logging.getLogger('').addHandler(console_handler)

logger = logging.getLogger(__name__)

# Load .env file
load_dotenv()

# Telegram Bot Token
TELEGRAM_BOT_TOKEN = os.getenv("MY_TELEGRAM_BOT_TOKEN")
# Retrieve the admin ID from the environment variables
ADMIN_ID = os.getenv("ADMIN_ID")

# Check if ADMIN_ID is not None before converting to int
if ADMIN_ID is not None:
    ADMIN_ID = int(ADMIN_ID)
else:
    print("Error: ADMIN_ID is not set in the environment variables.")
    # Handle the error, e.g., exit the script or provide a default value
    exit(1)

# Function to read allowed user IDs and budgets directly from .env file
def read_allowed_users_and_budgets():
    allowed_users = []
    budgets = []
    with open('.env', 'r') as file:
        for line in file:
            if line.startswith('ALLOWED_TELEGRAM_USER_IDS'):
                allowed_users = line.strip().split('=')[1].split(',')
            elif line.startswith('USER_BUDGETS'):
                budgets = line.strip().split('=')[1].split(',')
    return allowed_users, budgets

# Function to update .env file
def update_env_file(allowed_users, budgets):
    env_lines = []
    with open('.env', 'r') as file:
        for line in file:
            if line.startswith('ALLOWED_TELEGRAM_USER_IDS'):
                env_lines.append(f"ALLOWED_TELEGRAM_USER_IDS={','.join(allowed_users)}\n")
            elif line.startswith('USER_BUDGETS'):
                env_lines.append(f"USER_BUDGETS={','.join(budgets)}\n")
            else:
                env_lines.append(line)

    with open('.env', 'w') as file:
        file.writelines(env_lines)

# Function to display the button menu
def show_button_menu(update: Update, context: CallbackContext):
    keyboard = [
        [InlineKeyboardButton("Show Users", callback_data='show_users')],
        [InlineKeyboardButton("Add User", callback_data='add_user')],
        [InlineKeyboardButton("Delete User", callback_data='delete_user')],
        [InlineKeyboardButton("Edit User Budget", callback_data='edit_budget')],
        [InlineKeyboardButton("Start", callback_data='start')]
    ]
    reply_markup = InlineKeyboardMarkup(keyboard)
    update.effective_message.reply_text('Please choose:', reply_markup=reply_markup)

# Start command handler
def start(update: Update, context: CallbackContext):
    show_button_menu(update, context)
    print("Start button clicked")

# Callback query handler
def button(update: Update, context: CallbackContext):
    query = update.callback_query
    query.answer()

    if query.data == 'show_users':
        show_allowed_users(update, context)
        print("Show Users button clicked")
    elif query.data == 'add_user':
        update.effective_message.reply_text('Use /adduser <user_id> <budget> to add a user.')
        print("Add User button clicked")
    elif query.data == 'delete_user':
        update.effective_message.reply_text('Use /deleteuser <user_id> to delete a user.')
        print("Delete User button clicked")
    elif query.data == 'edit_budget':
        update.effective_message.reply_text('Use /editbudget <user_id> <new_budget> to edit a user\'s budget.')
        print("Edit User Budget button clicked")
    elif query.data == 'start':
        start(update, context)

# Show allowed users
def show_allowed_users(update: Update, context: CallbackContext):
    allowed_users, budgets = read_allowed_users_and_budgets()
    message = "Allowed Users and Budgets:\n"
    for user, budget in zip(allowed_users, budgets):
        message += f"User: {user}, Budget: {budget}\n"
    update.effective_message.reply_text(message)
    show_button_menu(update, context)

# Command to add a new user
def add_user(update: Update, context: CallbackContext):
    if len(context.args) < 2:
        update.message.reply_text("Usage: /adduser <user_id> <budget>")
        return

    new_user = context.args[0]
    new_budget = context.args[1]
    allowed_users, budgets = read_allowed_users_and_budgets()

    if new_user in allowed_users:
        update.message.reply_text("User already exists.")
        return

    allowed_users.append(new_user)
    budgets.append(new_budget)
    update_env_file(allowed_users, budgets)
    update.message.reply_text("User added successfully.")
    show_button_menu(update, context)
    print("User added successfully")

# Command to delete a user
def delete_user(update: Update, context: CallbackContext):
    if len(context.args) < 1:
        update.message.reply_text("Usage: /deleteuser <user_id>")
        return

    user_to_delete = context.args[0]
    allowed_users, budgets = read_allowed_users_and_budgets()

    if user_to_delete in allowed_users:
        index = allowed_users.index(user_to_delete)
        allowed_users.pop(index)
        budgets.pop(index)
        update_env_file(allowed_users, budgets)
        update.message.reply_text("User deleted successfully.")
        print("User deleted successfully")
    else:
        update.message.reply_text("User not found.")
        print("User not found")
    show_button_menu(update, context)

# Command to edit a user's budget
def edit_user_budget(update: Update, context: CallbackContext):
    if len(context.args) < 2:
        update.message.reply_text("Usage: /editbudget <user_id> <new_budget>")
        return

    user_to_edit = context.args[0]
    new_budget = context.args[1]
    allowed_users, budgets = read_allowed_users_and_budgets()

    if user_to_edit in allowed_users:
        index = allowed_users.index(user_to_edit)
        budgets[index] = new_budget
        update_env_file(allowed_users, budgets)
        update.message.reply_text("User's budget updated successfully.")
        print("User's budget updated successfully")
    else:
        update.message.reply_text("User not found.")
        print("User not found")
    show_button_menu(update, context)

# Error handler function
def error_handler(update: Update, context: CallbackContext):
    logger.warning('Update "%s" caused error "%s"', update, context.error)

# Main function to run the bot
def main():
    updater = Updater(TELEGRAM_BOT_TOKEN, use_context=True)
    dp = updater.dispatcher

    dp.add_handler(CommandHandler("start", start))
    dp.add_handler(CallbackQueryHandler(button))
    dp.add_handler(CommandHandler("showusers", show_allowed_users))
    dp.add_handler(CommandHandler("adduser", add_user, pass_args=True))
    dp.add_handler(CommandHandler("deleteuser", delete_user, pass_args=True))
    dp.add_handler(CommandHandler("editbudget", edit_user_budget, pass_args=True))
    dp.add_error_handler(error_handler)

    # Send start message to admin
    if ADMIN_ID is not None:
        updater.bot.send_message(chat_id=ADMIN_ID, text="Bot is running.")

    updater.start_polling()
    updater.idle()

if __name__ == '__main__':
    main()