自动回复Sticker v1.0.0

autoreplysticker.py

@@ -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