FIX: Ensuring the temp folder is deleted at the end

Author: joao-voltarelli

botcity/web/bot.py

@@ -30,7 +30,7 @@
 from selenium.webdriver.support.wait import WebDriverWait, TimeoutException, NoSuchElementException
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.common.print_page_options import PrintOptions
-from weakref import ReferenceType, ref
+from weakref import ref
 
 from . import config, cv2find, compat
 from .browsers import BROWSER_CONFIGS, Browser, PageLoadStrategy
@@ -45,13 +45,16 @@
 logger = logging.getLogger(__name__)
 
 
-def _cleanup(bot: ReferenceType[WebBot]):
-    if bot() is not None:
+def _cleanup(driver, temp_dir):
+    if driver() is not None:
+        try:
+            if driver().service.is_connectable():
+                driver().quit()
+        except Exception:
+            pass
+
+    if temp_dir:
         try:
-            bot().stop_browser()
-            temp_dir = bot()._botcity_temp_dir
-            if not temp_dir:
-                return None
             shutil.rmtree(temp_dir, ignore_errors=True)
         except Exception:
             pass
@@ -97,8 +100,6 @@ def __init__(self, headless=False):
         self._download_folder_path = os.getcwd()
         self._botcity_temp_dir = None
 
-        atexit.register(_cleanup, ref(self))
-
     def __enter__(self):
         pass
 
@@ -306,6 +307,8 @@ def check_driver():
         self._others_configurations()
         self.set_screen_resolution()
 
+        atexit.register(_cleanup, ref(self._driver), self.options._botcity_temp_dir)
+
     def _instantiate_driver(self, driver_class, func_def_options):
         """
         It is necessary to create this function because we isolated the instantiation of the driver,