__init__.py 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. from config import (
  2. BOT_TOKEN, API_ID, API_HASH, SUDO_USERS_ID, PHONE_NUMBER,
  3. LOG_GROUP_ID, FERNET_ENCRYPTION_KEY, MONGO_DB_URI,
  4. WELCOME_DELAY_KICK_SEC, ARQ_API_BASE_URL as ARQ_API,
  5. MAIN_CHATS, GBAN_LOG_GROUP_ID
  6. )
  7. from pyrogram import Client
  8. from pyromod import listen
  9. from motor.motor_asyncio import AsyncIOMotorClient as MongoClient
  10. from Python_ARQ import ARQ
  11. import time
  12. SUDOERS = SUDO_USERS_ID
  13. MAIN_CHATS = MAIN_CHATS
  14. GBAN_LOG_GROUP_ID = GBAN_LOG_GROUP_ID
  15. FERNET_ENCRYPTION_KEY = FERNET_ENCRYPTION_KEY
  16. WELCOME_DELAY_KICK_SEC = WELCOME_DELAY_KICK_SEC
  17. LOG_GROUP_ID = LOG_GROUP_ID
  18. MOD_LOAD = []
  19. MOD_NOLOAD = []
  20. bot_start_time = time.time()
  21. mongo_client = MongoClient(MONGO_DB_URI)
  22. db = mongo_client.wbb
  23. app2 = Client(
  24. "userbot",
  25. phone_number=PHONE_NUMBER,
  26. api_id=API_ID,
  27. api_hash=API_HASH
  28. )
  29. app = Client(
  30. "wbb",
  31. bot_token=BOT_TOKEN,
  32. api_id=API_ID,
  33. api_hash=API_HASH
  34. )
  35. BOT_ID = 0
  36. BOT_NAME = ""
  37. BOT_USERNAME = ""
  38. BOT_MENTION = ""
  39. BOT_DC_ID = 0
  40. USERBOT_ID = 0
  41. USERBOT_NAME = ""
  42. USERBOT_USERNAME = ""
  43. USERBOT_DC_ID = 0
  44. USERBOT_MENTION = ""
  45. def get_info(app, app2):
  46. global BOT_ID, BOT_NAME, BOT_USERNAME, BOT_DC_ID, BOT_MENTION
  47. global USERBOT_ID, USERBOT_NAME, USERBOT_USERNAME, USERBOT_DC_ID, USERBOT_MENTION
  48. getme = app.get_me()
  49. getme2 = app2.get_me()
  50. BOT_ID = getme.id
  51. USERBOT_ID = getme2.id
  52. if getme.last_name:
  53. BOT_NAME = getme.first_name + " " + getme.last_name
  54. else:
  55. BOT_NAME = getme.first_name
  56. BOT_USERNAME = getme.username
  57. BOT_MENTION = getme.mention
  58. BOT_DC_ID = getme.dc_id
  59. if getme2.last_name:
  60. USERBOT_NAME = getme2.first_name + " " + getme2.last_name
  61. else:
  62. USERBOT_NAME = getme2.first_name
  63. USERBOT_USERNAME = getme2.username
  64. USERBOT_MENTION = getme2.mention
  65. USERBOT_DC_ID = getme2.dc_id
  66. app.start()
  67. app2.start()
  68. get_info(app, app2)
  69. SUDOERS.append(USERBOT_ID)
  70. # ARQ client
  71. arq = ARQ(ARQ_API)