From 1395c7167871ee10c52481b6672c39f8e13290e8 Mon Sep 17 00:00:00 2001
From: xtaodada <xtao@xtaolink.cn>
Date: Sat, 3 Jul 2021 14:35:45 +0800
Subject: [PATCH] =?UTF-8?q?vip=20=E6=94=AF=E6=8C=81=E9=80=9A=E8=BF=87?=
 =?UTF-8?q?=E5=9B=BE=E7=89=87=E6=9F=A5=E6=89=BE=E7=9B=B8=E4=BC=BC=E5=8A=A8?=
 =?UTF-8?q?=E6=BC=AB=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 list.json |  4 ++--
 vip.py    | 68 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 70 insertions(+), 2 deletions(-)

diff --git a/list.json b/list.json
index 65394c4..67c5c9c 100644
--- a/list.json
+++ b/list.json
@@ -342,13 +342,13 @@
         },
         {
             "name": "vip",
-            "version": "0.31",
+            "version": "0.40",
             "section": "chat",
             "maintainer": "xtaodada",
             "size": "4.0 kb",
             "supported": true,
             "des-short": "vip 捐赠用户功能",
-            "des": "命令：天气：weather Pixiv：pixiv 百度：baidu Duckduckgo：duckduckgo。与 weather 插件冲突"
+            "des": "命令：天气：weather Pixiv：pixiv \n百度：baidu Duckduckgo：duckduckgo\n相似动漫：whatanime。与 weather 插件冲突"
         },
         {
             "name": "calculator",
diff --git a/vip.py b/vip.py
index 01e416d..6be85e4 100644
--- a/vip.py
+++ b/vip.py
@@ -1,5 +1,8 @@
+import io
+
 from requests import get
 from os import remove
+from telethon.tl.types import MessageMediaPhoto
 from pagermaid import bot
 from pagermaid.listener import listener
 from pagermaid.utils import obtain_message, alias_command
@@ -96,3 +99,68 @@ async def pixiv(context):
                 pass
     else:
         await context.edit(pixiv_text)
+
+
+@listener(is_plugin=True, outgoing=True, command=alias_command("whatanime"),
+          description="通过图片查找相似动漫。（需要回复图片）")
+async def whatanime(context):
+    reply = await context.get_reply_message()
+    if reply is not None:
+        if reply.media:
+            if not isinstance(reply.media, MessageMediaPhoto):
+                await context.edit("宁需要回复一张图片")
+                return
+        else:
+            await context.edit("宁需要回复一张图片")
+            return
+    else:
+        await context.edit("宁需要回复一张图片")
+        return
+    await context.edit("获取中。。。")
+    async with bot.conversation('PagerMaid_Modify_bot') as conversation:
+        await conversation.send_message(message='/whatanime_api', file=reply.photo)
+        chat_response = await conversation.get_response()
+        await bot.send_read_acknowledge(conversation.chat_id)
+        whatanime_text = chat_response.text
+    whatanime_list = whatanime_text.split('|||')
+    if len(whatanime_list) == 1:
+        await context.edit(whatanime_text)
+        return
+    link = whatanime_list[0]
+    name = whatanime_list[1]
+    native = whatanime_list[2]
+    episode = whatanime_list[3]
+    alias = whatanime_list[4]
+    r18 = whatanime_list[5]
+    percent = whatanime_list[6]
+    img = whatanime_list[7]
+    video = whatanime_list[8]
+    text = f'[{name}]({link}) （`{native}`）\n'
+    if bool(episode):
+        text += f'\n**集数：** `{episode}`'
+    if bool(alias):
+        text += f'\n**别名：** `{alias}`'
+    if bool(r18):
+        text += f'\n**R-18**'
+    text += f'\n**相似度：** `{percent}`'
+
+    r = get(img)
+    photo = io.BytesIO(r.content)
+    photo.name = f'{name}.png'
+    msg = await context.client.send_file(context.chat_id,
+                                         file=photo,
+                                         caption=text,
+                                         reply_to=reply.id)
+
+    if not video == '':
+        video_list = video.split('||')
+        video = video_list[0]
+        title = video_list[1]
+        start = video_list[2]
+        end = video_list[3]
+        text = f'`{title}`\n\n`{start}` - `{end}`'
+        await context.client.send_file(context.chat_id,
+                                       video,
+                                       caption=text,
+                                       reply_to=msg.id)
+    await context.delete()