FIX: Adding _botcity_temp_dir in all browsers

Author: kayqueGovetri

botcity/web/bot.py

@@ -47,26 +47,11 @@
 
 def _cleanup(bot: ReferenceType[WebBot]):
 
-    def get_botcity_temp_dir(opt, browser_name):
-        browsers = {
-            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
-        }
-        return browsers.get(browser_name, None)
-
     if bot() is not None:
         try:
             options = bot().options
-            name = bot().browser
             bot().stop_browser()
-            if not bot():
-                return None
-            temp_dir = get_botcity_temp_dir(options, name)
-            if not temp_dir:
-                return None
+            temp_dir = options._botcity_temp_dir
             shutil.rmtree(temp_dir, ignore_errors=True)
         except Exception:
             pass

botcity/web/browsers/chrome.py

@@ -70,7 +70,6 @@ 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

botcity/web/browsers/edge.py

@@ -67,7 +67,7 @@ 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

botcity/web/browsers/firefox.py

@@ -362,7 +362,7 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
     if not user_data_dir:
         temp_dir = tempfile.TemporaryDirectory(prefix="botcity_")
         user_data_dir = temp_dir.name
-        atexit.register(cleanup_temp_dir, temp_dir)
+        firefox_options._botcity_temp_dir = user_data_dir
     if binary_path:
         firefox_options.binary_location = str(binary_path)
     firefox_options.set_preference("profile", user_data_dir)

botcity/web/browsers/undetected_chrome.py

@@ -77,7 +77,7 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
     if not user_data_dir:
         temp_dir = tempfile.TemporaryDirectory(prefix="botcity_")
         user_data_dir = temp_dir.name
-        atexit.register(cleanup_temp_dir, temp_dir)
+        chrome_options._botcity_temp_dir = user_data_dir
 
     chrome_options.add_argument(f"--user-data-dir={user_data_dir}")