🐛 Ensure cookie uid IntStr is int

PaiGramTeam · Jan 10, 2025 · fccacb3 · fccacb3
1 parent f4481ed
commit fccacb3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/modules/apihelper/models/genshin/cookies.py b/modules/apihelper/models/genshin/cookies.py
@@ -1,8 +1,15 @@
-from typing import Optional, TypeVar
+from typing import Optional, Annotated, Union
 
-from pydantic import BaseModel
+from pydantic import BaseModel, BeforeValidator
 
-IntStr = TypeVar("IntStr", int, str)
+
+def int_str_check(value: Union[int, str]) -> Union[int, str]:
+    if value is not None:
+        int(value)
+    return value
+
+
+IntStr = Annotated[Union[int, str], BeforeValidator(int_str_check)]
 
 __all__ = ("CookiesModel",)
 

diff --git a/plugins/account/cookies.py b/plugins/account/cookies.py
@@ -205,6 +205,8 @@ async def parse_cookies(self, update: Update, context: CallbackContext, text: st
             )
             cookie = {x[0]: x[1] for x in wrapped}
             cookies = self.parse_cookie(cookie)
+            if cookies:
+                CookiesModel.model_validate(cookies)
         except (AttributeError, ValueError, IndexError) as exc:
             logger.info("用户 %s[%s] Cookies解析出现错误\ntext:%s", user.full_name, user.id, message.text)
             logger.debug("解析Cookies出现错误", exc_info=exc)