feat✨: 添加hook，更新其他

zhenxun/builtin_plugins/sign_in/_data_source.py

@@ -94,13 +94,7 @@ class SignManage:
         if not session.id1:
             return None
         now = datetime.now(pytz.timezone("Asia/Shanghai"))
-        user_console, _ = await UserConsole.get_or_create(
-            user_id=session.id1,
-            defaults={
-                "uid": await UserConsole.get_new_uid(),
-                "platform": session.platform,
-            },
-        )
+        user_console = await UserConsole.get_user(session.id1, session.platform)
         user, _ = await SignUser.get_or_create(
             user_id=session.id1,
             defaults={"user_console": user_console, "platform": session.platform},
@@ -112,7 +106,6 @@ class SignManage:
             or (new_log and now > new_log.create_time)
             or file_name in os.listdir(SIGN_TODAY_CARD_PATH)
         ):
-            user_console, _ = await UserConsole.get_or_create(user_id=session.id1)
             path = await get_card(user, nickname, -1, user_console.gold, "")
         else:
             path = await cls._handle_sign_in(user, nickname, session, is_view_card)

zhenxun/builtin_plugins/sign_in/goods_register.py

@@ -35,19 +35,14 @@ async def _():
         **{"好感度双倍加持卡Ⅰ_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"])
+        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",