优化

src/invoices.py

@@ -28,7 +28,12 @@ page.set.download_path(download_path)
 # 忽略 openpyxl 样式警告
 warnings.filterwarnings("ignore", category=UserWarning, module="openpyxl")
 
-while True:
+def page_get(url):
+    host = domain.switch_domain(country)
+    full_url = host + url
+    page.get(full_url, timeout=3)
+
+    while True:
         time.sleep(1)
         title = page.title
         if title == "Amazon Sign-In":

src/returns.py

@@ -22,7 +22,12 @@ helper.make_dir(download_path)
 # 设置下载路径，确保在打开浏览器前设置
 page.set.download_path(download_path)
 
-while True:
+def page_get(url):
+    host = domain.switch_domain(country)
+    full_url = host + url
+    page.get(full_url, timeout=3)
+
+    while True:
         time.sleep(1)
         title = page.title
         if title == "Amazon Sign-In":