__init__.py 1.9 KB

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