Merge pull request #973 from surapuramakhil/ft_job_portal_refactoring

Ft job portal refactoring

constants.py

@@ -1,3 +1,10 @@
+DATE_ALL_TIME = "all_time"
+DATE_MONTH = "month"
+DATE_WEEK = "week"
+DATE_24_HOURS = "24_hours"
+
+LINKEDIN = "linkedin"
+
 # constants used in application
 SECRETS_YAML = "secrets.yaml"
 WORK_PREFERENCES_YAML = "work_preferences.yaml"

requirements.txt

@@ -28,3 +28,4 @@ webdriver-manager==4.0.2
 pytest
 pytest-mock
 pytest-cov
+undetected_chromedriver

src/ai_hawk/authenticator.py

@@ -9,12 +9,6 @@
 
 from src.logging import logger
 
-def get_authenticator(driver, platform):
-    if platform == 'linkedin':
-        return LinkedInAuthenticator(driver)
-    else:
-        raise NotImplementedError(f"Platform {platform} not implemented yet.")
-
 class AIHawkAuthenticator(ABC):
 
     @property
@@ -56,7 +50,7 @@ def handle_login(self):
     def prompt_for_credentials(self):
         try:
             logger.debug("Enter credentials...")
-            check_interval = 4  # Interval to log the current URL
+            check_interval = 45  # Interval to log the current URL
             elapsed_time = 0
 
             while True:
@@ -74,7 +68,7 @@ def prompt_for_credentials(self):
                     break
                 else:
                     # Optionally wait for the password field (or any other element you expect on the login page)
-                    WebDriverWait(self.driver, 10).until(
+                    WebDriverWait(self.driver, 60).until(
                         EC.presence_of_element_located((By.ID, "password"))
                     )
                     logger.debug("Password field detected, waiting for login completion.")
@@ -88,35 +82,3 @@ def prompt_for_credentials(self):
     @abstractmethod
     def handle_security_checks(self):
         pass
-
-class LinkedInAuthenticator(AIHawkAuthenticator):
-
-    @property
-    def home_url(self):
-        return "https://www.linkedin.com"
-
-    def navigate_to_login(self):
-        return self.driver.get("https://www.linkedin.com/login")
-
-    def handle_security_checks(self):
-        try:
-            logger.debug("Handling security check...")
-            WebDriverWait(self.driver, 10).until(
-                EC.url_contains('https://www.linkedin.com/checkpoint/challengesV2/')
-            )
-            logger.warning("Security checkpoint detected. Please complete the challenge.")
-            WebDriverWait(self.driver, 300).until(
-                EC.url_contains('https://www.linkedin.com/feed/')
-            )
-            logger.info("Security check completed")
-        except TimeoutException:
-            logger.error("Security check not completed. Please try again later.")
-
-    @property
-    def is_logged_in(self):
-        keywords = ['feed', 'mynetwork','jobs','messaging','notifications']
-        return any(item in self.driver.current_url for item in keywords) and 'linkedin.com' in self.driver.current_url
-
-    def __init__(self, driver):
-        super().__init__(driver)
-        pass

src/ai_hawk/bot_facade.py

@@ -1,3 +1,4 @@
+from ai_hawk.job_manager import AIHawkJobManager
 from src.logging import logger
 
 
@@ -28,7 +29,7 @@ class AIHawkBotFacade:
     def __init__(self, login_component, apply_component):
         logger.debug("Initializing AIHawkBotFacade")
         self.login_component = login_component
-        self.apply_component = apply_component
+        self.apply_component : AIHawkJobManager = apply_component
         self.state = AIHawkBotState()
         self.job_application_profile = None
         self.resume = None