perf👌: 道具使用与签到

zhenxun/builtin_plugins/sign_in/_data_source.py

@@ -19,6 +19,7 @@ from zhenxun.utils.image_utils import BuildImage, ImageTemplate
 from zhenxun.utils.utils import get_user_avatar
 
 from ._random_event import random_event
+from .goods_register import driver
 from .utils import SIGN_TODAY_CARD_PATH, get_card
 
 ICON_PATH = IMAGE_PATH / "_icon"

zhenxun/builtin_plugins/sign_in/goods_register.py

@@ -12,59 +12,62 @@ from zhenxun.utils.decorator.shop import NotMeetUseConditionsException, shop_reg
 driver: Driver = nonebot.get_driver()
 
 
-@driver.on_startup
-async def _():
-    """
-    导入内置的三个商品
-    """
+# @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):
-        if session.id1:
-            user_console = await UserConsole.get_user(session.id1, 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(
+    name=("好感度双倍加持卡Ⅰ", "好感度双倍加持卡Ⅱ", "好感度双倍加持卡Ⅲ"),
+    price=(30, 150, 250),
+    des=(
+        "下次签到双倍好感度概率 + 10%（谁才是真命天子？）（同类商品将覆盖）",
+        "下次签到双倍好感度概率 + 20%（平平庸庸）（同类商品将覆盖）",
+        "下次签到双倍好感度概率 + 30%（金币才是真命天子！）（同类商品将覆盖）",
+    ),
+    load_status=True,
+    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):
+    if session.id1:
+        user_console = await UserConsole.get_user(session.id1, 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.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(
+    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.after_handle(name="测试道具A", load_status=False)
-    async def _(user_id: int, group_id: int):
-        print(user_id, group_id, "第一个使用后函数（after 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）")
+
+
+@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）")