ENH: Improving undetected-chrome finalization

Author: joao-voltarelli

botcity/web/browsers/undetected_chrome.py

@@ -1,3 +1,4 @@
+import atexit
 import json
 import os
 import platform
@@ -8,6 +9,8 @@
 from undetected_chromedriver.options import ChromeOptions
 from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
 
+from ..util import cleanup_temp_dir
+
 try:
     from undetected_chromedriver import Service as ChromeService  # noqa: F401, F403
 except ImportError:
@@ -77,6 +80,7 @@ def default_options(headless=False, download_folder_path=None, user_data_dir=Non
         temp_dir = tempfile.TemporaryDirectory(prefix="botcity_")
         user_data_dir = temp_dir.name
         chrome_options._botcity_temp_dir = user_data_dir
+        atexit.register(cleanup_temp_dir, temp_dir)
 
     chrome_options.add_argument(f"--user-data-dir={user_data_dir}")