From 44c59927be05af17ce2b7de90c1270e5378a8dfc Mon Sep 17 00:00:00 2001 From: yasirarism <55983182+yasirarism@users.noreply.github.com> Date: Tue, 4 Jul 2023 11:01:25 +0700 Subject: [PATCH] fix --- misskaty/core/misskaty_patch/decorators/command.py | 3 +++ misskaty/plugins/ytdl_plugins.py | 4 +--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/misskaty/core/misskaty_patch/decorators/command.py b/misskaty/core/misskaty_patch/decorators/command.py index 136ae3e8..fa27ff99 100644 --- a/misskaty/core/misskaty_patch/decorators/command.py +++ b/misskaty/core/misskaty_patch/decorators/command.py @@ -12,6 +12,7 @@ def command( group_only: typing.Union[bool, bool] = False, self_admin: typing.Union[bool, bool] = False, self_only: typing.Union[bool, bool] = False, + no_channel: typing.Union[bool, bool] = False, handler: typing.Optional[list] = None, filter: typing.Union[pyrogram.filters.Filter, pyrogram.filters.Filter] = None, *args, @@ -80,6 +81,8 @@ def command( async def decorator(client, message: pyrogram.types.Message): if is_disabled: return await message.reply_text("Sorry, this command has been disabled by owner.") + if no_channel: + return await message.reply_text("I'm cannot identify user. Use my command in private chat.") if ( self_admin and message.chat.type != pyrogram.enums.ChatType.SUPERGROUP diff --git a/misskaty/plugins/ytdl_plugins.py b/misskaty/plugins/ytdl_plugins.py index 7c59e7f6..a464c50b 100644 --- a/misskaty/plugins/ytdl_plugins.py +++ b/misskaty/plugins/ytdl_plugins.py @@ -35,13 +35,11 @@ def rand_key(): return str(uuid4())[:8] -@app.on_cmd("ytsearch", filter=~filters.channel) +@app.on_cmd("ytsearch", no_channel=True) @capture_err @ratelimiter @use_chat_lang() async def ytsearch(self: Client, ctx: Message, strings): - if ctx.sender_chat: - return await ctx.reply_msg(strings("no_channel")) if len(ctx.command) == 1: return await ctx.reply_msg(strings("no_query")) query = ctx.text.split(" ", maxsplit=1)[1]