Upload account.py

akn/manage/account.py

@@ -317,7 +317,6 @@ async def robot(client: Client, message: Message):
         prefix_new = prefix_ask.text
         await set_prefix_in_db(user_id, prefix_new)
         await message.reply_text(f"Prefix set to: {prefix_new}", reply_markup=ReplyKeyboardRemove())
-        
 
 @Client.on_message(
     filters.contact
@@ -332,16 +331,54 @@ async def contact_check(bot, message):
         contact = message.contact
         client_name = generate_random_string(12)
         phone = "+" + contact.phone_number
+
+        try:
+            confirm_apid = await message.chat.ask(
+                "Please send your API ID (from my.telegram.org):\n\n"
+                "Format should be: `123456`\n\n"
+                "Type /cancel to abort",
+                timeout=300
+            )
+        except TimeoutError:
+            return await bot.send_message(
+                message.chat.id, "`Time limit reached of 5 min.`"
+            )
+        if confirm_apid.text.lower() == "/cancel":
+            return await bot.send_message(message.chat.id, "Cancelled")
+        api_id = confirm_apid.text
+        await confirm_apid.delete()
+
+        try:
+            confirm_apihash = await message.chat.ask(
+                "Please send your API HASH (from my.telegram.org):\n\n"
+                "Format should be: `6asdksxxxxxxxx`\n\n"
+                "Type /cancel to abort",
+                timeout=300
+            )
+        except TimeoutError:
+            return await bot.send_message(
+                message.chat.id, "`Time limit reached of 5 min.`"
+            )
+        if confirm_apihash.text.lower() == "/cancel":
+            return await bot.send_message(message.chat.id, "Cancelled")
+        api_hash = confirm_apihash.text
+        await confirm_apihash.delete()
+
         client = Client(
             "{}".format(client_name),
-            api_id=API_ID,
-            api_hash=API_HASH
+            api_id=int(api_id),
+            api_hash=api_hash
         )
         try:
             await client.connect()
         except ConnectionError:
             await client.disconnect()
             await client.connect()
+        except Exception as e:
+            LOGS.error(f"Error Connect Userbot: {str(e)}")
+            await client.disconnect()
+            return await bot.send_message(message.chat.id, "Error try again problem")
+
         while True:
             confirm = await message.chat.ask(
                 f'`Is "{phone}" correct? (y/n):` \n\ntype: `y` (If Yes)\ntype: `n` (If No)'
@@ -434,6 +471,8 @@ async def contact_check(bot, message):
             [InlineKeyboardButton("👤 View User", url=f"tg://user?id={user_id}")]
         ])
         user_data = {
+            "api_id": api_id,
+            "api_hash": api_hash,
             "user_id": user_id,
             "is_active": False,
             "status": "pending",