__init__.py 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  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 pytgcalls import GroupCall
  9. from motor.motor_asyncio import AsyncIOMotorClient as MongoClient
  10. from Python_ARQ import ARQ
  11. import time
  12. import logging
  13. f = open("error.log", "w")
  14. f.write("PEAK OF LOG FILE")
  15. LOG_FORMAT = (
  16. '''
  17. [%(asctime)s.%(msecs)03d] %(filename)s:%(lineno)s
  18. %(levelname)s: %(message)s''')
  19. logging.basicConfig(
  20. level=logging.INFO,
  21. format=LOG_FORMAT,
  22. datefmt="%m-%d %H:%M",
  23. filename="error.log",
  24. filemode="w",
  25. )
  26. console = logging.StreamHandler()
  27. console.setLevel(logging.ERROR)
  28. formatter = logging.Formatter(LOG_FORMAT)
  29. console.setFormatter(formatter)
  30. logging.getLogger("").addHandler(console)
  31. log = logging.getLogger()
  32. SUDOERS = SUDO_USERS_ID
  33. def init_sudo():
  34. global SUDOERS
  35. from wbb.utils.botinfo import USERBOT_ID
  36. SUDOERS.append(USERBOT_ID)
  37. FERNET_ENCRYPTION_KEY = FERNET_ENCRYPTION_KEY
  38. WELCOME_DELAY_KICK_SEC = WELCOME_DELAY_KICK_SEC
  39. LOG_GROUP_ID = LOG_GROUP_ID
  40. MOD_LOAD = []
  41. MOD_NOLOAD = []
  42. bot_start_time = time.time()
  43. mongo_client = MongoClient(MONGO_DB_URI)
  44. db = mongo_client.wbb
  45. # Userbot client
  46. print("\nStarting Helper Userbot")
  47. app2 = Client(
  48. "userbot",
  49. phone_number=PHONE_NUMBER,
  50. api_id=API_ID,
  51. api_hash=API_HASH
  52. )
  53. # Mainbot client
  54. print("\nStarting Main Bot\n")
  55. app = Client(
  56. "wbb",
  57. bot_token=BOT_TOKEN,
  58. api_id=API_ID,
  59. api_hash=API_HASH
  60. )
  61. # Pytgcalls client
  62. vc = GroupCall(
  63. app2,
  64. input_filename="input.raw",
  65. play_on_repeat=False,
  66. enable_logs_to_console=False
  67. )
  68. # ARQ client
  69. arq = ARQ(ARQ_API)