fix(save_auth): 登录检测改为探测页实测会话有效性
不再读取用户标签页的 URL 判定——SPA hash 跳转后脚本读到的 可能是旧值, 登录界面也可能在其它标签页, 两种情况都会把成功 登录误判为失败。 改为: 回车后在同一 context 新开探测页访问首页, 由应用自己 回答是否已登录(有会话落 #/home, 无会话被重定向回 #/login), 与测试用例 ensure_login 的判定方式一致; 探测完即关, 不动 用户标签页。路由判断用精确 hash/path 解析, 参数含 login 字样不再误报。 验证: 用用户误报场景下保存的登录态实测, 访问直达 #/home, 证实会话判定有效。
This commit is contained in:
@@ -69,6 +69,22 @@ def _fill_credentials(page, account: dict) -> bool:
|
|||||||
return filled_user and filled_pwd
|
return filled_user and filled_pwd
|
||||||
|
|
||||||
|
|
||||||
|
def _is_on_login_page(url: str) -> bool:
|
||||||
|
"""精确判断是否仍在登录页:只看路由本身(hash 或 path)。
|
||||||
|
|
||||||
|
不做子串匹配——登录成功后的 URL 可能带 login 字样的参数
|
||||||
|
(如 ?from=login、#/loginLog 菜单),子串判断会把成功登录误判为失败。
|
||||||
|
"""
|
||||||
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
|
parsed = urlparse(url)
|
||||||
|
frag = (parsed.fragment or "").lstrip("/") # #/login → "login"
|
||||||
|
path = (parsed.path or "").rstrip("/") # /admin/login → "/admin/login"
|
||||||
|
frag_is_login = frag == "login" or frag.startswith("login/") or frag.startswith("login?")
|
||||||
|
path_is_login = path == "/login" or path.endswith("/login")
|
||||||
|
return frag_is_login or path_is_login
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
parser = argparse.ArgumentParser(
|
parser = argparse.ArgumentParser(
|
||||||
description="手动登录并保存浏览器登录态(绕过图形验证码)"
|
description="手动登录并保存浏览器登录态(绕过图形验证码)"
|
||||||
@@ -118,11 +134,23 @@ def main():
|
|||||||
|
|
||||||
input(">>> 按 Enter 保存登录态...")
|
input(">>> 按 Enter 保存登录态...")
|
||||||
|
|
||||||
# 检查是否真的登录成功了
|
# 不依赖用户标签页的 URL(SPA hash 跳转后读到的可能是旧值, 登录
|
||||||
current = page.url
|
# 界面也可能在别的标签页)——在 context 里新开探测页访问首页,
|
||||||
if "/login" in current or "/#/login" in current:
|
# 由应用自己回答"是否已登录": 有会话落在 #/home, 无会话被重定向
|
||||||
print(f"\n[警告] 当前仍在登录页: {current}")
|
# 回 #/login。探测完成即关闭, 不动用户的标签页。
|
||||||
print(" 要保存吗?(y/n): ", end="")
|
# (与测试用例 ensure_login 的判定方式一致)
|
||||||
|
print("[检测] 验证登录状态...")
|
||||||
|
probe = context.new_page()
|
||||||
|
try:
|
||||||
|
probe.goto(base_url, wait_until="domcontentloaded")
|
||||||
|
probe.wait_for_timeout(3000) # SPA hash 路由重定向的 settle
|
||||||
|
current = probe.url
|
||||||
|
finally:
|
||||||
|
probe.close()
|
||||||
|
|
||||||
|
if _is_on_login_page(current):
|
||||||
|
print(f"\n[警告] 登录态无效,访问首页被重定向回登录页: {current}")
|
||||||
|
print(" 登录可能未成功(验证码错误/未点击登录)。仍要保存吗?(y/n): ", end="")
|
||||||
choice = input().strip().lower()
|
choice = input().strip().lower()
|
||||||
if choice != "y":
|
if choice != "y":
|
||||||
print("已取消。")
|
print("已取消。")
|
||||||
|
|||||||
Reference in New Issue
Block a user