FIX: Ajust in import edge options and start driver

Author: kayqueGovetri

botcity/web/bot.py

@@ -254,8 +254,10 @@ def check_driver():
         self.capabilities = cap
         driver_path = self.driver_path or check_driver()
         self.driver_path = driver_path
-
-        self._driver = driver_class(options=opt, desired_capabilities=cap, executable_path=driver_path)
+        if BROWSER_CONFIGS.get(self.browser).get("driver") == "msedgedriver":
+            self._driver = driver_class(options=opt, capabilities=cap, executable_path=driver_path)
+        else:
+            self._driver = driver_class(options=opt, desired_capabilities=cap, executable_path=driver_path)
         self.set_screen_resolution()
 
     def stop_browser(self):

botcity/web/browsers/edge.py

@@ -5,14 +5,16 @@
 import time
 from typing import Dict
 
-from msedge.selenium_tools import Edge, EdgeOptions  # noqa: F401, F403
+# from msedge.selenium_tools import Edge, EdgeOptions  # noqa: F401, F403
 from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
+from selenium.webdriver.edge.options import Options
+from selenium.webdriver import Edge # noqa: F401, F403
 
 from ..util import cleanup_temp_dir
 
 
 def default_options(headless=False, download_folder_path=None, user_data_dir=None,
-                    page_load_strategy="normal") -> EdgeOptions:
+                    page_load_strategy="normal") -> Options:
     """Retrieve the default options for this browser curated by BotCity.
 
     Args:
@@ -26,7 +28,7 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
     Returns:
         EdgeOptions: The Edge options.
     """
-    edge_options = EdgeOptions()
+    edge_options = Options()
     try:
         page_load_strategy = page_load_strategy.value
     except AttributeError:

requirements.txt

@@ -1,6 +1,5 @@
 botcity-framework-base>=0.2.2
 beautifulsoup4
-msedge-selenium-tools==3.141.4
 numpy
 opencv-python
 pillow