nonebot-plugin-rollpig 0.2.0__tar.gz → 0.2.2__tar.gz

{nonebot_plugin_rollpig-0.2.0 → nonebot_plugin_rollpig-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nonebot_plugin_rollpig
-Version: 0.2.0
+Version: 0.2.2
 Summary: 今天是什么小猪
 License: MIT
 License-File: LICENSE

{nonebot_plugin_rollpig-0.2.0 → nonebot_plugin_rollpig-0.2.2}/nonebot_plugin_rollpig/__init__.py

@@ -1,22 +1,18 @@
+import json, random, datetime
+from pathlib import Path
+
 from nonebot import on_command, require
+from nonebot.adapters.onebot.v11 import Event, MessageSegment
+from nonebot.log import logger
 from nonebot.plugin import PluginMetadata
-from nonebot.adapters.onebot.v11 import MessageSegment
 
 # 确保依赖插件先被 NoneBot 注册
-require("nonebot_plugin_apscheduler")
 require("nonebot_plugin_htmlrender")
 require("nonebot_plugin_localstore")
 
-from nonebot_plugin_apscheduler import scheduler
 from nonebot_plugin_htmlrender import template_to_pic
 import nonebot_plugin_localstore as store
 
-from nonebot.log import logger
-import random
-import json
-import datetime
-from pathlib import Path
-
 # 插件配置页
 __plugin_meta__ = PluginMetadata(
     name="今天是什么小猪",
@@ -38,7 +34,7 @@ RES_DIR = PLUGIN_DIR / "resource"
 # 今日记录
 TODAY_PATH = store.get_plugin_data_file("today.json")
 
-cmd = on_command("今天是什么小猪", aliases={"今日小猪", "我是什么小猪"})
+cmd = on_command("今天是什么小猪", aliases={"今日小猪"})
 
 
 def load_json(path, default):
@@ -47,7 +43,6 @@ def load_json(path, default):
         return default
     return json.loads(path.read_text("utf-8"))
 
-
 def save_json(path, data):
     path.write_text(
         json.dumps(data, ensure_ascii=False, indent=2),
@@ -56,7 +51,6 @@ def save_json(path, data):
 
 def find_image_file(pig_id: str) -> Path | None:
     exts = ["png", "jpg", "jpeg", "webp", "gif"]
-
     for ext in exts:
         file = IMAGE_DIR / f"{pig_id}.{ext}"
         if file.exists():
@@ -64,48 +58,46 @@ def find_image_file(pig_id: str) -> Path | None:
     return None
 
 
-# 0 点自动清空
-@scheduler.scheduled_job("cron", hour=0, minute=0)
-def reset_today():
-    if TODAY_PATH.exists():
-        TODAY_PATH.unlink()
-    logger.info("已清空今日记录")
+# 载入小猪信息
+PIG_LIST = load_json(PIGINFO_PATH, [])
+if not PIG_LIST:
+    logger.error("小猪信息为空或不存在，请检查资源文件！")
 
 # 主函数
 @cmd.handle()
-async def _(bot, event):
+async def _(event: Event):
     today_str = datetime.date.today().isoformat()
+    user_id = str(event.user_id)
 
     # 读取今日缓存
-    today_data = load_json(TODAY_PATH, {})
-
-    # 确保当天有字典存储用户数据
-    if today_str not in today_data:
-        today_data[today_str] = {}
-
-    user_id = str(event.user_id)  # 使用 QQ 号作为 key
-
-    # 不重复抽
-    if user_id in today_data[today_str]:
-        pig = today_data[today_str][user_id]
-        await send_rendered_pig(event, pig)
+    today_cache = load_json(TODAY_PATH, {"date": "", "records": {}})
+    
+    # 检查日期，如果不是今天，则清空记录
+    if today_cache.get("date") != today_str:
+        today_cache = {"date": today_str, "records": {}}
+
+    user_records = today_cache["records"]
+
+    # 如果用户今天已经抽过，直接发送结果
+    if user_id in user_records:
+        pig = user_records[user_id]
+        await send_rendered_pig(pig)
         return
 
-    piglist = load_json(PIGINFO_PATH, [])
-    if not piglist:
-        logger.error("pig.json 里没找到小猪信息哦~")
+    if not PIG_LIST:
+        await cmd.finish("小猪信息加载失败，请检查后台报错！")
         return
 
-    # 随机
-    pig = random.choice(piglist)
+    # 随机抽取
+    pig = random.choice(PIG_LIST)
 
     # 保存当天该用户的抽取结果
-    today_data[today_str][user_id] = pig
-    save_json(TODAY_PATH, today_data)
+    user_records[user_id] = pig
+    save_json(TODAY_PATH, today_cache)
 
-    await send_rendered_pig(event, pig)
+    await send_rendered_pig(pig)
 
-async def send_rendered_pig(event, pig_data):
+async def send_rendered_pig(pig_data: dict):
 
     # 使用 id 字段作为图片名
     pig_id = pig_data.get("id", "")
@@ -130,4 +122,4 @@ async def send_rendered_pig(event, pig_data):
         },
     )
 
-    await cmd.finish(MessageSegment.image(pic))
+    await cmd.finish(MessageSegment.image(pic))

{nonebot_plugin_rollpig-0.2.0 → nonebot_plugin_rollpig-0.2.2}/nonebot_plugin_rollpig/resource/pig.json

@@ -238,5 +238,29 @@
         "name": "猪排",
         "description": "建议烤熟食用",
         "analysis": "充满危机的世界里，你深知比起花哨的外表，实实在在恢复的那几格饥饿值才是最靠谱的保障。"
+    },
+    {
+        "id": "pig_god",
+        "name": "智慧小猪之神",
+        "description": "能带来智慧和好运的神奇小猪。",
+        "analysis": "你拥有超凡的智慧和乐于助人的心。虽然看起来可爱呆萌，但你总能在关键时刻给予指引，是猪的守护神，象征着幸运和成功。"
+    },
+    {
+        "id": "tank_pig",
+        "name": "坦克猪",
+        "description": "等待驾驶的重装小猪",
+        "analysis": "你拥有呆萌外表和钢铁之心。平时或许处于“待机”状态，看起来有些慵懒，但只要找对启动方式（按F），你就能化身坚不可摧的堡垒，碾压一切困难。"
+    },
+    {
+        "id": "lao_zhu_li",
+        "name": "老猪历",
+        "description": "资深有经验的猪前辈",
+        "analysis": "你拥有深厚的资历和丰富的经验。在大家眼中，你德高望重，遇事沉稳，总能用积累的智慧指引方向，是值得信赖的可靠前辈。"
+    },
+    {
+        "id": "invisible_pig",
+        "name": "隐形猪",
+        "description": "擅长隐身的神秘小猪",
+        "analysis": "你自带低调气场，擅长隐藏行踪。在人群中你虽不显眼，但总能默默观察一切，像个神秘的旁观者，拥有独特的洞察力。"
     }
 ]

{nonebot_plugin_rollpig-0.2.0 → nonebot_plugin_rollpig-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nonebot_plugin_rollpig"
-version = "0.2.0"
+version = "0.2.2"
 description = "今天是什么小猪"
 authors = ["Bear_lele <lele2587342146@gmail.com>"]
 license = "MIT"