import asyncio from os import getpid, listdir from modules.utils import * from pyrogram.client import Client from pyrogram import filters from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, BotCommand, BotCommandScopeChat, ReplyKeyboardMarkup, ForceReply from pyrogram import idle # type: ignore from pyrogram.errors.exceptions import bad_request_400 pid = getpid() app = Client("holochecker", bot_token=configGet("bot_token", "bot"), api_id=configGet("api_id", "bot"), api_hash=configGet("api_hash", "bot")) @app.on_message(~ filters.scheduled & filters.command(["start"], prefixes=["/"])) async def cmd_start(app, msg): try: user_conf = configGet("stage", file=str(msg.from_user.id)) if user_conf["stage"] != 0: return except FileNotFoundError: jsonSave(jsonLoad(f"data{sep}user_default.json"), f"data{sep}users{sep}{msg.from_user.id}.json") user_conf = configGet("stage", file=str(msg.from_user.id)) await msg.reply_text(locale("start"), reply_markup=ReplyKeyboardMarkup(locale("welcome", "keyboards"))) @app.on_message(~ filters.scheduled & filters.command(["kill", "die", "reboot"], prefixes=["", "/"])) async def cmd_kill(app, msg): if (msg.from_user.id == configGet("owner")) or (msg.from_user.id in configGet("admins")): logWrite(f"Shutting down bot with pid {pid}") await msg.reply_text(f"Вимкнення бота з підом `{pid}`") killProc(pid) @app.on_message(~ filters.scheduled & (filters.regex(locale("welcome", "keyboards")[0][0]) | filters.regex(locale("return", "keyboards")[0][0]))) async def any_message(app, msg): await msg.reply_text(locale("privacy_notice", "message")) await msg.reply_text(locale("question1", "message"), reply_markup=ForceReply(placeholder=locale("question1", "force_reply"))) if __name__ == "__main__": logWrite(f"Starting up with pid {pid}") # Yes, it should be in some kind of async main() function but I don't give a shit. # I did compare performance, almost no difference and it's much more useful this way. Change my mind. app.start() # type: ignore app.send_message(configGet("owner"), f"Starting up with pid `{pid}`") # type: ignore # Registering user commands for fallback locale commands_list = [] for command in configGet("commands"): commands_list.append(BotCommand(command, configGet("commands")[command])) app.set_bot_commands(commands_list) # type: ignore # Registering admin commands commands_admin_list = [] for command in configGet("commands"): commands_admin_list.append(BotCommand(command, configGet("commands")[command])) for command in configGet("commands_admin"): commands_admin_list.append(BotCommand(command, configGet("commands_admin")[command])) for admin in configGet("admins"): try: app.set_bot_commands(commands_admin_list, scope=BotCommandScopeChat(chat_id=admin)) # type: ignore except bad_request_400.PeerIdInvalid: pass app.set_bot_commands(commands_admin_list, scope=BotCommandScopeChat(chat_id=configGet("owner"))) # type: ignore idle() app.send_message(configGet("owner"), f"Shutting with pid `{pid}`") # type: ignore app.stop() # type: ignore killProc(pid)