update 0.0.6.2

plugins/pix_gallery/pix_show_info.py

@@ -1,84 +1,84 @@
-from nonebot import on_command
-from utils.utils import get_message_text
-from utils.message_builder import image
-from nonebot.adapters.cqhttp import Bot, MessageEvent
-from nonebot.typing import T_State
-from .data_source import gen_keyword_pic, get_keyword_num
-from .model.pixiv_keyword_user import PixivKeywordUser
-import asyncio
-
-
-__zx_plugin_name__ = "查看pix图库"
-__plugin_usage__ = """
-usage：
-    查看pix图库
-    指令：
-        查看pix图库 ?[tags]: 查看指定tag图片数量，为空时查看整个图库
-""".strip()
-__plugin_des__ = "让我看看管理员私藏了多少货"
-__plugin_cmd__ = ["查看pix图库 ?[tags]"]
-__plugin_version__ = 0.1
-__plugin_author__ = "HibiKier"
-__plugin_settings__ = {
-    "level": 5,
-    "default_status": True,
-    "limit_superuser": False,
-    "cmd": ["查看pix图库"],
-}
-
-
-my_keyword = on_command("我的pix关键词", aliases={"我的pix关键字"}, priority=1, block=True)
-
-show_keyword = on_command("显示pix关键词", aliases={"显示pix关键字"}, priority=1, block=True)
-
-show_pix = on_command("查看pix图库", priority=1, block=True)
-
-
-@my_keyword.handle()
-async def _(bot: Bot, event: MessageEvent, state: T_State):
-    data = await PixivKeywordUser.get_all_user_dict()
-    if data.get(event.user_id) is None or not data[event.user_id]["keyword"]:
-        await my_keyword.finish("您目前没有提供任何Pixiv搜图关键字...", at_sender=True)
-    await my_keyword.send(
-        f"您目前提供的如下关键字：\n\t" + "，".join(data[event.user_id]["keyword"])
-    )
-
-
-@show_keyword.handle()
-async def _(bot: Bot, event: MessageEvent, state: T_State):
-    _pass_keyword, not_pass_keyword = await PixivKeywordUser.get_current_keyword()
-    if _pass_keyword or not_pass_keyword:
-        await show_keyword.send(
-            image(
-                b64=await asyncio.get_event_loop().run_in_executor(
-                    None,
-                    gen_keyword_pic,
-                    _pass_keyword,
-                    not_pass_keyword,
-                    str(event.user_id) in bot.config.superusers,
-                )
-            )
-        )
-    else:
-        if str(event.user_id) in bot.config.superusers:
-            await show_keyword.finish(f"目前没有已收录或待收录的搜索关键词...")
-        else:
-            await show_keyword.finish(f"目前没有已收录的搜索关键词...")
-
-
-@show_pix.handle()
-async def _(bot: Bot, event: MessageEvent, state: T_State):
-    keyword = get_message_text(event.json())
-    count, r18_count, count_, setu_count, r18_count_ = await get_keyword_num(keyword)
-    await show_pix.send(
-        f"PIX图库：{keyword}\n"
-        f"总数：{count + r18_count}\n"
-        f"美图：{count}\n"
-        f"R18：{r18_count}\n"
-        f"---------------\n"
-        f"Omega图库：{keyword}\n"
-        f"总数：{count_ + setu_count + r18_count_}\n"
-        f"美图：{count_}\n"
-        f"色图：{setu_count}\n"
-        f"R18：{r18_count_}"
-    )
+from nonebot import on_command
+from utils.utils import get_message_text
+from utils.message_builder import image
+from nonebot.adapters.cqhttp import Bot, MessageEvent
+from nonebot.typing import T_State
+from .data_source import gen_keyword_pic, get_keyword_num
+from .model.pixiv_keyword_user import PixivKeywordUser
+import asyncio
+
+
+__zx_plugin_name__ = "查看pix图库"
+__plugin_usage__ = """
+usage：
+    查看pix图库
+    指令：
+        查看pix图库 ?[tags]: 查看指定tag图片数量，为空时查看整个图库
+""".strip()
+__plugin_des__ = "让我看看管理员私藏了多少货"
+__plugin_cmd__ = ["查看pix图库 ?[tags]"]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+__plugin_settings__ = {
+    "level": 5,
+    "default_status": True,
+    "limit_superuser": False,
+    "cmd": ["查看pix图库"],
+}
+
+
+my_keyword = on_command("我的pix关键词", aliases={"我的pix关键字"}, priority=1, block=True)
+
+show_keyword = on_command("显示pix关键词", aliases={"显示pix关键字"}, priority=1, block=True)
+
+show_pix = on_command("查看pix图库", priority=1, block=True)
+
+
+@my_keyword.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    data = await PixivKeywordUser.get_all_user_dict()
+    if data.get(event.user_id) is None or not data[event.user_id]["keyword"]:
+        await my_keyword.finish("您目前没有提供任何Pixiv搜图关键字...", at_sender=True)
+    await my_keyword.send(
+        f"您目前提供的如下关键字：\n\t" + "，".join(data[event.user_id]["keyword"])
+    )
+
+
+@show_keyword.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    _pass_keyword, not_pass_keyword = await PixivKeywordUser.get_current_keyword()
+    if _pass_keyword or not_pass_keyword:
+        await show_keyword.send(
+            image(
+                b64=await asyncio.get_event_loop().run_in_executor(
+                    None,
+                    gen_keyword_pic,
+                    _pass_keyword,
+                    not_pass_keyword,
+                    str(event.user_id) in bot.config.superusers,
+                )
+            )
+        )
+    else:
+        if str(event.user_id) in bot.config.superusers:
+            await show_keyword.finish(f"目前没有已收录或待收录的搜索关键词...")
+        else:
+            await show_keyword.finish(f"目前没有已收录的搜索关键词...")
+
+
+@show_pix.handle()
+async def _(bot: Bot, event: MessageEvent, state: T_State):
+    keyword = get_message_text(event.json())
+    count, r18_count, count_, setu_count, r18_count_ = await get_keyword_num(keyword)
+    await show_pix.send(
+        f"PIX图库：{keyword}\n"
+        f"总数：{count + r18_count}\n"
+        f"美图：{count}\n"
+        f"R18：{r18_count}\n"
+        f"---------------\n"
+        f"Omega图库：{keyword}\n"
+        f"总数：{count_ + setu_count + r18_count_}\n"
+        f"美图：{count_}\n"
+        f"色图：{setu_count}\n"
+        f"R18：{r18_count_}"
+    )