feat✨: 更新内置插件

2026-10-21 04:00:04 +08:00 · 2024-02-25 03:18:34 +08:00
parent 131200a28e
commit eb0572ea77
83 changed files with 7588 additions and 450 deletions
@@ -0,0 +1,75 @@
+from decimal import Decimal
+
+import nonebot
+from nonebot.drivers import Driver
+from nonebot_plugin_session import EventSession
+
+from zhenxun.configs.config import Config
+from zhenxun.models.sign_user import SignUser
+from zhenxun.models.user_console import UserConsole
+from zhenxun.utils.decorator.shop import NotMeetUseConditionsException, shop_register
+
+driver: Driver = nonebot.get_driver()
+
+
+@driver.on_startup
+async def _():
+    """
+    导入内置的三个商品
+    """
+
+    @shop_register(
+        name=("好感度双倍加持卡Ⅰ", "好感度双倍加持卡Ⅱ", "好感度双倍加持卡Ⅲ"),
+        price=(30, 150, 250),
+        des=(
+            "下次签到双倍好感度概率 + 10%（谁才是真命天子？）（同类商品将覆盖）",
+            "下次签到双倍好感度概率 + 20%（平平庸庸）（同类商品将覆盖）",
+            "下次签到双倍好感度概率 + 30%（金币才是真命天子！）（同类商品将覆盖）",
+        ),
+        load_status=bool(Config.get_config("shop", "IMPORT_DEFAULT_SHOP_GOODS")),
+        icon=(
+            "favorability_card_1.png",
+            "favorability_card_2.png",
+            "favorability_card_3.png",
+        ),
+        **{"好感度双倍加持卡Ⅰ_prob": 0.1, "好感度双倍加持卡Ⅱ_prob": 0.2, "好感度双倍加持卡Ⅲ_prob": 0.3},  # type: ignore
+    )
+    async def _(session: EventSession, user_id: int, group_id: int, prob: float):
+        user_console, _ = await UserConsole.get_or_create(
+            user_id=session.id1,
+            defaults={
+                "uid": await UserConsole.get_new_uid(),
+                "platform": session.platform,
+            },
+        )
+        user, _ = await SignUser.get_or_create(
+            user_id=user_id,
+            defaults={"platform": session.platform, "user_console": user_console},
+        )
+        user.add_probability = Decimal(prob)
+        await user.save(update_fields=["add_probability"])
+
+    @shop_register(
+        name="测试道具A",
+        price=99,
+        des="随便侧而出",
+        load_status=False,
+        icon="sword.png",
+    )
+    async def _(user_id: int, group_id: int):
+        print(user_id, group_id, "使用测试道具")
+
+    @shop_register.before_handle(name="测试道具A", load_status=False)
+    async def _(user_id: int, group_id: int):
+        print(user_id, group_id, "第一个使用前函数（before handle）")
+
+    @shop_register.before_handle(name="测试道具A", load_status=False)
+    async def _(user_id: int, group_id: int):
+        print(user_id, group_id, "第二个使用前函数（before handle）222")
+        raise NotMeetUseConditionsException(
+            "太笨了！"
+        )  # 抛出异常，阻断使用，并返回信息
+
+    @shop_register.after_handle(name="测试道具A", load_status=False)
+    async def _(user_id: int, group_id: int):
+        print(user_id, group_id, "第一个使用后函数（after handle）")