From 1471cc5d4183b43dc26878ec36220e87a793d58b Mon Sep 17 00:00:00 2001
From: joey0629 <jjjoey020629@gmail.com>
Date: Tue, 23 Apr 2024 18:52:27 +0800
Subject: [PATCH] Update all headless mode v3

---
 main.py | 36 ++++++++++++++++++++++--------------
 1 file changed, 22 insertions(+), 14 deletions(-)

diff --git a/main.py b/main.py
index 022f14a..2f8fda7 100644
--- a/main.py
+++ b/main.py
@@ -10,7 +10,7 @@ import requests
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
 from webdriver_manager.chrome import ChromeDriverManager
-
+from selenium.common.exceptions import WebDriverException
 
 
 BOT_TOKEN = '6701395239:AAFE30dqvNihDdni9vYoAbWssO-X5yAmwho'
@@ -85,14 +85,20 @@ def find_pce(date:str):
     options.add_argument(
         'user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3')
     driver = webdriver.Chrome( options=options)
-    driver.get(pce_url+date)
-    print("Open")
-    time.sleep(0.5)
-    date = driver.find_element(By.XPATH, '//*[@id="home"]/h1').text.split(' ')[4]
-    value_1 = driver.find_element(By.XPATH, '//*[@id="home"]/div[2]/div/div/p[2]').text
-    value_2 = driver.find_element(By.XPATH, '//*[@id="home"]/div[2]/div/div/div[1]/table/tbody/tr[13]').text.strip(' ').split(' ')[-1]
-    driver.quit()
-    return date , value_1 , value_2
+
+    try:
+        print("Successfully accessed the website.")
+        driver.get(pce_url+date)
+        time.sleep(0.5)
+        date = driver.find_element(By.XPATH, '//*[@id="home"]/h1').text.split(' ')[4]
+        value_1 = driver.find_element(By.XPATH, '//*[@id="home"]/div[2]/div/div/p[2]').text
+        value_2 = driver.find_element(By.XPATH, '//*[@id="home"]/div[2]/div/div/div[1]/table/tbody/tr[13]').text.strip(' ').split(' ')[-1]
+        return date , value_1 , value_2
+    except WebDriverException:
+        print("Failed to access the website.")
+    finally:
+        driver.quit()
+        return None
 #非農就業人數
 def find_non_farm(url):
     from selenium import webdriver
@@ -199,7 +205,9 @@ if __name__ == "__main__":
     # broadcast_all_pce("February", "february-2024")
 
     print("Start Time:" , datetime.fromtimestamp(time.time()))
+    '''
     schedule.every().day.at("10:44").do(wrapper_function_pce, "February", "february-2024")
+    
     while True:
         if datetime.now().strftime("%Y-%m-%d %H:%M") == "2024-04-23 10:44":
             schedule.run_pending()
@@ -207,7 +215,7 @@ if __name__ == "__main__":
             print("Broadcast completed")
             break
         time.sleep(0.1)  # Check every 0.1 seconds
-
+    '''
     #NonFarm
     # date , message = find_non_farm(nonfarm_url)
     # print(date)
@@ -217,7 +225,7 @@ if __name__ == "__main__":
 
     #PCE
     #broadcast_all_pce("February", "february-2024")
-    # date, message1, message2 = find_pce("february-2024")
-    # print(date)
-    # print(message1)
-    # print(message2)
+    date, message1, message2 = find_pce("february-2024")
+    print(date)
+    print(message1)
+    print(message2)