FIX: Ajust to working in only bot temp dir

Author: kayqueGovetri

botcity/web/bot.py

@@ -44,14 +44,12 @@
 
 logger = logging.getLogger(__name__)
 
-
 def _cleanup(bot: ReferenceType[WebBot]):
 
     if bot() is not None:
         try:
-            options = bot().options
             bot().stop_browser()
-            temp_dir = options._botcity_temp_dir
+            temp_dir = bot()._botcity_temp_dir
             if not temp_dir: return None
             shutil.rmtree(temp_dir, ignore_errors=True)
         except Exception:
@@ -96,6 +94,7 @@ def __init__(self, headless=False):
         self._shift_hold = False
 
         self._download_folder_path = os.getcwd()
+        self._botcity_temp_dir = None
 
         atexit.register(_cleanup, ref(self))
 
@@ -372,6 +371,11 @@ def stop_browser(self):
             self.activate_tab(self.get_tabs()[-1])
         self._driver.close()
         self._driver.quit()
+        if self.options is not None:
+            try:
+                self._botcity_temp_dir = self.options._botcity_temp_dir
+            except Exception:
+                self._botcity_temp_dir = None
         self.options = None
         self.capabilities = None
         self._driver = None

botcity/web/browsers/chrome.py

@@ -70,10 +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._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

@@ -68,10 +68,11 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
     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._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}")
 

botcity/web/browsers/firefox.py

@@ -1,4 +1,3 @@
-import atexit
 import os
 import tempfile
 from typing import Dict
@@ -358,10 +357,11 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
     firefox_options.page_load_strategy = page_load_strategy
     if headless:
         firefox_options.add_argument('-headless')
+    firefox_options._botcity_temp_dir = None
     if not user_data_dir:
         temp_dir = tempfile.TemporaryDirectory(prefix="botcity_")
         user_data_dir = temp_dir.name
-    firefox_options._botcity_temp_dir = user_data_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

@@ -1,4 +1,3 @@
-import atexit
 import json
 import os
 import platform
@@ -73,10 +72,11 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
     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._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}")