59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
|
import asyncio
|
|||
|
|
|
|||
|
|
from core.custom import command
|
|||
|
|
from loguru import logger
|
|||
|
|
from pyrogram import Client, filters
|
|||
|
|
from pyrogram.errors import RPCError
|
|||
|
|
from pyrogram.types import Message
|
|||
|
|
from tools.constants import STORE_GHOST_DATA
|
|||
|
|
from tools.ghosts import get_ghost_to_read
|
|||
|
|
from tools.helpers import Parameters, delete_this, get_fullname
|
|||
|
|
from tools.storage import SimpleStore
|
|||
|
|
|
|||
|
|
|
|||
|
|
@Client.on_message(filters.incoming, group=-2)
|
|||
|
|
async def ghost_event(cli: Client, msg: Message):
|
|||
|
|
"""自动标记对话为<已读>"""
|
|||
|
|
if await get_ghost_to_read(msg.chat.id):
|
|||
|
|
try:
|
|||
|
|
await cli.read_history(msg.chat.id)
|
|||
|
|
except RPCError as e:
|
|||
|
|
logger.error(e)
|
|||
|
|
else:
|
|||
|
|
if msg.text or msg.caption:
|
|||
|
|
text = msg.text or msg.text
|
|||
|
|
text = f"Ghost | {msg.chat.title} | {get_fullname(msg.from_user)} | {text}"
|
|||
|
|
logger.debug(text)
|
|||
|
|
finally:
|
|||
|
|
await logger.complete()
|
|||
|
|
|
|||
|
|
|
|||
|
|
@Client.on_message(command('ghost'))
|
|||
|
|
async def ghost(_: Client, msg: Message):
|
|||
|
|
"""指令：将该对话标记为可自动<已读>状态"""
|
|||
|
|
_, opt = Parameters.get(msg)
|
|||
|
|
chat = msg.chat
|
|||
|
|
|
|||
|
|
async with SimpleStore(auto_flush=False) as store:
|
|||
|
|
ghost_data = store.get_data(STORE_GHOST_DATA)
|
|||
|
|
# ghost状态
|
|||
|
|
if opt == 'status':
|
|||
|
|
text = f"此对话是否开启ghost：{'✅' if chat.id in ghost_data else '❌'}"
|
|||
|
|
elif opt == 'list':
|
|||
|
|
tmp = '\n'.join(f'```{k} {v}```' for k, v in ghost_data.items())
|
|||
|
|
text = f"📢 已开启ghost的对话名单：\n{tmp}"
|
|||
|
|
# ghost开关
|
|||
|
|
else:
|
|||
|
|
if chat.id in ghost_data:
|
|||
|
|
text = "❌ 已关闭此对话的ghost"
|
|||
|
|
ghost_data.pop(chat.id, None)
|
|||
|
|
else:
|
|||
|
|
text = "✅ 已开启此对话的ghost"
|
|||
|
|
ghost_data[chat.id] = chat.title or get_fullname(msg.from_user)
|
|||
|
|
store.flush()
|
|||
|
|
|
|||
|
|
await msg.edit_text(text, parse_mode='md')
|
|||
|
|
await asyncio.sleep(1)
|
|||
|
|
if opt != 'status' and opt != 'list':
|
|||
|
|
await delete_this(msg)
|