This commit is contained in:
jxxghp
2023-07-07 08:16:26 +08:00
parent 99f30b28d1
commit ecec348886
2 changed files with 2 additions and 9 deletions

View File

@ -95,8 +95,7 @@ def update_cookie(
# 更新Cookie # 更新Cookie
state, message = SiteChain().update_cookie(site_info=site_info, state, message = SiteChain().update_cookie(site_info=site_info,
username=username, username=username,
password=password, password=password)
manual=True)
return schemas.Response(success=state, message=message) return schemas.Response(success=state, message=message)

View File

@ -147,14 +147,12 @@ class SiteChain(ChainBase):
self.remote_list() self.remote_list()
def update_cookie(self, site_info: Site, def update_cookie(self, site_info: Site,
username: str, password: str, username: str, password: str) -> Tuple[bool, str]:
manual=False) -> Tuple[bool, str]:
""" """
根据用户名密码更新站点Cookie 根据用户名密码更新站点Cookie
:param site_info: 站点信息 :param site_info: 站点信息
:param username: 用户名 :param username: 用户名
:param password: 密码 :param password: 密码
:param manual: 是否手动更新
:return: (是否成功, 错误信息) :return: (是否成功, 错误信息)
""" """
# 更新站点Cookie # 更新站点Cookie
@ -167,15 +165,11 @@ class SiteChain(ChainBase):
if result: if result:
cookie, ua, msg = result cookie, ua, msg = result
if not cookie: if not cookie:
if manual:
self.message.put(f"站点 {site_info.name} Cookie更新失败{msg}")
return False, msg return False, msg
self.siteoper.update(site_info.id, { self.siteoper.update(site_info.id, {
"cookie": cookie, "cookie": cookie,
"ua": ua "ua": ua
}) })
if manual:
self.message.put(f"站点 {site_info.name} Cookie更新成功")
return True, msg return True, msg
return False, "未知错误" return False, "未知错误"