feat(ddma eligibiltiy - v5 done
This commit is contained in:
@@ -9,10 +9,6 @@ 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):
|
||||
@@ -31,9 +27,6 @@ class AutomationDeltaDentalMAEligibilityCheck:
|
||||
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:
|
||||
@@ -52,145 +45,10 @@ class AutomationDeltaDentalMAEligibilityCheck:
|
||||
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)
|
||||
@@ -223,32 +81,6 @@ class AutomationDeltaDentalMAEligibilityCheck:
|
||||
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}"
|
||||
|
||||
Reference in New Issue
Block a user