Upload bots_manage.py

akn/manage/bots_manage.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 import time
+import os
 import uuid
 import asyncio
 import pyromod
@@ -74,13 +75,58 @@ async def userbotdashboardub(client: Client, callback: CallbackQuery):
     )
     await callback.answer()
 
+@Client.on_callback_query(filters.regex("^session_exportuball$"))
+async def export_all_sessionsub(client, callback):
+    user_data = await db_client.session.find_one({"user_id": callback.from_user.id})
+    if not user_data or not user_data.get("user_client"):
+        return await callback.answer("⚠️ No sessions to export!", show_alert=True)
+
+    sessions = user_data["user_client"]
+    export_text = ""
+    for index, session in enumerate(sessions, 1):
+        status = session.get("status", "unknown")
+        is_active = session.get("is_active", False)
+        session_str = session.get("session_string", "None")
+        export_text += (
+            f"Userbot #{index}\n"
+            f"Status: {status}\n"
+            f"Active: {is_active}\n"
+            f"Session: {session_str}\n\n"
+        )
+    filename = f"sessions_{callback.from_user.id}.txt"
+    with open(filename, "w") as f:
+        f.write(export_text)
+
+    try:
+        await callback.message.edit_text("Downloading export all session.....", reply_markup=None)
+        await callback.message.edit_media(
+            media=InputMediaDocument(
+                media=filename,
+                caption="📝 Here are your session details."
+            ),
+            reply_markup=InlineKeyboardMarkup([
+                [InlineKeyboardButton("🔙 Back", callback_data="my_userbotub")]
+            ])
+        )
+        os.remove(filename) 
+        await callback.answer()
+    except Exception as e:
+        LOGS.error(f"Error in export_all_sessionsub handler: {str(e)}")
+        await callback.answer(f"❌ Error: Please try again later.", True)
+        try:
+            os.remove(filename)
+        except:
+            pass
+
 @Client.on_callback_query(filters.regex("^ubsetting$"))
 async def userbotsettingsub(client: Client, callback: CallbackQuery):
     user_id = callback.from_user.id
     
     buttons = [
-        [InlineKeyboardButton("📝 Session Info", callback_data="session_infoub")],
-        [InlineKeyboardButton("🗑 Delete Userbot", callback_data="delete_userbotub")],
+        [InlineKeyboardButton("📝 Session Info", callback_data="session_infoub"),
+         InlineKeyboardButton("🗑 Delete Userbot", callback_data="delete_userbotub")
+        ],
+        [InlineKeyboardButton("📥 Export All Sessions", callback_data="session_exportuball")],
         [InlineKeyboardButton("🔙 Back", callback_data="my_userbotub")]
     ]
 
@@ -91,6 +137,62 @@ async def userbotsettingsub(client: Client, callback: CallbackQuery):
     )
     await callback.answer()
 
+@Client.on_callback_query(filters.regex("^delete_userbotub$"))
+async def sessionconfirmdeleteub(client, callback):
+    await callback.message.edit_text(
+        "⚠️ This will permanently delete your userbot!",
+        reply_markup=InlineKeyboardMarkup([
+            [InlineKeyboardButton("✅ Confirm", "confirm_rmdeleteub")],
+            [InlineKeyboardButton("❌ Cancel", "my_userbotub")]
+        ])
+    )
+
+@Client.on_callback_query(filters.regex("^confirm_rmdeleteub$"))
+async def sessionrmdeleteubnow(client, callback):
+    user_mention = callback.from_user.mention
+    user_data = await db_client.session.find_one({"user_id": callback.from_user.id})
+    if not user_data or not user_data.get("user_client"):
+        return await callback.answer(
+            "❌ You don't have any Userbot deployed yet!",
+            show_alert=True
+        )
+    try:
+        result = await db_client.session.update_one(
+            {
+                "_id": user_data["_id"],
+                "user_id": callback.from_user.id,
+                "user_client.user_id": callback.from_user.id
+            },
+            {"$pull": {"user_client": {"user_id": callback.from_user.id}}}
+        )
+        if result.modified_count == 1:
+            await callback.message.edit_text(
+                f"✅ User `{callback.from_user.id}` has been permanently removed",
+                reply_markup=InlineKeyboardMarkup([
+                    [InlineKeyboardButton("⭐ My Userbot", callback_data="my_userbotub")]
+                ])
+            )
+            await notify_admins_user(
+                client=client,
+                message=f"🛑 <b>Deleted Userbot</b>\n\n"
+                        f"👤 User: {user_mention} ({callback.from_user.id})\n",
+                user_id=callback.from_user.id,
+                is_button=False
+            )
+        else:
+            raise Exception("Database update failed")
+    except Exception as e:
+        LOGS.error(f"Error in sessionrmdeleteubnow handler: {str(e)}")
+        await callback.answer(f"❌ Error: Please try again later.", True)
+        await notify_admins_user(
+            client=client,
+            message=f"🚨 <b>Error in sessionrmdeleteubnow handler</b>\n\n"
+                    f"👤 User: {user_mention} ({callback.from_user.id})\n"
+                    f"🧠 Error: <code>{str(e)}</code>",
+            user_id=callback.from_user.id,
+            is_button=False
+        )
+
 @Client.on_callback_query(filters.regex("^session_infoub$"))
 async def infosessionstringub(client: Client, callback: CallbackQuery):
     user_data = await db_client.session.find_one({"user_id": callback.from_user.id})
@@ -182,7 +284,7 @@ async def asksessionstringub(client: Client, callback: CallbackQuery):
             message=f"🚨 <b>Error in asksessionstringub handler</b>\n\n"
                     f"👤 User: {user_mention} ({callback.from_user.id})\n"
                     f"🧠 Error: <code>{str(e)}</code>",
-            user_id=user_id,
+            user_id=callback.from_user.id,
             is_button=False
         )