From 115efb38762039540a5dc31af6c47ac5d7e5974c Mon Sep 17 00:00:00 2001
From: Niko <1377382065@qq.com>
Date: Fri, 22 May 2026 10:52:52 +0800
Subject: [PATCH] fix bugs

---
 src/main/java/com/ets/scraper/EtsScraper.java | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/ets/scraper/EtsScraper.java b/src/main/java/com/ets/scraper/EtsScraper.java
index 7eea214..8a4d6b8 100644
--- a/src/main/java/com/ets/scraper/EtsScraper.java
+++ b/src/main/java/com/ets/scraper/EtsScraper.java
@@ -178,10 +178,11 @@ public class EtsScraper {
 
                  // Download captcha image
                 downloadCaptcha(page);
+                screenshot(page, "2_after_captcha", downloadDir);
 
                  // Close dialog again after page reload
                 closeNotificationDialog(page);
-                screenshot(page, "2_after_close_dialog", downloadDir);
+                screenshot(page, "3_after_close_dialog", downloadDir);
 
                  // Recognize captcha and perform login
                 boolean loggedin = doLoginWithCaptcha(page);
@@ -344,7 +345,6 @@ public class EtsScraper {
                 Path captchaPath = SCREENSHOT_DIR.resolve("captcha.png");
                 java.nio.file.Files.write(captchaPath, body);
                 System.out.println("[+] Captcha saved to: " + captchaPath);
-                screenshot(page, "3_after_captcha", downloadDir);
                 System.out.println("[+] Captcha size: " + body.length + " bytes");
             }
         } catch (Exception e) {
@@ -393,7 +393,6 @@ public class EtsScraper {
             System.err.println("[-] Screenshot failed: " + e.getMessage());
           }
       }
-    }
 
     public static void sleep(long ms) {
         try {