81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
"""Chrome/Edge 独立 Profile：启用「继续上次会话」，让 Session Cookie 可跨次打开保留。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import logging
|
||
import os
|
||
from typing import Any, Dict
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
RESTORE_CONTINUE_WHERE_LEFT_OFF = 1
|
||
|
||
|
||
def preferences_path(profile_dir: str) -> str:
|
||
return os.path.join(os.path.abspath(profile_dir), "Default", "Preferences")
|
||
|
||
|
||
def ensure_session_restore(profile_dir: str) -> Dict[str, Any]:
|
||
"""启动浏览器前写入 session.restore_on_startup=1；Crashed → Normal。"""
|
||
profile_abs = os.path.abspath(profile_dir)
|
||
prefs = preferences_path(profile_abs)
|
||
os.makedirs(os.path.dirname(prefs), exist_ok=True)
|
||
data: Dict[str, Any] = {}
|
||
existed = os.path.isfile(prefs)
|
||
if existed:
|
||
try:
|
||
with open(prefs, "r", encoding="utf-8") as fh:
|
||
data = json.load(fh)
|
||
except Exception:
|
||
logger.exception("chrome_prefs_read_failed path=%s", prefs)
|
||
data = {}
|
||
|
||
before = int((data.get("session") or {}).get("restore_on_startup") or 0)
|
||
exit_before = str((data.get("profile") or {}).get("exit_type") or "")
|
||
|
||
session = data.setdefault("session", {})
|
||
session["restore_on_startup"] = RESTORE_CONTINUE_WHERE_LEFT_OFF
|
||
|
||
profile = data.setdefault("profile", {})
|
||
if str(profile.get("exit_type") or "").lower() == "crashed":
|
||
profile["exit_type"] = "Normal"
|
||
|
||
try:
|
||
with open(prefs, "w", encoding="utf-8") as fh:
|
||
json.dump(data, fh, ensure_ascii=False, separators=(",", ":"))
|
||
except Exception as exc:
|
||
logger.warning("chrome_prefs_write_failed path=%s err=%s", prefs, exc)
|
||
return {
|
||
"ok": False,
|
||
"path": prefs,
|
||
"existed": existed,
|
||
"restore_before": before,
|
||
"restore_after": before,
|
||
"exit_type_before": exit_before,
|
||
"error": str(exc),
|
||
}
|
||
|
||
after = int((data.get("session") or {}).get("restore_on_startup") or 0)
|
||
exit_after = str((data.get("profile") or {}).get("exit_type") or "")
|
||
logger.info(
|
||
"chrome_session_restore_ensured profile_dir=%s prefs=%s existed=%s "
|
||
"restore_on_startup %s->%s exit_type %s->%s",
|
||
profile_abs,
|
||
prefs,
|
||
existed,
|
||
before,
|
||
after,
|
||
exit_before or "-",
|
||
exit_after or "-",
|
||
)
|
||
return {
|
||
"ok": True,
|
||
"path": prefs,
|
||
"existed": existed,
|
||
"restore_before": before,
|
||
"restore_after": after,
|
||
"exit_type_before": exit_before,
|
||
"exit_type_after": exit_after,
|
||
}
|