98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
|
import asyncio
|
|||
|
|
|
|||
|
|
from core import command
|
|||
|
|
from core.custom import is_traced
|
|||
|
|
from loguru import logger
|
|||
|
|
from pyrogram import Client
|
|||
|
|
from pyrogram.errors import BadRequest, FloodWait, RPCError
|
|||
|
|
from pyrogram.types import Message
|
|||
|
|
from tools.constants import REACTIONS, STORE_TRACE_DATA
|
|||
|
|
from tools.helpers import Parameters, delete_this
|
|||
|
|
from tools.storage import SimpleStore
|
|||
|
|
|
|||
|
|
|
|||
|
|
@Client.on_message(is_traced(), group=-4)
|
|||
|
|
async def trace_event(cli: Client, msg: Message):
|
|||
|
|
user = msg.from_user
|
|||
|
|
async with SimpleStore(auto_flush=False) as store:
|
|||
|
|
try:
|
|||
|
|
emoji = store.get_data(STORE_TRACE_DATA).get(user.id)
|
|||
|
|
await cli.send_reaction(
|
|||
|
|
msg.chat.id, msg.message_id, emoji
|
|||
|
|
)
|
|||
|
|
except BadRequest:
|
|||
|
|
failure = f"Group named <{msg.chat.title}> can't use {emoji} to react."
|
|||
|
|
store.data[STORE_TRACE_DATA].pop(user.id, None)
|
|||
|
|
store.flush()
|
|||
|
|
await cli.send_message('me', failure)
|
|||
|
|
except RPCError as e:
|
|||
|
|
logger.error(e)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@Client.on_message(command('trace'))
|
|||
|
|
async def trace(cli: Client, msg: Message):
|
|||
|
|
"""群组中追着丢emoji
|
|||
|
|
指令：-trace
|
|||
|
|
用法：-trace <emoji|clear|list> 用于回复一条消息
|
|||
|
|
"""
|
|||
|
|
cmd, opt = Parameters.get(msg)
|
|||
|
|
replied_msg = msg.reply_to_message
|
|||
|
|
|
|||
|
|
if not opt and not replied_msg:
|
|||
|
|
await msg.edit_text(f'❗️ Use `{cmd}` to reply a message.')
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if opt != 'clear' and opt != 'list':
|
|||
|
|
emoji = '💩' if opt not in REACTIONS else opt
|
|||
|
|
user = replied_msg.from_user
|
|||
|
|
try:
|
|||
|
|
await cli.send_reaction(
|
|||
|
|
msg.chat.id,
|
|||
|
|
replied_msg.message_id,
|
|||
|
|
emoji
|
|||
|
|
)
|
|||
|
|
except RPCError as e:
|
|||
|
|
logger.warning(e)
|
|||
|
|
await msg.edit_text(f"❗️ Can't use {emoji} in this chat.")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
async with SimpleStore() as store:
|
|||
|
|
trace_data = store.get_data(STORE_TRACE_DATA)
|
|||
|
|
if opt != 'clear' and opt != 'list':
|
|||
|
|
# 追踪列表中没有，则添加
|
|||
|
|
if not trace_data.get(user.id):
|
|||
|
|
trace_data[user.id] = emoji
|
|||
|
|
text = f"✅ 添加 {user.mention(style='md')} 到trace列表"
|
|||
|
|
logger.success(text)
|
|||
|
|
# 追踪列表有，则删除
|
|||
|
|
elif trace_data.pop(user.id, False):
|
|||
|
|
text = f"✅ 将 {user.mention(style='md')} 从trace列表移除"
|
|||
|
|
logger.success(text)
|
|||
|
|
# 删除失败？？
|
|||
|
|
else:
|
|||
|
|
text = f"❌ 竟然将 {user.mention(style='md')} 从trace列表移除失败！！!"
|
|||
|
|
logger.warning(text)
|
|||
|
|
elif opt == 'clear':
|
|||
|
|
trace_data.clear()
|
|||
|
|
text = "✅ 已清空trace名单"
|
|||
|
|
elif opt == 'list':
|
|||
|
|
tmp = '\n'.join(f"`{k}` | {v}" for k, v in trace_data.items())
|
|||
|
|
text = f"📢 trace名单：\n{tmp}"
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
await msg.edit_text(text, parse_mode='md')
|
|||
|
|
|
|||
|
|
except FloodWait as e:
|
|||
|
|
await asyncio.sleep(e.x)
|
|||
|
|
await msg.edit_text(text, parse_mode='md')
|
|||
|
|
|
|||
|
|
except RPCError as e:
|
|||
|
|
logger.error(e)
|
|||
|
|
await msg.edit_text(f"Network error | `{e}`")
|
|||
|
|
|
|||
|
|
finally:
|
|||
|
|
if opt != 'clear' and opt != 'list':
|
|||
|
|
await asyncio.sleep(3)
|
|||
|
|
await delete_this(msg)
|
|||
|
|
await logger.complete()
|