from selenium import webdriver from selenium.common.exceptions import WebDriverException, TimeoutException from selenium.webdriver.chrome.service import Service from selenium.webdriver.common.by import By from selenium.webdriver.common.keys import Keys from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC from webdriver_manager.chrome import ChromeDriverManager import time import os import base64 import stat import pickle import traceback import urllib class AutomationDeltaDentalMAEligibilityCheck: def __init__(self, data): self.headless = False self.driver = None self.data = data.get("data", {}) if isinstance(data, dict) else {} # Flatten values for convenience self.memberId = self.data.get("memberId", "") self.dateOfBirth = self.data.get("dateOfBirth", "") self.massddma_username = self.data.get("massddmaUsername", "") self.massddma_password = self.data.get("massddmaPassword", "") self.download_dir = os.path.abspath("seleniumDownloads") os.makedirs(self.download_dir, exist_ok=True) self.cookies_path = os.path.abspath("cookies.pkl") def config_driver(self): options = webdriver.ChromeOptions() if self.headless: options.add_argument("--headless") # Add PDF download preferences prefs = { "download.default_directory": self.download_dir, "plugins.always_open_pdf_externally": True, "download.prompt_for_download": False, "download.directory_upgrade": True } options.add_experimental_option("prefs", prefs) s = Service(ChromeDriverManager().install()) driver = webdriver.Chrome(service=s, options=options) self.driver = driver def _origin_from_url(self, url): p = urllib.parse.urlparse(url) origin = f"{p.scheme}://{p.netloc}" return origin def try_cookie_login(self, url): try: if not os.path.exists(self.cookies_path): print(f"[try_cookie_login] cookies file not found at: {self.cookies_path}") return "NO_COOKIES" origin = self._origin_from_url(url) print(f"origin {origin}") print(f"[try_cookie_login] Attempting cookie-login. origin={origin}, cookies_path={self.cookies_path}") # load origin first self.driver.get(origin) time.sleep(1) try: cookies = pickle.load(open(self.cookies_path, "rb")) print(f"[try_cookie_login] Loaded {len(cookies)} cookies from file.") except Exception as e: print("[try_cookie_login] Failed to load cookies.pkl:", e) traceback.print_exc() return "FAILED" added = 0 for c in cookies: try: self.driver.add_cookie(c) added += 1 except Exception as ex: print(f"[try_cookie_login] warning adding cookie {c.get('name')}: {ex}") print(f"[try_cookie_login] Added {added}/{len(cookies)} cookies to browser.") # now load target url time.sleep(2) self.driver.get(origin) time.sleep(2) # detect logged-in state try: WebDriverWait(self.driver, 8).until( EC.presence_of_element_located((By.XPATH, "//a[text()='Member Eligibility' or contains(., 'Member Eligibility')]")) ) # refresh cookie file with any updates try: current = self.driver.get_cookies() pickle.dump(current, open(self.cookies_path, "wb")) print("[try_cookie_login] Cookie-login success; cookies.pkl refreshed.") except Exception as e: print("[try_cookie_login] Warning saving refreshed cookies:", e) return "Success" except TimeoutException: print("[try_cookie_login] Cookie-login did not find dashboard element.") return "FAILED" except Exception as e: print("[try_cookie_login] Unexpected exception:", e) traceback.print_exc() return "FAILED" def handle_cookies(self, url, create_if_missing=True): try: origin = self._origin_from_url(url) self.driver.get(origin) time.sleep(1) if not os.path.exists(self.cookies_path): if not create_if_missing: print("[handle_cookies] cookies file missing and create_if_missing=False") return False try: input("add ?") cookies = self.driver.get_cookies() pickle.dump(cookies, open(self.cookies_path, "wb")) print(f"[handle_cookies] Saved {len(cookies)} cookies to {self.cookies_path}") return True except Exception as e: print("[handle_cookies] failed to save cookies:", e) traceback.print_exc() return False else: try: cookies = pickle.load(open(self.cookies_path, "rb")) print(f"[handle_cookies] Loaded {len(cookies)} cookies from {self.cookies_path}") except Exception as e: print("[handle_cookies] failed to load cookies.pkl:", e) traceback.print_exc() return False # ensure on origin to add cookies self.driver.get(origin) time.sleep(1) added = 0 for c in cookies: try: self.driver.add_cookie(c) added += 1 except Exception: pass print(f"[handle_cookies] Re-applied {added}/{len(cookies)} cookies.") time.sleep(1) self.driver.refresh() time.sleep(2) try: new_cookies = self.driver.get_cookies() pickle.dump(new_cookies, open(self.cookies_path, "wb")) print(f"[handle_cookies] Refreshed cookies.pkl with {len(new_cookies)} cookies.") except Exception as e: print("[handle_cookies] failed to refresh cookies.pkl:", e) traceback.print_exc() return True except Exception as e: print("[handle_cookies] Unexpected exception:", e) traceback.print_exc() return False def login(self, url): wait = WebDriverWait(self.driver, 30) try: print(f"[login] Starting login for url: {url}") cookie_attempt = self.try_cookie_login(url) print(f"[login] cookie_attempt result: {cookie_attempt}") if cookie_attempt == "Success": print("[login] Logged in via cookies.") return "Success" elif cookie_attempt == "NO_COOKIES": print("[login] No cookies present; will do credential login.") else: print("[login] Cookie-login failed; falling back to username/password login.") # credential login flow self.driver.get(url) print("[login] On login page, attempting to fill credentials.") email_field = wait.until(EC.presence_of_element_located((By.XPATH, "//input[@name='username' and @type='text']"))) email_field.clear() email_field.send_keys(self.massddma_username) password_field = wait.until(EC.presence_of_element_located((By.XPATH, "//input[@name='password' and @type='password']"))) password_field.clear() password_field.send_keys(self.massddma_password) # remember me try: remember_me_checkbox = wait.until(EC.element_to_be_clickable( (By.XPATH, "//label[.//span[contains(text(),'Remember me')]]") )) remember_me_checkbox.click() except: print("[login] Remember me checkbox not found (continuing).") login_button = wait.until(EC.element_to_be_clickable((By.XPATH, "//button[@type='submit' and @aria-label='Sign in']"))) login_button.click() # OTP detection try: otp_candidate = WebDriverWait(self.driver, 30).until( EC.presence_of_element_located( (By.XPATH, "//input[contains(@aria-lable,'Verification code') or contains(@placeholder,'Enter your verification code')]") ) ) if otp_candidate: print("[login] OTP input detected -> OTP_REQUIRED") return "OTP_REQUIRED" except TimeoutException: print("[login] No OTP input detected in allowed time.") # check for dashboard element try: logged_in_el = WebDriverWait(self.driver, 8).until( EC.presence_of_element_located((By.XPATH, "//a[text()='Member Eligibility' or contains(., 'Member Eligibility')]")) ) if logged_in_el: print("[login] Credential login succeeded; calling handle_cookies to save/refresh cookies.") try: self.handle_cookies(self.driver.current_url, create_if_missing=True) except Exception as e: print("[login] handle_cookies raised:", e) traceback.print_exc() return "Success" except TimeoutException: cur = self.driver.current_url or "" print(f"[login] Post-login element not found. Current URL: {cur}") if "dashboard" in cur or "providers" in cur: print("[login] URL looks like dashboard; treating as success and saving cookies.") try: self.handle_cookies(self.driver.current_url, create_if_missing=True) except Exception: pass return "Success" return "ERROR:LOGIN FAILED - unable to detect success or OTP" except Exception as e: print("[login] Exception during login:", e) return f"ERROR:LOGIN FAILED: {e}" def step1(self): wait = WebDriverWait(self.driver, 30) try: # Fill Member ID member_id_input = wait.until(EC.presence_of_element_located((By.XPATH, '//input[@placeholder="Search by member ID"]'))) member_id_input.clear() member_id_input.send_keys(self.memberId) # Fill DOB parts try: dob_parts = self.dateOfBirth.split("-") year = dob_parts[0] # "1964" month = dob_parts[1].zfill(2) # "04" day = dob_parts[2].zfill(2) # "17" except Exception as e: print(f"Error parsing DOB: {e}") return "ERROR: PARSING DOB" # 1) locate the specific member DOB container dob_container = wait.until( EC.presence_of_element_located( (By.XPATH, "//div[@data-testid='member-search_date-of-birth']") ) ) # 2) find the editable spans *inside that container* using relative XPaths month_elem = dob_container.find_element(By.XPATH, ".//span[@data-type='month' and @contenteditable='true']") day_elem = dob_container.find_element(By.XPATH, ".//span[@data-type='day' and @contenteditable='true']") year_elem = dob_container.find_element(By.XPATH, ".//span[@data-type='year' and @contenteditable='true']") # Helper to click, select-all and type (pure send_keys approach) def replace_with_sendkeys(el, value): # focus (same as click) el.click() # select all (Ctrl+A) and delete (some apps pick up BACKSPACE better — we use BACKSPACE after select) el.send_keys(Keys.CONTROL, "a") el.send_keys(Keys.BACKSPACE) # type the value el.send_keys(value) # optionally blur or tab out if app expects it # el.send_keys(Keys.TAB) replace_with_sendkeys(month_elem, month) time.sleep(0.05) replace_with_sendkeys(day_elem, day) time.sleep(0.05) replace_with_sendkeys(year_elem, year) # Click Continue button continue_btn = wait.until(EC.element_to_be_clickable((By.XPATH, '//button[@data-testid="member-search_search-button"]'))) continue_btn.click() # Check for error message try: error_msg = WebDriverWait(self.driver, 5).until(EC.presence_of_element_located( (By.XPATH, '//div[@data-testid="member-search-result-no-results"]') )) if error_msg: print("Error: Invalid Member ID or Date of Birth.") return "ERROR: INVALID MEMBERID OR DOB" except TimeoutException: pass return "Success" except Exception as e: print(f"Error while step1 i.e Cheking the MemberId and DOB in: {e}") return "ERROR:STEP1" def step2(self): wait = WebDriverWait(self.driver, 90) try: # 1) find the eligibility inside the correct cell status_link = wait.until(EC.presence_of_element_located(( By.XPATH, "(//tbody//tr)[1]//a[contains(@href, 'member-eligibility-search')]" ))) eligibilityText = status_link.text.strip().lower() try: WebDriverWait(self.driver, 30).until( lambda d: d.execute_script("return document.readyState") == "complete" ) except Exception: print("Warning: document.readyState did not become 'complete' within timeout") # Give some time for lazy content to finish rendering (adjust if needed) time.sleep(0.6) # Get total page size and DPR total_width = int(self.driver.execute_script( "return Math.max(document.body.scrollWidth, document.documentElement.scrollWidth, document.documentElement.clientWidth);" )) total_height = int(self.driver.execute_script( "return Math.max(document.body.scrollHeight, document.documentElement.scrollHeight, document.documentElement.clientHeight);" )) dpr = float(self.driver.execute_script("return window.devicePixelRatio || 1;")) # Set device metrics to the full page size so Page.captureScreenshot captures everything # Note: Some pages are extremely tall; if you hit memory limits, you can capture in chunks. self.driver.execute_cdp_cmd('Emulation.setDeviceMetricsOverride', { "mobile": False, "width": total_width, "height": total_height, "deviceScaleFactor": dpr, "screenOrientation": {"angle": 0, "type": "portraitPrimary"} }) # Small pause for layout to settle after emulation change time.sleep(0.15) # Capture screenshot (base64 PNG) result = self.driver.execute_cdp_cmd("Page.captureScreenshot", {"format": "png", "fromSurface": True}) image_data = base64.b64decode(result.get('data', '')) screenshot_path = os.path.join(self.download_dir, f"ss_{self.memberId}.png") with open(screenshot_path, "wb") as f: f.write(image_data) # Restore original metrics to avoid affecting further interactions try: self.driver.execute_cdp_cmd('Emulation.clearDeviceMetricsOverride', {}) except Exception: # non-fatal: continue pass print("Screenshot saved at:", screenshot_path) output = { "status": "success", "eligibility": eligibilityText, "ss_path": screenshot_path } print(output) return output except Exception as e: print("ERROR in step2:", e) # Empty the download folder (remove files / symlinks only) try: dl = os.path.abspath(self.download_dir) if os.path.isdir(dl): for name in os.listdir(dl): item = os.path.join(dl, name) try: if os.path.isfile(item) or os.path.islink(item): os.remove(item) print(f"[cleanup] removed: {item}") except Exception as rm_err: print(f"[cleanup] failed to remove {item}: {rm_err}") print(f"[cleanup] emptied download dir: {dl}") else: print(f"[cleanup] download dir does not exist: {dl}") except Exception as cleanup_exc: print(f"[cleanup] unexpected error while cleaning downloads dir: {cleanup_exc}") return {"status": "error", "message": str(e)} finally: # Keep your existing quit behavior; if you want the driver to remain open for further # actions, remove or change this. if self.driver: try: self.driver.quit() except Exception: pass def main_workflow(self, url): try: self.config_driver() self.driver.maximize_window() time.sleep(3) login_result = self.login(url) if login_result.startswith("ERROR"): return {"status": "error", "message": login_result} if login_result == "OTP_REQUIRED": return {"status": "otp_required", "message": "OTP required after login"} step1_result = self.step1() if step1_result.startswith("ERROR"): return {"status": "error", "message": step1_result} step2_result = self.step2() if step2_result.get("status") == "error": return {"status": "error", "message": step2_result.get("message")} return step2_result except Exception as e: return { "status": "error", "message": e } finally: try: if self.driver: self.driver.quit() except Exception: pass