iShotaBot/modules/repeater.py
2024-06-19 20:47:36 +08:00

55 lines
1.4 KiB
Python
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

from pyrogram import Client, filters, ContinuePropagation
from pyrogram.types import Message
from init import bot
last_msg = {}
last_repeat_msg = {}
repeat_count = {}
@bot.on_message(filters.incoming & filters.group)
async def repeater_handler(client: Client, message: Message):
global last_msg, last_repeat_msg, repeat_count
group_id = message.chat.id
try:
last_msg[group_id]
except KeyError:
last_msg[group_id] = {}
try:
last_repeat_msg[group_id]
except KeyError:
last_repeat_msg[group_id] = ""
msg = t_msg = message.text
if not msg:
raise ContinuePropagation
if (
msg.startswith("/")
or msg.startswith("!")
or msg.startswith(",")
or msg.startswith("")
):
raise ContinuePropagation
last_msg_text = last_msg[group_id].get("text", "")
last_msg_id = last_msg[group_id].get("id", 0)
if message.id == last_msg_id:
raise ContinuePropagation
if msg != last_msg_text or msg == last_repeat_msg[group_id]:
last_msg[group_id] = {"text": msg, "id": message.id}
repeat_count[group_id] = 0
else:
repeat_count[group_id] += 1
last_repeat_msg[group_id] = ""
if repeat_count[group_id] >= 2:
await client.send_message(group_id, t_msg)
repeat_count[group_id] = 0
last_msg[group_id] = {}
last_repeat_msg[group_id] = msg
raise ContinuePropagation