Almost Done

worker.py

@@ -10,6 +10,7 @@ from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException
+from datetime import datetime
 
 class QuantumBot:
     def __init__(self, socketio, app):
@@ -53,30 +54,14 @@ class QuantumBot:
     def login(self, username, password):
         try:
             self.micro_status("Navigating to login page...")
-            self.driver.get("https://gateway.quantumepay.com/account/login")
-            
-            self.micro_status("Waiting for login form to be ready...")
-            
-            # --- Definitive Fix: Wait for the element to be fully interactable ---
-            username_field = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.element_to_be_clickable((By.ID, "Username"))
-            )
-            username_field.send_keys(username)
-            
-            password_field = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.element_to_be_clickable((By.ID, "Password"))
-            )
-            password_field.send_keys(password)
-            
-            login_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.element_to_be_clickable((By.ID, "login"))
-            )
-            login_button.click()
-
+            self.driver.get("https://gateway.quantumepay.com/")
+            time.sleep(2)
+            self.micro_status("Entering credentials...")
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.ID, "Username"))).send_keys(username)
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.ID, "Password"))).send_keys(password)
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.ID, "login"))).click()
             self.micro_status("Waiting for OTP screen...")
-            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(
-                EC.presence_of_element_located((By.ID, "code1"))
-            )
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.presence_of_element_located((By.ID, "code1")))
             return True, None
         except Exception as e:
             error_message = f"Error during login: {str(e)}"; print(f"[Bot Log] ERROR during login: {error_message}")
@@ -87,31 +72,54 @@ class QuantumBot:
             self.micro_status(f"Submitting OTP...")
             otp_digits = list(otp)
             for i in range(6):
-                WebDriverWait(self.driver, 5).until(EC.element_to_be_clickable((By.ID, f"code{i+1}"))).send_keys(otp_digits[i])
-
-            login_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.ID, "login")))
-            login_button.click()
-            
+                self.driver.find_element(By.ID, f"code{i+1}").send_keys(otp_digits[i])
+            WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.ID, "login"))).click()
             self.micro_status("Verifying login success...")
             WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//span[text()='Payments']")))
             return True, None
         except Exception as e:
             error_message = f"Error during OTP submission: {str(e)}"; print(f"[Bot Log] ERROR during OTP submission: {error_message}")
             return False, error_message
+            
+    def _get_calendar_months(self):
+        try:
+            titles = self.driver.find_elements(By.XPATH, "//div[contains(@class, 'vc-title')]")
+            return [datetime.strptime(title.text, "%B %Y") for title in titles]
+        except Exception: return []
+
+    def _select_date_in_calendar(self, target_date):
+        target_month_str = target_date.strftime("%B %Y")
+        self.micro_status(f"Navigating calendar to {target_month_str}...")
+        for _ in range(24): 
+            visible_months = self._get_calendar_months()
+            if any(d.strftime("%B %Y") == target_month_str for d in visible_months):
+                self.micro_status(f"Found month. Selecting day {target_date.day}.")
+                day_format = "%#d" if os.name == 'nt' else "%-d"
+                expected_aria_label = target_date.strftime(f"%A, %B {day_format}, %Y")
+                day_xpath = f"//span[@aria-label='{expected_aria_label}']"
+                day_element = WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.XPATH, day_xpath)))
+                self.driver.execute_script("arguments[0].click();", day_element); return
+            if target_date < visible_months[0]: self.driver.find_element(By.XPATH, "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-left')]").click()
+            else: self.driver.find_element(By.XPATH, "//div[contains(@class, 'vc-arrow') and contains(@class, 'is-right')]").click()
+            time.sleep(0.5)
+        raise Exception(f"Could not navigate to date {target_date.strftime('%Y-%m-%d')}")
+
+    def _set_date_range(self, start_date_str, end_date_str):
+        self.micro_status("Opening calendar to set date range...")
+        date_button = WebDriverWait(self.driver, self.DEFAULT_TIMEOUT).until(EC.element_to_be_clickable((By.XPATH, "//button[.//span[contains(text(), '-')]]")))
+        time.sleep(1); date_button.click()
+        start_date = datetime.strptime(start_date_str, "%Y-%m-%d"); end_date = datetime.strptime(end_date_str, "%Y-%m-%d")
+        self._select_date_in_calendar(start_date); time.sleep(1)
+        self._select_date_in_calendar(end_date); time.sleep(1)
+        self.driver.find_element(By.TAG_NAME, "body").click()
+        time.sleep(3)
 
     def process_patient_list(self, patient_data, workflow, date_range=None):
         results = []
-        # Create a copy of the list to iterate over, so we can modify the original if needed
         records_to_process = list(patient_data)
-
         for index, record in enumerate(records_to_process):
-            if self.termination_event.is_set():
-                print("[Bot Log] Termination detected.")
-                break
-            
-            patient_name = record['Name']
-            patient_prn = record.get('PRN', '')
-            
+            if self.termination_event.is_set(): print("[Bot Log] Termination detected."); break
+            patient_name = record['Name']; patient_prn = record.get('PRN', '')
             if not patient_prn or not str(patient_prn).strip():
                 status = 'Skipped - No PRN'
                 self.micro_status(f"Skipping '{patient_name}' (No PRN).")
@@ -122,7 +130,6 @@ class QuantumBot:
                     status = self._process_single_void(patient_name, patient_prn)
                 elif workflow == 'refund':
                     status = self._process_single_refund(patient_name, patient_prn, date_range['start_date'], date_range['end_date'])
-
             results.append({'Name': patient_name, 'PRN': patient_prn, 'Status': status})
             with self.app.app_context():
                 self.socketio.emit('log_update', {'name': patient_name, 'prn': patient_prn, 'status': status})
@@ -203,7 +210,7 @@ class QuantumBot:
                 return 'Bad'
         except Exception as e:
             print(f"An error occurred while processing {patient_name}: {e}"); return 'Error'
-
+            
     def shutdown(self):
         try:
             if self.driver: self.driver.quit()