STYLE: Implement Exception in try/except

Author: kayqueGovetri

botcity/web/bot.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+import atexit
 import base64
 import functools
 import glob
@@ -25,6 +28,7 @@
 from selenium.webdriver.remote.webelement import WebElement
 from selenium.webdriver.support.wait import WebDriverWait, TimeoutException, NoSuchElementException
 from selenium.webdriver.support import expected_conditions as EC
+from weakref import ReferenceType, ref
 
 from . import config, cv2find
 from .browsers import BROWSER_CONFIGS, Browser, PageLoadStrategy
@@ -39,6 +43,14 @@
 logger = logging.getLogger(__name__)
 
 
+def _cleanup(bot: ReferenceType[WebBot]):
+    if bot() is not None:
+        try:
+            bot().stop_browser()
+        except Exception:
+            pass
+
+
 class WebBot(BaseBot):
     KEYS = Keys
     DEFAULT_DIMENSIONS = (1600, 900)
@@ -77,6 +89,8 @@ def __init__(self, headless=False):
 
         self._download_folder_path = os.getcwd()
 
+        atexit.register(_cleanup, ref(self))
+
     def __enter__(self):
         pass