ENH: Insert property in options class edge and chrome

Author: kayqueGovetri

botcity/web/bot.py

@@ -49,26 +49,18 @@ def _cleanup(bot: ReferenceType[WebBot]):
 
     def get_botcity_temp_dir(opt, browser_name):
         browsers = {
-            Browser.CHROME: ("--botcity-temp-dir=", opt.arguments if opt else []),
-            Browser.EDGE: ("--botcity-temp-dir=", opt.arguments if opt else []),
-            Browser.UNDETECTED_CHROME: ("--botcity-temp-dir=", opt.arguments if opt else []),
+            Browser.CHROME: opt._botcity_temp_dir if opt else None,
+            Browser.EDGE: opt._botcity_temp_dir if opt else None,
+            Browser.UNDETECTED_CHROME: opt._botcity_temp_dir if opt else None,
             Browser.FIREFOX: None,
             Browser.IE: None
         }
-        items = browsers.get(browser_name, None)
-        if not items:
-            return None
-
-        result = next(filter(lambda x: x.startswith(items[0]), items[1]), None)
-        if not result:
-            return None
+        response = browsers.get(browser_name, None)
 
-        match_result = re.search(rf'{items[0]}([^ ]+)', result)
-
-        if not match_result:
+        if not response:
             return None
 
-        return match_result.group(1)
+        return response
 
     if bot() is not None:
         try:

botcity/web/browsers/chrome.py

@@ -70,11 +70,11 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
             chrome_options.add_argument("--no-sandbox")
     except AttributeError:
         pass
-
+    chrome_options._botcity_temp_dir = None
     if not user_data_dir:
         temp_dir = tempfile.TemporaryDirectory(prefix="botcity_")
         user_data_dir = temp_dir.name
-        chrome_options.add_argument(f"--botcity-temp-dir={user_data_dir}")
+        chrome_options._botcity_temp_dir = user_data_dir
 
     chrome_options.add_argument(f"--user-data-dir={user_data_dir}")
 

botcity/web/browsers/edge.py

@@ -67,11 +67,11 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
             edge_options.add_argument("--no-sandbox")
     except AttributeError:
         pass
-
+    edge_options._botcity_temp_dir = None
     if not user_data_dir:
         temp_dir = tempfile.TemporaryDirectory(prefix="botcity_")
         user_data_dir = temp_dir.name
-        edge_options.add_argument(f"--botcity-temp-dir={user_data_dir}")
+        edge_options._botcity_temp_dir = user_data_dir
 
     edge_options.add_argument(f"--user-data-dir={user_data_dir}")