Add auto update driver

Signed-off-by: GarfieldGod <1351763110@qq.com>

Author: GarfieldGod

src/core/auto_clock.py

@@ -9,10 +9,13 @@
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 
+from src.utils.const import Key, AppPath
 from src.utils.log import Log
 from src.core.clock import clock
 from src.core.login import login
 from src.core.captcha import captcha, Selectors
+from src.utils.utils import Utils
+
 
 @dataclass
 class Config:
@@ -25,6 +28,7 @@ class Config:
     wait_time: int = 2
     always_retry: bool = False
     show_web_page: bool = True
+    auto_update_driver: bool = True
 
 class AutoClock:
     def __init__(self, config: Config):
@@ -37,12 +41,35 @@ def __init__(self, config: Config):
         self.remote_url = config.remote_url
         self.always_retry = config.always_retry
         self.show_web_page = config.show_web_page
+        self.auto_update_driver = config.auto_update_driver
         self.driver = None
         try:
             self.driver = self.create_driver()
         except Exception as e:
             Log.error(f"Create driver error: {e}")
-            raise Exception(f"Failed to create WebDriver: {e}")
+
+            if 'version' in str(e):
+                try:
+                    if not self.auto_update_driver:
+                        raise Exception("Edge version not support, need to redownload driver.")
+
+                    ok, driver_path = Utils.download_edge_web_driver()
+                    if not ok:
+                        raise Exception("Redownload Edge web driver error.")
+
+                    self.driver_path = driver_path
+                    self.driver = self.create_driver()
+
+                    if self.driver is not None:
+                        data = Utils.read_dict_from_json(AppPath.DataJson)
+                        data[Key.DriverPath] = self.driver_path
+                        Utils.write_dict_to_file(AppPath.DataJson, data)
+
+                except Exception as e:
+                    Log.error(f"Redownload and Create driver error: {e}")
+                    raise Exception(f"Failed to create WebDriver: {e}")
+            else:
+                raise Exception(f"Failed to create WebDriver: {e}")
 
     def create_driver(self):
         # 创建浏览器驱动