🐛 添加 Cookie 时,显示 「无法获取cookie_name」 的错误提示

This commit is contained in:
2024-10-28 13:17:36 +08:00
parent 438b23a0b9
commit 0985705c22
2 changed files with 15 additions and 5 deletions
+2 -2
View File
@@ -74,7 +74,7 @@ class CookieClientManager(ClientManager):
await config.add_cookie(new_anonymous_cookie)
@classmethod
async def add_user_cookie(cls, content: str):
async def add_user_cookie(cls, content: str, cookie_name=str | None) -> Cookie:
"""添加用户 cookie"""
from ..platform import site_manager
@@ -82,7 +82,7 @@ class CookieClientManager(ClientManager):
if not await cookie_site.validate_cookie(content):
raise ValueError()
cookie = Cookie(site_name=cls._site_name, content=content)
cookie.cookie_name = await cookie_site.get_cookie_name(content)
cookie.cookie_name = cookie_name if cookie_name else await cookie_site.get_cookie_name(content)
cookie.cd = cls._default_cd
cookie_id = await config.add_cookie(cookie)
return await config.get_cookie_by_id(cookie_id)