64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
"""登录助手 — 解析浏览器导出的 cookies JSON，保存为 session.json。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
|
|
def get_session_path() -> str:
|
|
if getattr(sys, "frozen", False):
|
|
base = os.path.dirname(sys.executable)
|
|
else:
|
|
base = os.path.dirname(os.path.abspath(__file__))
|
|
return os.path.join(base, "session.json")
|
|
|
|
|
|
def parse_json_and_save(json_text: str) -> int:
|
|
"""解析浏览器导出的 cookies JSON 数组，提取目标域名的 cookie 并保存。
|
|
返回保存的 cookie 数量。
|
|
"""
|
|
raw = json.loads(json_text)
|
|
if not isinstance(raw, list):
|
|
raise ValueError("JSON 必须是数组格式")
|
|
|
|
# 过滤目标域名
|
|
target = "10.4.188.1"
|
|
cookies = []
|
|
seen = set()
|
|
for c in raw:
|
|
domain = c.get("domain", "")
|
|
if target not in domain:
|
|
continue
|
|
name = c.get("name", "").strip()
|
|
path = c.get("path", "/")
|
|
if not name:
|
|
continue
|
|
key = (name, domain, path)
|
|
if key in seen:
|
|
continue
|
|
seen.add(key)
|
|
cookies.append({
|
|
"name": name,
|
|
"value": c.get("value", ""),
|
|
"domain": domain,
|
|
"path": path,
|
|
})
|
|
|
|
data = {"cookies": cookies, "saved_from": f"https://{target}"}
|
|
session_path = get_session_path()
|
|
os.makedirs(os.path.dirname(session_path) or ".", exist_ok=True)
|
|
with open(session_path, "w", encoding="utf-8") as f:
|
|
json.dump(data, f, ensure_ascii=False, indent=2)
|
|
return len(cookies)
|
|
|
|
|
|
def load_session_info() -> dict | None:
|
|
"""读取已保存的 session 信息用于 UI 展示。"""
|
|
try:
|
|
with open(get_session_path(), "r") as f:
|
|
return json.load(f)
|
|
except (FileNotFoundError, json.JSONDecodeError):
|
|
return None
|