mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-27 18:51:13 +00:00
qu 人形 @fish_quote_reply_bot
All checks were successful
Github commit to telegram / build (push) Successful in 37s
All checks were successful
Github commit to telegram / build (push) Successful in 37s
Co-authored-by: omg-xtao <100690902+omg-xtao@users.noreply.github.com>
This commit is contained in:
parent
242bb5bcc5
commit
e109dc513d
61
qu/main.py
Normal file
61
qu/main.py
Normal file
@ -0,0 +1,61 @@
|
||||
from typing import Union, Tuple
|
||||
|
||||
from pyrogram.types import User, Chat
|
||||
|
||||
from pagermaid.enums import Client, Message
|
||||
from pagermaid.listener import listener
|
||||
|
||||
|
||||
def get_text(opt: str, opt1: str, user1: Union[User, Chat], user2: Union[User, Chat]) -> str:
|
||||
first = user1.mention
|
||||
second = user2.mention
|
||||
return f"{first} {opt} {second} {opt1}!"
|
||||
|
||||
|
||||
def get_second_chat(message: Message, me: User) -> Union[User, Chat]:
|
||||
if not message.reply_to_message:
|
||||
return me
|
||||
reply = message.reply_to_message
|
||||
if reply.from_user:
|
||||
return reply.from_user
|
||||
if reply.sender_chat:
|
||||
return reply.sender_chat
|
||||
return me
|
||||
|
||||
|
||||
def get_opt(message: Message) -> Tuple[str, str]:
|
||||
text = message.arguments
|
||||
# 分割命令和参数
|
||||
opt, _, opt1 = text.partition(' ')
|
||||
# opt1 为空的话给 opt 加个"了"
|
||||
if opt and not opt1:
|
||||
opt += "了"
|
||||
return opt, opt1
|
||||
|
||||
|
||||
@listener(command="qu", description=r"使用“qu”代替“/”,以拙劣地模仿 @fish_quote_reply_bot")
|
||||
async def qu(bot: Client, message: Message):
|
||||
me = await bot.get_me()
|
||||
# 分割命令和参数
|
||||
opt, opt1 = get_opt(message)
|
||||
# opt 为空报错
|
||||
if not opt:
|
||||
return await message.edit("E:未指定内容")
|
||||
user1 = me
|
||||
user2 = get_second_chat(message, me)
|
||||
text = get_text(opt, opt1, user1, user2)
|
||||
await message.edit(text, disable_web_page_preview=True)
|
||||
|
||||
|
||||
@listener(command="uq", description=r"使用“uq”代替“\”,以拙劣地模仿 @fish_quote_reply_bot")
|
||||
async def uq(bot: Client, message: Message):
|
||||
me = await bot.get_me()
|
||||
# 分割命令和参数
|
||||
opt, opt1 = get_opt(message)
|
||||
# opt 为空报错
|
||||
if not opt:
|
||||
return await message.edit("E:未指定内容")
|
||||
user1 = me
|
||||
user2 = get_second_chat(message, me)
|
||||
text = get_text(opt, opt1, user2, user1)
|
||||
await message.edit(text, disable_web_page_preview=True)
|
Loading…
Reference in New Issue
Block a user