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