Spaces:
Sleeping
Sleeping
| from platform import python_version | |
| from threading import RLock | |
| from time import gmtime, strftime | |
| from time import time as t | |
| from pyrogram import Client, __version__ | |
| from pyrogram.raw.all import layer | |
| from pyrogram.types import BotCommand | |
| from Powers import (API_HASH, API_ID, BOT_TOKEN, LOG_DATETIME, LOGFILE, LOGGER, | |
| MESSAGE_DUMP, NO_LOAD, UPTIME, WORKERS, load_cmds, | |
| scheduler) | |
| from Powers.database import MongoDB | |
| from Powers.plugins import all_plugins | |
| from Powers.plugins.scheduled_jobs import * | |
| from Powers.supports import * | |
| from Powers.vars import Config | |
| INITIAL_LOCK = RLock() | |
| class Gojo(Client): | |
| """Starts the Pyrogram Client on the Bot Token when we do 'python3 -m Powers'""" | |
| def __init__(self): | |
| # name = Powers | |
| super().__init__( | |
| "Gojo_Satoru", | |
| bot_token=BOT_TOKEN, | |
| plugins=dict(root="Powers.plugins", exclude=NO_LOAD), | |
| api_id=API_ID, | |
| api_hash=API_HASH, | |
| workers=WORKERS, | |
| ) | |
| async def start(self, use_qr=False, except_ids=[]): | |
| """Start the bot.""" | |
| await super().start(use_qr=use_qr, except_ids=except_ids) | |
| await self.set_bot_commands( | |
| [ | |
| BotCommand( | |
| "start", "To check weather the bot is alive or not"), | |
| BotCommand("help", "To get help menu"), | |
| BotCommand("donate", "To buy me a coffee"), | |
| BotCommand("bug", "To report bugs") | |
| ] | |
| ) | |
| meh = await self.get_me() # Get bot info from pyrogram client | |
| LOGGER.info("Starting bot...") | |
| Config.BOT_ID = meh.id | |
| Config.BOT_NAME = meh.first_name | |
| Config.BOT_USERNAME = meh.username | |
| startmsg = await self.send_message(MESSAGE_DUMP, "<i>Starting Bot...</i>") | |
| # Show in Log that bot has started | |
| LOGGER.info( | |
| f"Pyrogram v{__version__} (Layer - {layer}) started on {meh.username}", | |
| ) | |
| LOGGER.info(f"Python Version: {python_version()}\n") | |
| # Get cmds and keys | |
| cmd_list = await load_cmds(await all_plugins()) | |
| await load_support_users() | |
| await cache_support() | |
| LOGGER.info(f"Dev Users: {SUPPORT_USERS['Dev']}") | |
| LOGGER.info(f"Sudo Users: {SUPPORT_USERS['Sudo']}") | |
| LOGGER.info(f"Whitelist users: {SUPPORT_USERS['White']}") | |
| LOGGER.info(f"Plugins Loaded: {cmd_list}") | |
| if BDB_URI: | |
| scheduler.add_job(send_wishish, 'cron', [ | |
| self], hour=0, minute=0, second=0) | |
| scheduler.start() | |
| # Send a message to MESSAGE_DUMP telling that the | |
| # bot has started and has loaded all plugins! | |
| await startmsg.edit_text( | |
| ( | |
| f"<b><i>@{meh.username} started on Pyrogram v{__version__} (Layer - {layer})</i></b>\n" | |
| f"\n<b>Python:</b> <u>{python_version()}</u>\n" | |
| "\n<b>Loaded Plugins:</b>\n" | |
| f"<i>{cmd_list}</i>\n" | |
| ), | |
| ) | |
| LOGGER.info("Bot Started Successfully!\n") | |
| async def stop(self): | |
| """Stop the bot and send a message to MESSAGE_DUMP telling that the bot has stopped.""" | |
| runtime = strftime("%Hh %Mm %Ss", gmtime(t() - UPTIME)) | |
| LOGGER.info("Uploading logs before stopping...!\n") | |
| # Send Logs to MESSAGE_DUMP and LOG_CHANNEL | |
| scheduler.remove_all_jobs() | |
| target = MESSAGE_DUMP or OWNER_ID | |
| await self.send_document( | |
| target, | |
| document=LOGFILE, | |
| caption=( | |
| "Bot Stopped!\n\n" f"Uptime: {runtime}\n" f"<code>{LOG_DATETIME}</code>" | |
| ), | |
| ) | |
| await super().stop() | |
| MongoDB.close() | |
| LOGGER.info( | |
| f"""Bot Stopped. | |
| Logs have been uploaded to the MESSAGE_DUMP Group! | |
| Runtime: {runtime}s\n | |
| """, | |
| ) | |