Merge pull request #884 from 49Simon/fix-saving-failed-applications

fix: prevent failed applications from being saved in success.json

Co-authored-by: 49Simon <85809106+49Simon@users.noreply.github.com>

src/ai_hawk/job_manager.py

@@ -4,7 +4,6 @@
 import time
 from itertools import product
 from pathlib import Path
-from turtle import color
 from datetime import datetime
 
 from inputimeout import inputimeout, TimeoutOccurred
@@ -14,6 +13,7 @@
 
 from ai_hawk.linkedIn_easy_applier import AIHawkEasyApplier
 from config import JOB_MAX_APPLICATIONS, JOB_MIN_APPLICATIONS, MINIMUM_WAIT_TIME_IN_SECONDS
+from src.custom_exception import JobNotSuitableException
 from src.job import Job
 from src.logging import logger
 
@@ -412,6 +412,10 @@ def apply_jobs(self):
                     self.easy_applier_component.job_apply(job)
                     self.write_to_file(job, "success")
                     logger.debug(f"Applied to job: {job.title} at {job.company}")
+            except JobNotSuitableException as e:
+                logger.debug(f"Job not suitable for application: {job.title} at {job.company}")
+                self.write_to_file(job, "skipped", str(e))
+                continue
             except Exception as e:
                 logger.error(f"Failed to apply for {job.title} at {job.company}: {e}",exc_info=True)
                 self.write_to_file(job, "failed", f"Application error: {str(e)}")

src/ai_hawk/linkedIn_easy_applier.py

@@ -22,6 +22,7 @@
 from jobContext import JobContext
 from job_application import JobApplication
 from job_application_saver import ApplicationSaver
+from src.custom_exception import JobNotSuitableException
 import src.utils as utils
 from src.logging import logger
 from src.job import Job
@@ -157,7 +158,7 @@ def job_apply(self, job: Job):
 
             # Todo: add this job to skip list with it's reason
             if not self.gpt_answerer.is_job_suitable():
-                return
+                raise JobNotSuitableException("LLM score determined this job is not suitable for you")
 
             logger.debug("Attempting to click 'Easy Apply' button")
             actions = ActionChains(self.driver)

src/custom_exception.py

@@ -0,0 +1,4 @@
+class JobNotSuitableException(Exception):
+    def __init__(self, message):
+        self.message = message
+        super().__init__(self.message)