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