mirror of
https://github.com/TeamPGM/PagerMaid_Plugins.git
synced 2024-11-22 00:35:36 +00:00
自动回复Sticker v1.0.0
This commit is contained in:
Pentacene
GitHub
parent
5a2676801f
commit
90eb51f5db
181
autoreplysticker.py
Normal file
181
autoreplysticker.py
Normal file
@ -0,0 +1,181 @@
|
||||
""" Pagermaid Plugins AutoReplySticker """
|
||||
# ______ _
|
||||
# | ___ \ | |
|
||||
# | |_/ /__ _ __ | |_ __ _ ___ ___ _ __ ___
|
||||
# | __/ _ \ '_ \| __/ _` |/ __/ _ \ '_ \ / _ \
|
||||
# | | | __/ | | | || (_| | (_| __/ | | | __/
|
||||
# \_| \___|_| |_|\__\__,_|\___\___|_| |_|\___|
|
||||
#
|
||||
|
||||
from os import mkdir
|
||||
from os.path import exists
|
||||
from asyncio import sleep
|
||||
from random import randint
|
||||
import yaml
|
||||
from telethon.tl.functions.messages import GetAllStickersRequest
|
||||
from telethon.tl.functions.messages import GetStickerSetRequest
|
||||
from telethon.tl.types import InputStickerSetID
|
||||
from pagermaid import log
|
||||
from pagermaid.listener import listener
|
||||
|
||||
async def ars_check(context):
|
||||
try:
|
||||
config = yaml.load(open(r"./plugins/autoreplysticker/config.yml"), Loader=yaml.FullLoader)
|
||||
except FileNotFoundError:
|
||||
await context.edit("自动回复贴纸的相关设置不存在。\n请使用 `-ars help` 查看设置方法")
|
||||
return
|
||||
_sticker_id = config['sticker_id']
|
||||
_sticker_hash = config['sticker_hash']
|
||||
_num = config['num']
|
||||
_time = config['time']
|
||||
_noti = await context.reply(
|
||||
'您当前的设置为:\n'
|
||||
f'sticker_id: {_sticker_id}\n'
|
||||
f'sticker_hash: {_sticker_hash}\n'
|
||||
f'time: {_time}\n'
|
||||
f'num: {_num}')
|
||||
await context.delete()
|
||||
await sleep(15)
|
||||
await _noti.delete()
|
||||
|
||||
async def ars_getall(context):
|
||||
sticker_sets = await context.client(GetAllStickersRequest(0))
|
||||
sticker_pack_list = []
|
||||
for sticker_set in sticker_sets.sets:
|
||||
if len(sticker_pack_list) < 10:
|
||||
text = "我发现了一个Sticker Pack，名为\n" + sticker_set.title + "\n" + "ID为： `" + \
|
||||
str(sticker_set.id) + "` \n" + "Hash为： `" + str(sticker_set.access_hash) + \
|
||||
"` \n" + "共有" + str(sticker_set.count) + "张"
|
||||
sticker_pack_list.extend([text])
|
||||
else:
|
||||
sticker_pack_list_old = sticker_pack_list
|
||||
send_text = '\n\n'.join(sticker_pack_list_old)
|
||||
await context.client.send_message(context.chat_id, send_text)
|
||||
sticker_pack_list = []
|
||||
await sleep(2)
|
||||
sendtext = '\n\n'.join(sticker_pack_list)
|
||||
await context.client.send_message(context.chat_id, sendtext)
|
||||
await context.delete()
|
||||
|
||||
async def ars_help(context):
|
||||
await context.reply('111')
|
||||
await context.delete()
|
||||
|
||||
def set_state(name, state):
|
||||
file_name = "./plugins/autoreplysticker/config.yml"
|
||||
if exists(file_name):
|
||||
with open(file_name) as f:
|
||||
doc = yaml.safe_load(f)
|
||||
doc[name] = state
|
||||
with open(file_name, 'w') as f:
|
||||
yaml.safe_dump(doc, f, default_flow_style=False)
|
||||
else:
|
||||
dc = {name: state}
|
||||
with open(file_name, 'w', encoding='utf-8') as f:
|
||||
yaml.dump(dc, f)
|
||||
|
||||
def get_name(sender):
|
||||
"""
|
||||
get_name(Message.sender)
|
||||
"""
|
||||
username = sender.username
|
||||
first_name = sender.first_name
|
||||
last_name = sender.last_name
|
||||
_id = sender.id
|
||||
if username == None:
|
||||
if last_name == None:
|
||||
name = f'[{first_name}](tg://user?id={_id})'
|
||||
else:
|
||||
name = f'[{first_name}{last_name}](tg://user?id={_id})'
|
||||
else:
|
||||
name = f'@{username}'
|
||||
return name
|
||||
|
||||
def process_link(chatid, msgid):
|
||||
"""
|
||||
process_link(chat_id, message_id)
|
||||
return https://t.me/c/chat_id/message_id
|
||||
"""
|
||||
if chatid < 0:
|
||||
if chatid < -1000000000000:
|
||||
chatid *= -1
|
||||
chatid -= 1000000000000
|
||||
else:
|
||||
chatid *= -1
|
||||
link = f'https://t.me/c/{chatid}/{msgid}'
|
||||
return link
|
||||
|
||||
@listener(is_plugin=True, outgoing=True, command="ars")
|
||||
async def ars(context):
|
||||
if not exists('./plugins/autoreplysticker'):
|
||||
mkdir('./plugins/autoreplysticker')
|
||||
|
||||
if context.parameter[0] == 'set':
|
||||
if len(context.parameter) < 5:
|
||||
await context.reply('请正确输入 `-ars set` <sticker_id> <sticker_hash> <time> <num>')
|
||||
await context.delete()
|
||||
return
|
||||
try:
|
||||
set_state('sticker_id', context.parameter[1])
|
||||
set_state('sticker_hash', context.parameter[2])
|
||||
set_state('time', context.parameter[3])
|
||||
num_list = []
|
||||
for i in range(4, len(context.parameter)):
|
||||
num_list.append(context.parameter[i])
|
||||
set_state('num', num_list)
|
||||
except:
|
||||
await context.reply('设置失败,请手动设置`./plugins/autoreplysticker/config.yml`')
|
||||
return
|
||||
_noti = await context.reply('设置成功')
|
||||
await sleep(10)
|
||||
await _noti.delete()
|
||||
await ars_check(context)
|
||||
|
||||
elif context.parameter[0] == 'check':
|
||||
await ars_check(context)
|
||||
elif context.parameter[0] == 'getall':
|
||||
await ars_getall(context)
|
||||
elif context.parameter[0] == 'help':
|
||||
await ars_help(context)
|
||||
|
||||
@listener(incoming=True)
|
||||
async def process_message(context):
|
||||
reply_user_id = 0
|
||||
link = process_link(context.chat_id, context.id)
|
||||
me = await context.client.get_me()
|
||||
try:
|
||||
reply = await context.get_reply_message()
|
||||
reply_user_id = reply.sender.id
|
||||
if reply.sticker:
|
||||
return
|
||||
except:
|
||||
pass
|
||||
|
||||
try:
|
||||
config = yaml.load(open(r"./plugins/autoreplysticker/config.yml"), Loader=yaml.FullLoader)
|
||||
_sticker_id = int(config['sticker_id'])
|
||||
_sticker_hash = int(config['sticker_hash'])
|
||||
_num = config['num']
|
||||
_time = int(config['time'])
|
||||
except:
|
||||
return
|
||||
|
||||
if (reply and reply_user_id == me.id):
|
||||
stickers = await context.client(
|
||||
GetStickerSetRequest(
|
||||
stickerset=InputStickerSetID(
|
||||
id=_sticker_id, access_hash=_sticker_hash)))
|
||||
try:
|
||||
i = randint(0, len(_num)-1)
|
||||
sticker = await context.client.send_file(
|
||||
context.chat_id,
|
||||
stickers.documents[int(_num[i])],
|
||||
reply_to=context.id)
|
||||
await sleep(_time)
|
||||
await sticker.delete()
|
||||
if context.chat_id < 0:
|
||||
await log(
|
||||
f'#被回复\n在 [{context.chat.title}]({process_link(context.chat_id, context.id)})\n'
|
||||
f'获得了 {get_name(context.sender)} 的回复')
|
||||
except:
|
||||
pass
|
||||
Loading…
Reference in New Issue
Block a user