From e9efa343dd95d01591a8c84765d041286c9a2fa9 Mon Sep 17 00:00:00 2001 From: Tilak Patel Date: Sun, 13 Oct 2024 12:52:35 -0400 Subject: [PATCH] reuse extractor --- src/aihawk_job_manager.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/src/aihawk_job_manager.py b/src/aihawk_job_manager.py index 007adce73..406384ead 100644 --- a/src/aihawk_job_manager.py +++ b/src/aihawk_job_manager.py @@ -43,6 +43,7 @@ def __init__(self, driver): self.driver = driver self.set_old_answers = set() self.easy_applier_component = None + self.extractor = None logger.debug("AIHawkJobManager initialized successfully") def set_parameters(self, parameters): @@ -217,11 +218,15 @@ def get_jobs_from_page(self): def apply_jobs(self): job_list = [] - for e in EXTRACTORS: - extracted_jobs = e.get_job_list(self.driver) - if len(extracted_jobs) > 0: - job_list = extracted_jobs # break when we find a valid extractor - break + if self.extractor is not None: # we found a working extractor + job_list = self.extractor.get_job_list(self.driver) + else: + for e in EXTRACTORS: + extracted_jobs = e.get_job_list(self.driver) + if len(extracted_jobs) > 0: + job_list = extracted_jobs # break when we find a valid extractor + self.extractor = e + break for job in job_list: