From 82fb61ec26506ceec4e00864e650401410a358fd Mon Sep 17 00:00:00 2001
From: kinoshitakenta <boxuan@techmation.com.tw>
Date: Thu, 3 Jul 2025 17:06:25 +0800
Subject: [PATCH] fix: prompt re-login on expired Selenium session to prevent
 crash

---
 main.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/main.py b/main.py
index 621d610..bc6d863 100644
--- a/main.py
+++ b/main.py
@@ -9,6 +9,7 @@ import logging
 import os
 from pathlib import Path
 
+from selenium.common.exceptions import InvalidSessionIdException
 from utils.cli import cli
 from utils.driver import get_driver
 from utils.EIP_action import ActionType, Action
@@ -98,6 +99,9 @@ def main(opt):
             if keep_login_status(driver, login_info):
                 action_agent.run(action_code)
 
+    except InvalidSessionIdException:
+        print("Session has expired, please re-login.")
+
     except KeyboardInterrupt:
         pass