mirror of
https://github.com/TeamPGM/PagerMaid-Pyro.git
synced 2024-11-21 23:08:02 +00:00
🔖 Update to v1.4.11
feat: support sub command fix: reload plugin
This commit is contained in:
parent
7cb481d0fc
commit
19b102486d
@ -22,8 +22,8 @@ from pagermaid.scheduler import scheduler
|
|||||||
import pyromod.listen
|
import pyromod.listen
|
||||||
from pyrogram import Client
|
from pyrogram import Client
|
||||||
|
|
||||||
pgm_version = "1.4.10"
|
pgm_version = "1.4.11"
|
||||||
pgm_version_code = 1410
|
pgm_version_code = 1411
|
||||||
CMD_LIST = {}
|
CMD_LIST = {}
|
||||||
module_dir = __path__[0]
|
module_dir = __path__[0]
|
||||||
working_dir = getcwd()
|
working_dir = getcwd()
|
||||||
|
@ -6,10 +6,10 @@ from typing import Optional, List, Tuple, Dict
|
|||||||
|
|
||||||
from pydantic import BaseModel, ValidationError
|
from pydantic import BaseModel, ValidationError
|
||||||
|
|
||||||
|
import pagermaid.modules
|
||||||
from pagermaid import Config, logs
|
from pagermaid import Config, logs
|
||||||
from pagermaid.enums import Message
|
from pagermaid.enums import Message
|
||||||
from pagermaid.common.cache import cache
|
from pagermaid.common.cache import cache
|
||||||
from pagermaid.modules import plugin_list as active_plugins
|
|
||||||
from pagermaid.utils import client
|
from pagermaid.utils import client
|
||||||
from pagermaid.services import sqlite
|
from pagermaid.services import sqlite
|
||||||
|
|
||||||
@ -33,6 +33,7 @@ class LocalPlugin(BaseModel):
|
|||||||
@property
|
@property
|
||||||
def load_status(self) -> bool:
|
def load_status(self) -> bool:
|
||||||
"""插件加载状态"""
|
"""插件加载状态"""
|
||||||
|
active_plugins = pagermaid.modules.plugin_list
|
||||||
return self.name in active_plugins
|
return self.name in active_plugins
|
||||||
|
|
||||||
def remove(self):
|
def remove(self):
|
||||||
@ -307,6 +308,7 @@ class PluginManager:
|
|||||||
) -> Tuple[List[str], List[LocalPlugin], List[LocalPlugin]]:
|
) -> Tuple[List[str], List[LocalPlugin], List[LocalPlugin]]:
|
||||||
"""Get plugins status"""
|
"""Get plugins status"""
|
||||||
all_local_plugins = self.plugins
|
all_local_plugins = self.plugins
|
||||||
|
active_plugins = pagermaid.modules.plugin_list
|
||||||
disabled_plugins = []
|
disabled_plugins = []
|
||||||
inactive_plugins = []
|
inactive_plugins = []
|
||||||
for plugin in all_local_plugins:
|
for plugin in all_local_plugins:
|
||||||
|
52
pagermaid/enums/command.py
Normal file
52
pagermaid/enums/command.py
Normal file
@ -0,0 +1,52 @@
|
|||||||
|
from typing import NewType, Callable, Any, Awaitable, Union, TYPE_CHECKING, Optional
|
||||||
|
|
||||||
|
from ..inject import inject
|
||||||
|
|
||||||
|
if TYPE_CHECKING:
|
||||||
|
from . import Client, Message
|
||||||
|
|
||||||
|
CommandHandlerFunc = NewType("CommandHandlerFunc", Callable[[Any, Any], Awaitable[Any]])
|
||||||
|
CommandHandlerDecorator = NewType(
|
||||||
|
"CommandHandlerDecorator",
|
||||||
|
Callable[[Union["CommandHandler", CommandHandlerFunc]], "CommandHandler"],
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class CommandHandler:
|
||||||
|
def __init__(self, func: CommandHandlerFunc, command: Optional[str]) -> None:
|
||||||
|
self._pgp_func__: CommandHandlerFunc = func
|
||||||
|
self._pgp_command__: Optional[str] = command
|
||||||
|
self._pgp_raw_handler = None
|
||||||
|
|
||||||
|
def func(self) -> CommandHandlerFunc:
|
||||||
|
return self._pgp_func__
|
||||||
|
|
||||||
|
def set_handler(self, handler):
|
||||||
|
self._pgp_raw_handler = handler
|
||||||
|
|
||||||
|
def get_handler(self):
|
||||||
|
return self._pgp_raw_handler
|
||||||
|
|
||||||
|
async def handler(self, client: "Client", message: "Message"):
|
||||||
|
func = self.func()
|
||||||
|
if data := inject(message, func):
|
||||||
|
await func(**data)
|
||||||
|
else:
|
||||||
|
if func.__code__.co_argcount == 0:
|
||||||
|
await func()
|
||||||
|
if func.__code__.co_argcount == 1:
|
||||||
|
await func(message)
|
||||||
|
elif func.__code__.co_argcount == 2:
|
||||||
|
await func(client, message)
|
||||||
|
|
||||||
|
def sub_command(self, **kwargs) -> CommandHandlerDecorator:
|
||||||
|
if self._pgp_command__ is None:
|
||||||
|
raise ValueError("Cannot add subcommand to a handler without a command")
|
||||||
|
if self._pgp_raw_handler is None:
|
||||||
|
raise ValueError("Cannot add subcommand to a handler without init")
|
||||||
|
from pagermaid.listener import listener
|
||||||
|
|
||||||
|
def decorator(func: CommandHandlerFunc) -> CommandHandlerFunc:
|
||||||
|
return listener(__parent_command=self._pgp_command__, **kwargs)(func)
|
||||||
|
|
||||||
|
return decorator
|
@ -103,12 +103,14 @@ class Hook:
|
|||||||
logs.info(f"[shutdown]: {type(exception)}: {exception}")
|
logs.info(f"[shutdown]: {type(exception)}: {exception}")
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def command_pre(message: Message, command):
|
async def command_pre(message: Message, command, sub_command):
|
||||||
cors = []
|
cors = []
|
||||||
try:
|
try:
|
||||||
for pre in hook_functions["command_pre"]:
|
for pre in hook_functions["command_pre"]:
|
||||||
try:
|
try:
|
||||||
data = inject(message, pre, command=command)
|
data = inject(
|
||||||
|
message, pre, command=command, sub_command=sub_command
|
||||||
|
)
|
||||||
except Exception as exception:
|
except Exception as exception:
|
||||||
logs.info(f"[process_error]: {type(exception)}: {exception}")
|
logs.info(f"[process_error]: {type(exception)}: {exception}")
|
||||||
continue
|
continue
|
||||||
@ -124,12 +126,14 @@ class Hook:
|
|||||||
logs.info(f"[command_pre]: {type(exception)}: {exception}")
|
logs.info(f"[command_pre]: {type(exception)}: {exception}")
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def command_post(message: Message, command):
|
async def command_post(message: Message, command, sub_command):
|
||||||
cors = []
|
cors = []
|
||||||
try:
|
try:
|
||||||
for post in hook_functions["command_post"]:
|
for post in hook_functions["command_post"]:
|
||||||
try:
|
try:
|
||||||
data = inject(message, post, command=command)
|
data = inject(
|
||||||
|
message, post, command=command, sub_command=sub_command
|
||||||
|
)
|
||||||
except Exception as exception:
|
except Exception as exception:
|
||||||
logs.info(f"[process_error]: {type(exception)}: {exception}")
|
logs.info(f"[process_error]: {type(exception)}: {exception}")
|
||||||
continue
|
continue
|
||||||
|
@ -17,8 +17,8 @@ from pyrogram.handlers import MessageHandler, EditedMessageHandler
|
|||||||
|
|
||||||
from pagermaid import help_messages, logs, Config, bot, read_context, all_permissions
|
from pagermaid import help_messages, logs, Config, bot, read_context, all_permissions
|
||||||
from pagermaid.common.ignore import ignore_groups_manager
|
from pagermaid.common.ignore import ignore_groups_manager
|
||||||
|
from pagermaid.enums.command import CommandHandler, CommandHandlerDecorator
|
||||||
from pagermaid.group_manager import Permission
|
from pagermaid.group_manager import Permission
|
||||||
from pagermaid.inject import inject
|
|
||||||
from pagermaid.single_utils import (
|
from pagermaid.single_utils import (
|
||||||
Message,
|
Message,
|
||||||
AlreadyInConversationError,
|
AlreadyInConversationError,
|
||||||
@ -41,9 +41,11 @@ from pyromod.utils import mod_filters
|
|||||||
_lock = asyncio.Lock()
|
_lock = asyncio.Lock()
|
||||||
|
|
||||||
|
|
||||||
def listener(**args):
|
def listener(**args) -> CommandHandlerDecorator:
|
||||||
"""Register an event listener."""
|
"""Register an event listener."""
|
||||||
|
parent_command = args.get("__parent_command")
|
||||||
command = args.get("command")
|
command = args.get("command")
|
||||||
|
allow_parent = args.get("allow_parent", False)
|
||||||
disallow_alias = args.get("disallow_alias", False)
|
disallow_alias = args.get("disallow_alias", False)
|
||||||
need_admin = args.get("need_admin", False)
|
need_admin = args.get("need_admin", False)
|
||||||
description = args.get("description")
|
description = args.get("description")
|
||||||
@ -70,15 +72,20 @@ def listener(**args):
|
|||||||
priority = 0
|
priority = 0
|
||||||
|
|
||||||
if command is not None:
|
if command is not None:
|
||||||
if command in help_messages:
|
if parent_command is None and command in help_messages:
|
||||||
if help_messages[alias_command(command)]["priority"] <= priority:
|
if help_messages[alias_command(command)]["priority"] <= priority:
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"{lang('error_prefix')} {lang('command')} \"{command}\" {lang('has_reg')}"
|
f"{lang('error_prefix')} {lang('command')} \"{command}\" {lang('has_reg')}"
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
block_process = True
|
block_process = True
|
||||||
pattern = rf"^(,|,){alias_command(command, disallow_alias)}(?: |$)([\s\S]*)"
|
real_command = (
|
||||||
sudo_pattern = rf"^(/){alias_command(command, disallow_alias)}(?: |$)([\s\S]*)"
|
alias_command(command, disallow_alias)
|
||||||
|
if parent_command is None
|
||||||
|
else f"{parent_command} {command}"
|
||||||
|
)
|
||||||
|
pattern = rf"^(,|,){real_command}(?: |$)([\s\S]*)"
|
||||||
|
sudo_pattern = rf"^(/){real_command}(?: |$)([\s\S]*)"
|
||||||
if pattern is not None and not pattern.startswith("(?i)"):
|
if pattern is not None and not pattern.startswith("(?i)"):
|
||||||
args["pattern"] = f"(?i){pattern}"
|
args["pattern"] = f"(?i){pattern}"
|
||||||
else:
|
else:
|
||||||
@ -137,7 +144,14 @@ def listener(**args):
|
|||||||
if "block_process" in args:
|
if "block_process" in args:
|
||||||
del args["block_process"]
|
del args["block_process"]
|
||||||
|
|
||||||
def decorator(function):
|
def decorator(function) -> CommandHandler:
|
||||||
|
func = CommandHandler(
|
||||||
|
function,
|
||||||
|
alias_command(command, disallow_alias)
|
||||||
|
if command and parent_command is None
|
||||||
|
else None,
|
||||||
|
)
|
||||||
|
|
||||||
async def handler(client: Client, message: Message):
|
async def handler(client: Client, message: Message):
|
||||||
try:
|
try:
|
||||||
# ignore
|
# ignore
|
||||||
@ -149,11 +163,15 @@ def listener(**args):
|
|||||||
except BaseException:
|
except BaseException:
|
||||||
pass
|
pass
|
||||||
try:
|
try:
|
||||||
parameter = message.matches[0].group(2).split(" ")
|
arguments = message.matches[0].group(2)
|
||||||
|
parameter = arguments.split(" ")
|
||||||
if parameter == [""]:
|
if parameter == [""]:
|
||||||
parameter = []
|
parameter = []
|
||||||
|
if parent_command is not None and command is not None:
|
||||||
|
parameter.insert(0, command)
|
||||||
|
arguments = f"{command} {arguments}".strip()
|
||||||
message.parameter = parameter
|
message.parameter = parameter
|
||||||
message.arguments = message.matches[0].group(2)
|
message.arguments = arguments
|
||||||
except BaseException:
|
except BaseException:
|
||||||
message.parameter = None
|
message.parameter = None
|
||||||
message.arguments = None
|
message.arguments = None
|
||||||
@ -164,18 +182,18 @@ def listener(**args):
|
|||||||
read_context[(message.chat.id, message.id)] = True
|
read_context[(message.chat.id, message.id)] = True
|
||||||
|
|
||||||
if command:
|
if command:
|
||||||
await Hook.command_pre(message, command)
|
await Hook.command_pre(
|
||||||
if data := inject(message, function):
|
message,
|
||||||
await function(**data)
|
parent_command or command,
|
||||||
else:
|
command if parent_command else None,
|
||||||
if function.__code__.co_argcount == 0:
|
)
|
||||||
await function()
|
await func.handler(client, message)
|
||||||
if function.__code__.co_argcount == 1:
|
|
||||||
await function(message)
|
|
||||||
elif function.__code__.co_argcount == 2:
|
|
||||||
await function(client, message)
|
|
||||||
if command:
|
if command:
|
||||||
await Hook.command_post(message, command)
|
await Hook.command_post(
|
||||||
|
message,
|
||||||
|
parent_command or command,
|
||||||
|
command if parent_command else None,
|
||||||
|
)
|
||||||
except StopPropagation as e:
|
except StopPropagation as e:
|
||||||
raise StopPropagation from e
|
raise StopPropagation from e
|
||||||
except KeyboardInterrupt as e:
|
except KeyboardInterrupt as e:
|
||||||
@ -235,30 +253,38 @@ def listener(**args):
|
|||||||
await Hook.process_error_exec(message, command, exc_info, exc_format)
|
await Hook.process_error_exec(message, command, exc_info, exc_format)
|
||||||
if (message.chat.id, message.id) in read_context:
|
if (message.chat.id, message.id) in read_context:
|
||||||
del read_context[(message.chat.id, message.id)]
|
del read_context[(message.chat.id, message.id)]
|
||||||
if block_process:
|
if block_process or (parent_command and not allow_parent):
|
||||||
message.stop_propagation()
|
message.stop_propagation()
|
||||||
message.continue_propagation()
|
message.continue_propagation()
|
||||||
|
|
||||||
bot.add_handler(
|
bot.dispatcher.add_handler(
|
||||||
MessageHandler(handler, filters=base_filters), group=0 + priority
|
MessageHandler(handler, filters=base_filters),
|
||||||
|
group=0 + priority,
|
||||||
|
first=parent_command and not allow_parent,
|
||||||
)
|
)
|
||||||
if command:
|
if command:
|
||||||
bot.add_handler(
|
bot.dispatcher.add_handler(
|
||||||
MessageHandler(handler, filters=sudo_filters), group=50 + priority
|
MessageHandler(handler, filters=sudo_filters),
|
||||||
|
group=50 + priority,
|
||||||
|
first=parent_command and not allow_parent,
|
||||||
)
|
)
|
||||||
if not ignore_edited:
|
if not ignore_edited:
|
||||||
bot.add_handler(
|
bot.dispatcher.add_handler(
|
||||||
EditedMessageHandler(handler, filters=base_filters), group=1 + priority
|
EditedMessageHandler(handler, filters=base_filters),
|
||||||
|
group=1 + priority,
|
||||||
|
first=parent_command and not allow_parent,
|
||||||
)
|
)
|
||||||
if command:
|
if command:
|
||||||
bot.add_handler(
|
bot.dispatcher.add_handler(
|
||||||
EditedMessageHandler(handler, filters=sudo_filters),
|
EditedMessageHandler(handler, filters=sudo_filters),
|
||||||
group=51 + priority,
|
group=51 + priority,
|
||||||
|
first=parent_command and not allow_parent,
|
||||||
)
|
)
|
||||||
|
|
||||||
return handler
|
func.set_handler(handler)
|
||||||
|
return func
|
||||||
|
|
||||||
if description is not None and command is not None:
|
if description is not None and command is not None and parent_command is None:
|
||||||
if parameters is None:
|
if parameters is None:
|
||||||
parameters = ""
|
parameters = ""
|
||||||
help_messages.update(
|
help_messages.update(
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
""" PagerMaid module for different ways to avoid users. """
|
""" PagerMaid module for different ways to avoid users. """
|
||||||
|
|
||||||
from pagermaid import log
|
from pagermaid import log
|
||||||
|
from pagermaid.enums.command import CommandHandler
|
||||||
from pagermaid.single_utils import sqlite
|
from pagermaid.single_utils import sqlite
|
||||||
from pagermaid.utils import lang
|
from pagermaid.utils import lang
|
||||||
from pagermaid.enums import Client, Message
|
from pagermaid.enums import Client, Message
|
||||||
@ -14,14 +15,25 @@ from pagermaid.listener import listener
|
|||||||
description=lang("ghost_des"),
|
description=lang("ghost_des"),
|
||||||
parameters="<true|false|status>",
|
parameters="<true|false|status>",
|
||||||
)
|
)
|
||||||
async def ghost(client: Client, message: Message):
|
async def ghost(message: Message):
|
||||||
"""Toggles ghosting of a user."""
|
"""Toggles ghosting of a user."""
|
||||||
if len(message.parameter) != 1:
|
if len(message.parameter) != 1:
|
||||||
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||||||
return
|
return
|
||||||
|
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||||||
|
|
||||||
|
|
||||||
|
ghost: "CommandHandler"
|
||||||
|
|
||||||
|
|
||||||
|
@ghost.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
outgoing=True,
|
||||||
|
command="true",
|
||||||
|
)
|
||||||
|
async def ghost_true(client: Client, message: Message):
|
||||||
myself = await client.get_me()
|
myself = await client.get_me()
|
||||||
self_user_id = myself.id
|
self_user_id = myself.id
|
||||||
if message.parameter[0] == "true":
|
|
||||||
if message.chat.id == self_user_id:
|
if message.chat.id == self_user_id:
|
||||||
return await message.edit(lang("ghost_e_mark"))
|
return await message.edit(lang("ghost_e_mark"))
|
||||||
sqlite[f"ghosted.chat_id.{str(message.chat.id)}"] = True
|
sqlite[f"ghosted.chat_id.{str(message.chat.id)}"] = True
|
||||||
@ -29,7 +41,16 @@ async def ghost(client: Client, message: Message):
|
|||||||
await log(
|
await log(
|
||||||
f"{lang('ghost_set_f')} ChatID {str(message.chat.id)} {lang('ghost_set_l')}"
|
f"{lang('ghost_set_f')} ChatID {str(message.chat.id)} {lang('ghost_set_l')}"
|
||||||
)
|
)
|
||||||
elif message.parameter[0] == "false":
|
|
||||||
|
|
||||||
|
@ghost.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
outgoing=True,
|
||||||
|
command="false",
|
||||||
|
)
|
||||||
|
async def ghost_false(client: Client, message: Message):
|
||||||
|
myself = await client.get_me()
|
||||||
|
self_user_id = myself.id
|
||||||
if message.chat.id == self_user_id:
|
if message.chat.id == self_user_id:
|
||||||
await message.edit(lang("ghost_e_mark"))
|
await message.edit(lang("ghost_e_mark"))
|
||||||
return
|
return
|
||||||
@ -41,13 +62,18 @@ async def ghost(client: Client, message: Message):
|
|||||||
await log(
|
await log(
|
||||||
f"{lang('ghost_set_f')} ChatID {str(message.chat.id)} {lang('ghost_cancel')}"
|
f"{lang('ghost_set_f')} ChatID {str(message.chat.id)} {lang('ghost_cancel')}"
|
||||||
)
|
)
|
||||||
elif message.parameter[0] == "status":
|
|
||||||
|
|
||||||
|
@ghost.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
outgoing=True,
|
||||||
|
command="status",
|
||||||
|
)
|
||||||
|
async def ghost_status(message: Message):
|
||||||
if sqlite.get(f"ghosted.chat_id.{str(message.chat.id)}", None):
|
if sqlite.get(f"ghosted.chat_id.{str(message.chat.id)}", None):
|
||||||
await message.edit(lang("ghost_e_exist"))
|
await message.edit(lang("ghost_e_exist"))
|
||||||
else:
|
else:
|
||||||
await message.edit(lang("ghost_e_noexist"))
|
await message.edit(lang("ghost_e_noexist"))
|
||||||
else:
|
|
||||||
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
|
||||||
|
|
||||||
|
|
||||||
@listener(
|
@listener(
|
||||||
@ -58,20 +84,42 @@ async def ghost(client: Client, message: Message):
|
|||||||
description=lang("deny_des"),
|
description=lang("deny_des"),
|
||||||
parameters="<true|false|status>",
|
parameters="<true|false|status>",
|
||||||
)
|
)
|
||||||
async def deny(client: Client, message: Message):
|
async def deny(message: Message):
|
||||||
"""Toggles denying of a user."""
|
"""Toggles denying of a user."""
|
||||||
if len(message.parameter) != 1:
|
if len(message.parameter) != 1:
|
||||||
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||||||
return
|
return
|
||||||
|
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
||||||
|
|
||||||
|
|
||||||
|
deny: "CommandHandler"
|
||||||
|
|
||||||
|
|
||||||
|
@deny.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
outgoing=True,
|
||||||
|
need_admin=True,
|
||||||
|
command="true",
|
||||||
|
)
|
||||||
|
async def deny_true(client: Client, message: Message):
|
||||||
myself = await client.get_me()
|
myself = await client.get_me()
|
||||||
self_user_id = myself.id
|
self_user_id = myself.id
|
||||||
if message.parameter[0] == "true":
|
|
||||||
if message.chat.id == self_user_id:
|
if message.chat.id == self_user_id:
|
||||||
return await message.edit(lang("ghost_e_mark"))
|
return await message.edit(lang("ghost_e_mark"))
|
||||||
sqlite[f"denied.chat_id.{str(message.chat.id)}"] = True
|
sqlite[f"denied.chat_id.{str(message.chat.id)}"] = True
|
||||||
await message.safe_delete()
|
await message.safe_delete()
|
||||||
await log(f"ChatID {str(message.chat.id)} {lang('deny_set')}")
|
await log(f"ChatID {str(message.chat.id)} {lang('deny_set')}")
|
||||||
elif message.parameter[0] == "false":
|
|
||||||
|
|
||||||
|
@deny.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
outgoing=True,
|
||||||
|
need_admin=True,
|
||||||
|
command="false",
|
||||||
|
)
|
||||||
|
async def deny_false(client: Client, message: Message):
|
||||||
|
myself = await client.get_me()
|
||||||
|
self_user_id = myself.id
|
||||||
if message.chat.id == self_user_id:
|
if message.chat.id == self_user_id:
|
||||||
await message.edit(lang("ghost_e_mark"))
|
await message.edit(lang("ghost_e_mark"))
|
||||||
return
|
return
|
||||||
@ -81,13 +129,19 @@ async def deny(client: Client, message: Message):
|
|||||||
return await message.edit(lang("deny_e_noexist"))
|
return await message.edit(lang("deny_e_noexist"))
|
||||||
await message.safe_delete()
|
await message.safe_delete()
|
||||||
await log(f"ChatID {str(message.chat.id)} {lang('deny_cancel')}")
|
await log(f"ChatID {str(message.chat.id)} {lang('deny_cancel')}")
|
||||||
elif message.parameter[0] == "status":
|
|
||||||
|
|
||||||
|
@deny.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
outgoing=True,
|
||||||
|
need_admin=True,
|
||||||
|
command="status",
|
||||||
|
)
|
||||||
|
async def deny_status(message: Message):
|
||||||
if sqlite.get(f"denied.chat_id.{str(message.chat.id)}", None):
|
if sqlite.get(f"denied.chat_id.{str(message.chat.id)}", None):
|
||||||
await message.edit(lang("deny_e_exist"))
|
await message.edit(lang("deny_e_exist"))
|
||||||
else:
|
else:
|
||||||
await message.edit(lang("deny_e_noexist"))
|
await message.edit(lang("deny_e_noexist"))
|
||||||
else:
|
|
||||||
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
|
|
||||||
|
|
||||||
|
|
||||||
@listener(is_plugin=False, incoming=True, outgoing=False, ignore_edited=True)
|
@listener(is_plugin=False, incoming=True, outgoing=False, ignore_edited=True)
|
||||||
|
@ -134,7 +134,7 @@ async def mixpanel_init_id(bot: Client):
|
|||||||
|
|
||||||
|
|
||||||
@Hook.command_postprocessor()
|
@Hook.command_postprocessor()
|
||||||
async def mixpanel_report(bot: Client, message: Message, command):
|
async def mixpanel_report(bot: Client, message: Message, command, sub_command):
|
||||||
if not Config.ALLOW_ANALYTIC:
|
if not Config.ALLOW_ANALYTIC:
|
||||||
return
|
return
|
||||||
await set_people(bot)
|
await set_people(bot)
|
||||||
@ -144,11 +144,14 @@ async def mixpanel_report(bot: Client, message: Message, command):
|
|||||||
sender_id = message.sender_chat.id if message.sender_chat else sender_id
|
sender_id = message.sender_chat.id if message.sender_chat else sender_id
|
||||||
if sender_id < 0 and message.outgoing:
|
if sender_id < 0 and message.outgoing:
|
||||||
sender_id = bot.me.id
|
sender_id = bot.me.id
|
||||||
|
properties = {"command": command, "bot_id": bot.me.id}
|
||||||
|
if sub_command:
|
||||||
|
properties["sub_command"] = sub_command
|
||||||
bot.loop.create_task(
|
bot.loop.create_task(
|
||||||
mp.track(
|
mp.track(
|
||||||
str(sender_id),
|
str(sender_id),
|
||||||
f"Function {command}",
|
f"Function {command}",
|
||||||
{"command": command, "bot_id": bot.me.id},
|
properties,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -1,3 +1,4 @@
|
|||||||
|
from pagermaid.enums.command import CommandHandler
|
||||||
from pagermaid.single_utils import sqlite
|
from pagermaid.single_utils import sqlite
|
||||||
from pagermaid.listener import listener
|
from pagermaid.listener import listener
|
||||||
from pagermaid.group_manager import (
|
from pagermaid.group_manager import (
|
||||||
@ -29,11 +30,21 @@ def from_msg_get_sudo_id(message: Message) -> int:
|
|||||||
parameters="{on|off|add|remove|gaddp|gaddu|gdelp|gdelu|glist|uaddp|udelp|list}",
|
parameters="{on|off|add|remove|gaddp|gaddu|gdelp|gdelu|glist|uaddp|udelp|list}",
|
||||||
description=lang("sudo_des"),
|
description=lang("sudo_des"),
|
||||||
)
|
)
|
||||||
async def sudo_change(client: Client, message: Message):
|
async def sudo_change(message: Message):
|
||||||
"""To enable or disable sudo of your userbot."""
|
"""To enable or disable sudo of your userbot."""
|
||||||
input_str = message.arguments
|
await edit_delete(message, lang("arg_error"))
|
||||||
|
|
||||||
|
|
||||||
|
sudo_change: "CommandHandler"
|
||||||
|
|
||||||
|
|
||||||
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="on",
|
||||||
|
need_admin=True,
|
||||||
|
)
|
||||||
|
async def sudo_on(message: Message):
|
||||||
sudo = get_sudo_list()
|
sudo = get_sudo_list()
|
||||||
if input_str == "on":
|
|
||||||
if _status_sudo():
|
if _status_sudo():
|
||||||
return await edit_delete(message, lang("sudo_has_enabled"))
|
return await edit_delete(message, lang("sudo_has_enabled"))
|
||||||
sqlite["sudo_enable"] = True
|
sqlite["sudo_enable"] = True
|
||||||
@ -46,7 +57,15 @@ async def sudo_change(client: Client, message: Message):
|
|||||||
return await message.edit(
|
return await message.edit(
|
||||||
text,
|
text,
|
||||||
)
|
)
|
||||||
elif input_str == "off":
|
|
||||||
|
|
||||||
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="off",
|
||||||
|
need_admin=True,
|
||||||
|
)
|
||||||
|
async def sudo_off(message: Message):
|
||||||
|
sudo = get_sudo_list()
|
||||||
if _status_sudo():
|
if _status_sudo():
|
||||||
del sqlite["sudo_enable"]
|
del sqlite["sudo_enable"]
|
||||||
text = f"__{lang('sudo_disable')}__\n"
|
text = f"__{lang('sudo_disable')}__\n"
|
||||||
@ -59,7 +78,15 @@ async def sudo_change(client: Client, message: Message):
|
|||||||
text,
|
text,
|
||||||
)
|
)
|
||||||
await edit_delete(message, lang("sudo_has_disabled"))
|
await edit_delete(message, lang("sudo_has_disabled"))
|
||||||
elif input_str == "add":
|
|
||||||
|
|
||||||
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="add",
|
||||||
|
need_admin=True,
|
||||||
|
)
|
||||||
|
async def sudo_add(message: Message):
|
||||||
|
sudo = get_sudo_list()
|
||||||
from_id = from_msg_get_sudo_id(message)
|
from_id = from_msg_get_sudo_id(message)
|
||||||
if from_id in sudo:
|
if from_id in sudo:
|
||||||
return await edit_delete(message, f"__{lang('sudo_add')}__")
|
return await edit_delete(message, f"__{lang('sudo_add')}__")
|
||||||
@ -70,7 +97,15 @@ async def sudo_change(client: Client, message: Message):
|
|||||||
await message.edit(f"__{lang('sudo_add')}__")
|
await message.edit(f"__{lang('sudo_add')}__")
|
||||||
else:
|
else:
|
||||||
await message.edit(f"__{lang('sudo_add_chat')}__")
|
await message.edit(f"__{lang('sudo_add_chat')}__")
|
||||||
elif input_str == "remove":
|
|
||||||
|
|
||||||
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="remove",
|
||||||
|
need_admin=True,
|
||||||
|
)
|
||||||
|
async def sudo_remove(message: Message):
|
||||||
|
sudo = get_sudo_list()
|
||||||
from_id = from_msg_get_sudo_id(message)
|
from_id = from_msg_get_sudo_id(message)
|
||||||
if from_id not in sudo:
|
if from_id not in sudo:
|
||||||
return await edit_delete(message, f"__{lang('sudo_no')}__")
|
return await edit_delete(message, f"__{lang('sudo_no')}__")
|
||||||
@ -80,7 +115,15 @@ async def sudo_change(client: Client, message: Message):
|
|||||||
await message.edit(f"__{lang('sudo_remove')}__")
|
await message.edit(f"__{lang('sudo_remove')}__")
|
||||||
else:
|
else:
|
||||||
await message.edit(f"__{lang('sudo_remove_chat')}__")
|
await message.edit(f"__{lang('sudo_remove_chat')}__")
|
||||||
elif input_str == "list":
|
|
||||||
|
|
||||||
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="list",
|
||||||
|
need_admin=True,
|
||||||
|
)
|
||||||
|
async def sudo_list(client: Client, message: Message):
|
||||||
|
sudo = get_sudo_list()
|
||||||
if len(sudo) == 0:
|
if len(sudo) == 0:
|
||||||
return await edit_delete(message, f"__{lang('sudo_no_one')}__")
|
return await edit_delete(message, f"__{lang('sudo_no_one')}__")
|
||||||
text = f"**{lang('sudo_list')}**\n\n"
|
text = f"**{lang('sudo_list')}**\n\n"
|
||||||
@ -99,56 +142,108 @@ async def sudo_change(client: Client, message: Message):
|
|||||||
for j in permissions.get_permissions_for_user(str(i)):
|
for j in permissions.get_permissions_for_user(str(i)):
|
||||||
text += f" • {'-' if j[2] == 'ejection' else ''}{j[1]}\n"
|
text += f" • {'-' if j[2] == 'ejection' else ''}{j[1]}\n"
|
||||||
except Exception:
|
except Exception:
|
||||||
text += (
|
text += f"• `{i}` - {' '.join(permissions.get_roles_for_user(str(i)))}\n"
|
||||||
f"• `{i}` - {' '.join(permissions.get_roles_for_user(str(i)))}\n"
|
|
||||||
)
|
|
||||||
await message.edit(text)
|
await message.edit(text)
|
||||||
elif len(message.parameter) > 0:
|
|
||||||
if len(message.parameter) == 2:
|
|
||||||
|
def check_parameter_length(length: int, check_permission: bool):
|
||||||
|
def decorator(func):
|
||||||
|
async def wrapper(message: Message):
|
||||||
|
if len(message.parameter) != length:
|
||||||
|
return await edit_delete(message, lang("arg_error"))
|
||||||
|
if check_permission:
|
||||||
|
sudo = get_sudo_list()
|
||||||
from_id = from_msg_get_sudo_id(message)
|
from_id = from_msg_get_sudo_id(message)
|
||||||
if message.parameter[0] == "glist":
|
if from_id not in sudo:
|
||||||
if not (
|
return await edit_delete(message, f"__{lang('sudo_no')}__")
|
||||||
data := permissions.get_permissions_for_user(
|
|
||||||
str(message.parameter[1])
|
return await func(message)
|
||||||
|
|
||||||
|
return wrapper
|
||||||
|
|
||||||
|
return decorator
|
||||||
|
|
||||||
|
|
||||||
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="glist",
|
||||||
|
need_admin=True,
|
||||||
)
|
)
|
||||||
):
|
@check_parameter_length(2, False)
|
||||||
|
async def sudo_glist(message: Message):
|
||||||
|
if not (data := permissions.get_permissions_for_user(str(message.parameter[1]))):
|
||||||
return await edit_delete(message, f"__{lang('sudo_group_list')}__")
|
return await edit_delete(message, f"__{lang('sudo_group_list')}__")
|
||||||
text = f"**{message.parameter[1]} {lang('sudo_group_list')}**\n\n"
|
text = f"**{message.parameter[1]} {lang('sudo_group_list')}**\n\n"
|
||||||
for i in data:
|
for i in data:
|
||||||
text += f" • `{'-' if i[2] == 'ejection' else ''}{i[1]}`\n"
|
text += f" • `{'-' if i[2] == 'ejection' else ''}{i[1]}`\n"
|
||||||
return await message.edit(text)
|
return await message.edit(text)
|
||||||
if from_id not in sudo:
|
|
||||||
return await edit_delete(message, f"__{lang('sudo_no')}__")
|
|
||||||
elif message.parameter[0] == "gaddu":
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="gaddu",
|
||||||
|
need_admin=True,
|
||||||
|
)
|
||||||
|
@check_parameter_length(2, True)
|
||||||
|
async def sudo_gaddu(message: Message):
|
||||||
|
from_id = from_msg_get_sudo_id(message)
|
||||||
add_user_to_group(str(from_id), message.parameter[1])
|
add_user_to_group(str(from_id), message.parameter[1])
|
||||||
return await message.edit(lang("sudo_group_add_user"))
|
return await message.edit(lang("sudo_group_add_user"))
|
||||||
elif message.parameter[0] == "gdelu":
|
|
||||||
|
|
||||||
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="gdelu",
|
||||||
|
need_admin=True,
|
||||||
|
)
|
||||||
|
@check_parameter_length(2, True)
|
||||||
|
async def sudo_gdelu(message: Message):
|
||||||
|
from_id = from_msg_get_sudo_id(message)
|
||||||
remove_user_from_group(str(from_id), message.parameter[1])
|
remove_user_from_group(str(from_id), message.parameter[1])
|
||||||
return await message.edit(lang("sudo_group_del_user"))
|
return await message.edit(lang("sudo_group_del_user"))
|
||||||
elif message.parameter[0] == "uaddp":
|
|
||||||
|
|
||||||
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="uaddp",
|
||||||
|
need_admin=True,
|
||||||
|
)
|
||||||
|
@check_parameter_length(2, True)
|
||||||
|
async def sudo_uaddp(message: Message):
|
||||||
|
from_id = from_msg_get_sudo_id(message)
|
||||||
add_permission_for_user(str(from_id), Permission(message.parameter[1]))
|
add_permission_for_user(str(from_id), Permission(message.parameter[1]))
|
||||||
return await message.edit(lang("sudo_user_add_per"))
|
return await message.edit(lang("sudo_user_add_per"))
|
||||||
elif message.parameter[0] == "udelp":
|
|
||||||
remove_permission_for_user(
|
|
||||||
str(from_id), Permission(message.parameter[1])
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="udelp",
|
||||||
|
need_admin=True,
|
||||||
)
|
)
|
||||||
|
@check_parameter_length(2, True)
|
||||||
|
async def sudo_udelp(message: Message):
|
||||||
|
from_id = from_msg_get_sudo_id(message)
|
||||||
|
remove_permission_for_user(str(from_id), Permission(message.parameter[1]))
|
||||||
return await message.edit(lang("sudo_user_del_per"))
|
return await message.edit(lang("sudo_user_del_per"))
|
||||||
else:
|
|
||||||
return await edit_delete(message, lang("arg_error"))
|
|
||||||
if len(message.parameter) == 3:
|
@sudo_change.sub_command(
|
||||||
if message.parameter[0] == "gaddp":
|
is_plugin=False,
|
||||||
add_permission_for_group(
|
command="gaddp",
|
||||||
message.parameter[1], Permission(message.parameter[2])
|
need_admin=True,
|
||||||
)
|
)
|
||||||
|
@check_parameter_length(3, False)
|
||||||
|
async def sudo_gaddp(message: Message):
|
||||||
|
add_permission_for_group(message.parameter[1], Permission(message.parameter[2]))
|
||||||
return await message.edit(lang("sudo_group_add_per"))
|
return await message.edit(lang("sudo_group_add_per"))
|
||||||
elif message.parameter[0] == "gdelp":
|
|
||||||
remove_permission_for_group(
|
|
||||||
message.parameter[1], Permission(message.parameter[2])
|
@sudo_change.sub_command(
|
||||||
|
is_plugin=False,
|
||||||
|
command="gdelp",
|
||||||
|
need_admin=True,
|
||||||
)
|
)
|
||||||
|
@check_parameter_length(3, False)
|
||||||
|
async def sudo_gdelp(message: Message):
|
||||||
|
remove_permission_for_group(message.parameter[1], Permission(message.parameter[2]))
|
||||||
return await message.edit(lang("sudo_group_del_per"))
|
return await message.edit(lang("sudo_group_del_per"))
|
||||||
else:
|
|
||||||
return await edit_delete(message, lang("arg_error"))
|
|
||||||
else:
|
|
||||||
await edit_delete(message, lang("arg_error"))
|
|
||||||
else:
|
|
||||||
await edit_delete(message, lang("arg_error"))
|
|
||||||
|
@ -21,6 +21,7 @@ along with pyromod. If not, see <https://www.gnu.org/licenses/>.
|
|||||||
import asyncio
|
import asyncio
|
||||||
import contextlib
|
import contextlib
|
||||||
import functools
|
import functools
|
||||||
|
from collections import OrderedDict
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from typing import Optional, List, Union
|
from typing import Optional, List, Union
|
||||||
|
|
||||||
@ -425,3 +426,24 @@ class Dispatcher(pyrogram.dispatcher.Dispatcher): # noqa
|
|||||||
lock.release()
|
lock.release()
|
||||||
|
|
||||||
self.loop.create_task(fn())
|
self.loop.create_task(fn())
|
||||||
|
|
||||||
|
@patchable
|
||||||
|
def add_handler(self, handler, group: int, first: bool):
|
||||||
|
if not first:
|
||||||
|
return self.oldadd_handler(handler, group)
|
||||||
|
|
||||||
|
async def fn():
|
||||||
|
for lock in self.locks_list:
|
||||||
|
await lock.acquire()
|
||||||
|
|
||||||
|
try:
|
||||||
|
if group not in self.groups:
|
||||||
|
self.groups[group] = []
|
||||||
|
self.groups = OrderedDict(sorted(self.groups.items()))
|
||||||
|
|
||||||
|
self.groups[group].insert(0, handler)
|
||||||
|
finally:
|
||||||
|
for lock in self.locks_list:
|
||||||
|
lock.release()
|
||||||
|
|
||||||
|
self.loop.create_task(fn())
|
||||||
|
Loading…
Reference in New Issue
Block a user