mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2025-12-29 17:44:50 +00:00
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
# SPDX-License-Identifier: MIT
|
|
# Copyright (c) 2018-2022 Amano Team
|
|
|
|
import html
|
|
|
|
import regex
|
|
from pyrogram import Client, filters
|
|
from pyrogram.errors import MessageEmpty
|
|
from pyrogram.types import Message
|
|
|
|
from misskaty import app
|
|
from misskaty.core.decorator.ratelimiter import ratelimiter
|
|
|
|
|
|
@app.on_message(filters.regex(r"^s/(.+)?/(.+)?(/.+)?") & filters.reply)
|
|
@ratelimiter
|
|
async def sed(self: Client, ctx: Message):
|
|
exp = regex.split(r"(?<![^\\]\\)/", ctx.text)
|
|
pattern = exp[1]
|
|
replace_with = exp[2].replace(r"\/", "/")
|
|
flags = exp[3] if len(exp) > 3 else ""
|
|
|
|
rflags = 0
|
|
|
|
count = 0 if "g" in flags else 1
|
|
if "i" in flags and "s" in flags:
|
|
rflags = regex.I | regex.S
|
|
elif "i" in flags:
|
|
rflags = regex.I
|
|
elif "s" in flags:
|
|
rflags = regex.S
|
|
|
|
text = ctx.reply_to_message.text or ctx.reply_to_message.caption
|
|
|
|
if not text:
|
|
return
|
|
|
|
try:
|
|
res = regex.sub(
|
|
pattern, replace_with, text, count=count, flags=rflags, timeout=1
|
|
)
|
|
except TimeoutError:
|
|
return await ctx.reply_msg("Oops, your regex pattern has run for too long.")
|
|
except regex.error as e:
|
|
return await ctx.reply_msg(str(e))
|
|
else:
|
|
try:
|
|
await self.send_msg(
|
|
ctx.chat.id,
|
|
f"<pre>{html.escape(res)}</pre>",
|
|
reply_to_message_id=ctx.reply_to_message.id,
|
|
)
|
|
except MessageEmpty:
|
|
return await ctx.reply_msg(
|
|
"Please reply message to use this feature.", del_in=5
|
|
)
|
|
except Exception as e:
|
|
return await ctx.reply_msg(f"ERROR: {str(e)}")
|