suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-04 02:26:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✏️ 改改变量名

Browse Source
This commit is contained in:
suyiiyii 2024-10-21 20:43:08 +08:00
parent f5d30b998c
commit 8d32145ea8
Signed by: suyiiyii
GPG Key ID: 044704CB29B8AD85
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
nonebot_bison/utils/site.py
View File

@ -49,15 +49,13 @@ class CookieClientManager(ClientManager):
@classmethod @classmethod
async def refresh_anonymous_cookie(cls): async def refresh_anonymous_cookie(cls):
"""移除已有的匿名cookie，添加一个新的匿名cookie""" """移除已有的匿名cookie，添加一个新的匿名cookie"""
anonymous_cookies = await config.get_cookie(cls._site_name, is_anonymous=True) existing_anonymous_cookies = await config.get_cookie(cls._site_name, is_anonymous=True)
anonymous_cookie = Cookie(site_name=cls._site_name, content="{}", is_universal=True, is_anonymous=True) new_anonymous_cookie = Cookie(site_name=cls._site_name, content="{}", is_universal=True, is_anonymous=True)
for cookie in anonymous_cookies: for cookie in existing_anonymous_cookies:
if not cookie.is_anonymous:
continue
await config.delete_cookie_by_id(cookie.id) await config.delete_cookie_by_id(cookie.id)
anonymous_cookie.id = cookie.id # 保持原有的id new_anonymous_cookie.id = cookie.id # 保持原有的id
anonymous_cookie.last_usage = datetime.now() # 使得第一次请求优先使用用户 cookie new_anonymous_cookie.last_usage = datetime.now() # 使得第一次请求优先使用用户 cookie
await config.add_cookie(anonymous_cookie) await config.add_cookie(new_anonymous_cookie)
@classmethod @classmethod
async def add_user_cookie(cls, content: str): async def add_user_cookie(cls, content: str):
@ -81,7 +79,7 @@ class CookieClientManager(ClientManager):
logger.trace(f"请求成功: {cookie.id} {resp.request.url}") logger.trace(f"请求成功: {cookie.id} {resp.request.url}")
cookie.status = "success" cookie.status = "success"
else: else:
logger.warning(f"请求失败:{cookie.id} {resp.request.url}, 状态码: {resp.status_code}") logger.warning(f"请求失败: {cookie.id} {resp.request.url}, 状态码: {resp.status_code}")
cookie.status = "failed" cookie.status = "failed"
cookie.last_usage = datetime.now() cookie.last_usage = datetime.now()
await config.update_cookie(cookie) await config.update_cookie(cookie)