add travel code notify

browser.py

@@ -0,0 +1,62 @@
+from selenium import webdriver
+from selenium.common.exceptions import NoSuchElementException
+from selenium.webdriver.common.by import By
+from loguru import logger
+from fake_useragent import UserAgent
+from functools import cache
+import time
+
+
+class AbstractChrome:
+    def __init__(self, headless: bool = True):
+        self.chrome_options = webdriver.ChromeOptions()
+        self.chrome_options.add_argument(f'user-agent={self._user_aget}')
+        self.chrome_options.add_experimental_option('excludeSwitches', ['enable-logging'])
+
+        if headless:
+            self.chrome_options.add_argument('--mute-audio')
+            self.chrome_options.add_argument('--auto-open-devtools-for-tabs')
+            self.chrome_options.add_argument('--headless')
+            self.chrome_options.add_argument('--window-size=1440,900')
+        self.chrome_options.add_argument('--disable-gpu')
+        self.chrome_options.add_argument('--disable-dev-shm-usage')
+        self.chrome_options.add_argument("--log-level=3")
+        self.driver = webdriver.Chrome(
+            options=self.chrome_options
+        )
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        self.driver.close()
+
+    @property
+    @cache
+    def _user_aget(self):
+        ua = UserAgent()
+        return ua.googlechrome
+
+
+class PrimaryLocationClicker(AbstractChrome):
+
+    def click(self, auth_url: str):
+        self.driver.get(auth_url)
+        try:
+            element = self.driver.find_element(By.CSS_SELECTOR, '[data-uia="set-primary-location-action"]')
+            element.click()
+        except NoSuchElementException as e:
+            self.driver.save_screenshot(f'{time.time()}.png')
+            logger.error(e)
+
+
+class TravelVerificationRetriever(AbstractChrome):
+
+    def get_code(self, auth_url: str) -> str:
+        self.driver.get(auth_url)
+        try:
+            element = self.driver.find_element(By.CSS_SELECTOR, '[data-uia="travel-verification-otp"]')
+            return element.text
+        except NoSuchElementException as e:
+            self.driver.save_screenshot(f'{time.time()}.png')
+            logger.error(e)