3978b16d7d
- Add United/DentalHub pre-authorization Selenium worker, helpers, backend route/processor/client, and frontend OTP modal + wired button - Add Tufts SCO pre-authorization Selenium worker, helpers, backend route/processor/client, and frontend OTP modal + wired button - Fix btnSubmitAuthorization selector for UnitedDH preauth step2 - Fix Tufts SCO preauth step3 to target <span>pre-authorization</span> button - Cloud storage search: default to "both" mode so patient folder names match on first search - Cloud storage folder panel: auto-scroll to panel when opened from search results - Cloud storage files: show inline image thumbnails and PDF badge in file grid Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
343 lines
15 KiB
Python
343 lines
15 KiB
Python
import os
|
|
import time
|
|
import asyncio
|
|
from typing import Dict, Any
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
|
from selenium.webdriver.support import expected_conditions as EC
|
|
from selenium.common.exceptions import WebDriverException, TimeoutException
|
|
|
|
from selenium_UnitedDH_preAuthWorker import AutomationUnitedDHPreAuth
|
|
|
|
sessions: Dict[str, Dict[str, Any]] = {}
|
|
|
|
SESSION_OTP_TIMEOUT = int(os.getenv("SESSION_OTP_TIMEOUT", "120"))
|
|
|
|
|
|
def make_session_entry() -> str:
|
|
import uuid
|
|
sid = str(uuid.uuid4())
|
|
sessions[sid] = {
|
|
"status": "created",
|
|
"created_at": time.time(),
|
|
"last_activity": time.time(),
|
|
"bot": None,
|
|
"driver": None,
|
|
"otp_event": asyncio.Event(),
|
|
"otp_value": None,
|
|
"result": None,
|
|
"message": None,
|
|
}
|
|
return sid
|
|
|
|
|
|
async def cleanup_session(sid: str, message: str | None = None):
|
|
s = sessions.get(sid)
|
|
if not s:
|
|
return
|
|
try:
|
|
if s.get("status") not in ("completed", "error", "not_found"):
|
|
s["status"] = "error"
|
|
if message:
|
|
s["message"] = message
|
|
ev = s.get("otp_event")
|
|
if ev and not ev.is_set():
|
|
ev.set()
|
|
finally:
|
|
sessions.pop(sid, None)
|
|
print(f"[helpers_uniteddh_preauth] cleaned session {sid}")
|
|
|
|
|
|
async def _remove_session_later(sid: str, delay: int = 20):
|
|
await asyncio.sleep(delay)
|
|
await cleanup_session(sid)
|
|
|
|
|
|
def _minimize_browser(bot):
|
|
try:
|
|
if bot and bot.driver:
|
|
try:
|
|
bot.driver.get("about:blank")
|
|
except Exception:
|
|
pass
|
|
try:
|
|
bot.driver.minimize_window()
|
|
print("[UnitedDH PreAuth] Browser minimized after error")
|
|
return
|
|
except Exception:
|
|
pass
|
|
try:
|
|
bot.driver.set_window_position(-10000, -10000)
|
|
print("[UnitedDH PreAuth] Browser moved off-screen after error")
|
|
except Exception:
|
|
pass
|
|
except Exception as e:
|
|
print(f"[UnitedDH PreAuth] Could not hide browser: {e}")
|
|
|
|
|
|
async def start_uniteddh_preauth_run(sid: str, data: dict, url: str):
|
|
"""
|
|
Run the United/DentalHub pre-authorization workflow.
|
|
Login/OTP handling mirrors helpers_uniteddh_claim.py exactly.
|
|
Pre-auth steps call selenium_UnitedDH_preAuthWorker.
|
|
"""
|
|
s = sessions.get(sid)
|
|
if not s:
|
|
return {"status": "error", "message": "session not found"}
|
|
|
|
s["status"] = "running"
|
|
s["last_activity"] = time.time()
|
|
|
|
bot = None
|
|
try:
|
|
bot = AutomationUnitedDHPreAuth(data)
|
|
bot.config_driver()
|
|
|
|
s["bot"] = bot
|
|
s["driver"] = bot.driver
|
|
s["last_activity"] = time.time()
|
|
|
|
try:
|
|
bot.driver.maximize_window()
|
|
bot.driver.get(url)
|
|
await asyncio.sleep(1)
|
|
except Exception as e:
|
|
s["status"] = "error"
|
|
s["message"] = f"Navigation failed: {e}"
|
|
await cleanup_session(sid)
|
|
return {"status": "error", "message": s["message"]}
|
|
|
|
# --- Login ---
|
|
try:
|
|
login_result = bot.login(url)
|
|
except WebDriverException as wde:
|
|
s["status"] = "error"
|
|
s["message"] = f"Selenium driver error during login: {wde}"
|
|
await cleanup_session(sid, s["message"])
|
|
return {"status": "error", "message": s["message"]}
|
|
except Exception as e:
|
|
s["status"] = "error"
|
|
s["message"] = f"Unexpected error during login: {e}"
|
|
await cleanup_session(sid, s["message"])
|
|
return {"status": "error", "message": s["message"]}
|
|
|
|
# ── Already logged in ────────────────────────────────────────────────
|
|
if isinstance(login_result, str) and login_result == "ALREADY_LOGGED_IN":
|
|
print("[UnitedDH PreAuth] Session persisted — skipping OTP")
|
|
s["status"] = "running"
|
|
s["message"] = "Session persisted"
|
|
|
|
# ── OTP required ─────────────────────────────────────────────────────
|
|
elif isinstance(login_result, str) and login_result == "OTP_REQUIRED":
|
|
s["status"] = "waiting_for_otp"
|
|
s["message"] = "OTP required for login - please enter OTP in browser"
|
|
s["last_activity"] = time.time()
|
|
|
|
driver = s["driver"]
|
|
max_polls = SESSION_OTP_TIMEOUT
|
|
login_success = False
|
|
|
|
print(f"[UnitedDH PreAuth OTP] Waiting for user to enter OTP (polling browser for {SESSION_OTP_TIMEOUT}s)...")
|
|
|
|
for poll in range(max_polls):
|
|
await asyncio.sleep(1)
|
|
s["last_activity"] = time.time()
|
|
|
|
try:
|
|
otp_value = s.get("otp_value")
|
|
if otp_value:
|
|
print(f"[UnitedDH PreAuth OTP] OTP received from app: {otp_value}")
|
|
try:
|
|
otp_input = driver.find_element(By.XPATH,
|
|
"//input[contains(@name,'otp') or contains(@name,'code') or @type='tel' or contains(@aria-label,'Verification') or contains(@placeholder,'code')]"
|
|
)
|
|
otp_input.clear()
|
|
otp_input.send_keys(otp_value)
|
|
try:
|
|
verify_btn = driver.find_element(By.XPATH, "//button[@type='button' and @aria-label='Verify']")
|
|
verify_btn.click()
|
|
print("[UnitedDH PreAuth OTP] Clicked verify button (aria-label)")
|
|
except:
|
|
try:
|
|
verify_btn = driver.find_element(By.XPATH, "//button[contains(text(),'Verify') or contains(text(),'Submit') or @type='submit']")
|
|
verify_btn.click()
|
|
print("[UnitedDH PreAuth OTP] Clicked verify button (text/type)")
|
|
except:
|
|
otp_input.send_keys("\n")
|
|
print("[UnitedDH PreAuth OTP] Pressed Enter as fallback")
|
|
print("[UnitedDH PreAuth OTP] OTP typed and submitted via app")
|
|
s["otp_value"] = None
|
|
await asyncio.sleep(3)
|
|
except Exception as type_err:
|
|
print(f"[UnitedDH PreAuth OTP] Failed to type OTP from app: {type_err}")
|
|
|
|
current_url = driver.current_url.lower()
|
|
print(f"[UnitedDH PreAuth OTP Poll {poll+1}/{max_polls}] URL: {current_url[:60]}...")
|
|
|
|
if "member" in current_url or "dashboard" in current_url or "eligibility" in current_url or "home" in current_url:
|
|
try:
|
|
WebDriverWait(driver, 5).until(
|
|
EC.presence_of_element_located((By.XPATH,
|
|
'//input[@placeholder="Search by member ID"] | //input[contains(@placeholder,"Search")] | //*[contains(@class,"dashboard")]'
|
|
))
|
|
)
|
|
print("[UnitedDH PreAuth OTP] Dashboard/search element found - login successful!")
|
|
login_success = True
|
|
break
|
|
except TimeoutException:
|
|
print("[UnitedDH PreAuth OTP] On member page but search input not found, continuing to poll...")
|
|
|
|
try:
|
|
driver.find_element(By.XPATH,
|
|
"//input[contains(@name,'otp') or contains(@name,'code') or @type='tel' or contains(@aria-label,'Verification') or contains(@placeholder,'code') or contains(@placeholder,'Code')]"
|
|
)
|
|
print(f"[UnitedDH PreAuth OTP Poll {poll+1}] OTP input still visible - waiting...")
|
|
except:
|
|
if "login" in current_url or "app/login" in current_url:
|
|
print("[UnitedDH PreAuth OTP] OTP input gone, trying to navigate to dashboard...")
|
|
try:
|
|
driver.get("https://app.dentalhub.com/app/dashboard")
|
|
await asyncio.sleep(2)
|
|
except:
|
|
pass
|
|
|
|
except Exception as poll_err:
|
|
print(f"[UnitedDH PreAuth OTP Poll {poll+1}] Error: {poll_err}")
|
|
|
|
if not login_success:
|
|
try:
|
|
print("[UnitedDH PreAuth OTP] Final attempt - navigating to dashboard...")
|
|
driver.get("https://app.dentalhub.com/app/dashboard")
|
|
await asyncio.sleep(3)
|
|
WebDriverWait(driver, 10).until(
|
|
EC.presence_of_element_located((By.XPATH,
|
|
'//input[@placeholder="Search by member ID"] | //input[contains(@placeholder,"Search")] | //*[contains(@class,"dashboard")]'
|
|
))
|
|
)
|
|
print("[UnitedDH PreAuth OTP] Dashboard element found - login successful!")
|
|
login_success = True
|
|
except TimeoutException:
|
|
s["status"] = "error"
|
|
s["message"] = "OTP timeout - login not completed"
|
|
await cleanup_session(sid)
|
|
return {"status": "error", "message": "OTP not completed in time"}
|
|
except Exception as final_err:
|
|
s["status"] = "error"
|
|
s["message"] = f"OTP verification failed: {final_err}"
|
|
await cleanup_session(sid)
|
|
return {"status": "error", "message": s["message"]}
|
|
|
|
if login_success:
|
|
s["status"] = "running"
|
|
s["message"] = "Login successful after OTP"
|
|
print("[UnitedDH PreAuth OTP] Proceeding to pre-auth steps...")
|
|
|
|
# ── Login succeeded without OTP ───────────────────────────────────────
|
|
elif isinstance(login_result, str) and login_result == "SUCCESS":
|
|
print("[UnitedDH PreAuth] Login succeeded without OTP")
|
|
s["status"] = "running"
|
|
s["message"] = "Login succeeded"
|
|
|
|
# ── Login error ───────────────────────────────────────────────────────
|
|
elif isinstance(login_result, str) and login_result.startswith("ERROR"):
|
|
s["status"] = "error"
|
|
s["message"] = login_result
|
|
await cleanup_session(sid)
|
|
return {"status": "error", "message": login_result}
|
|
|
|
# --- Pre-auth steps ---
|
|
for step_name, step_fn in [
|
|
("step1_search_patient", bot.step1_search_patient),
|
|
("step2_click_preauth_button", bot.step2_click_preauth_button),
|
|
("step3_continue_prefilled", bot.step3_continue_prefilled),
|
|
("step4_select_insurance_ok", bot.step4_select_insurance_ok),
|
|
("step5_practitioner_continue", bot.step5_practitioner_continue),
|
|
("step6_fill_preauth_form", bot.step6_fill_preauth_form),
|
|
("step7_attach_files", bot.step7_attach_files),
|
|
("step8_submit_preauth", bot.step8_submit_preauth),
|
|
]:
|
|
result = step_fn()
|
|
print(f"[UnitedDH PreAuth] {step_name} result: {result}")
|
|
if isinstance(result, str) and result.startswith("ERROR"):
|
|
s["status"] = "error"
|
|
s["message"] = result
|
|
_minimize_browser(bot)
|
|
asyncio.create_task(_remove_session_later(sid, 30))
|
|
return {"status": "error", "message": result}
|
|
|
|
# --- Step 9: PDF + pre-auth number ---
|
|
step9_result = bot.step9_save_confirmation_pdf()
|
|
print(f"[UnitedDH PreAuth] step9 result: {step9_result}")
|
|
if isinstance(step9_result, str) and step9_result.startswith("ERROR"):
|
|
print(f"[UnitedDH PreAuth] step9 warning (non-fatal): {step9_result}")
|
|
step9_result = {}
|
|
|
|
pdf_path = step9_result.get("pdf_path") if isinstance(step9_result, dict) else None
|
|
pdf_url = None
|
|
if pdf_path:
|
|
import os as _os
|
|
filename = _os.path.basename(pdf_path)
|
|
port = _os.getenv("PORT", "5002")
|
|
url_host = _os.getenv("HOST", "localhost")
|
|
pdf_url = f"http://{url_host}:{port}/downloads/{filename}"
|
|
print(f"[UnitedDH PreAuth] pdf_url: {pdf_url}")
|
|
|
|
preauth_number = step9_result.get("preAuthNumber") if isinstance(step9_result, dict) else None
|
|
|
|
result = {
|
|
"status": "success",
|
|
"message": "United/DentalHub pre-authorization submitted successfully",
|
|
"preAuthNumber": preauth_number,
|
|
"pdf_url": pdf_url,
|
|
}
|
|
s["status"] = "completed"
|
|
s["result"] = result
|
|
s["message"] = "completed"
|
|
|
|
# Close browser window (session preserved in profile via UnitedSCO browser manager)
|
|
try:
|
|
from unitedsco_browser_manager import get_browser_manager as _gbm
|
|
_gbm().quit_driver()
|
|
print("[UnitedDH PreAuth] Browser closed - session preserved in profile")
|
|
except Exception as close_err:
|
|
print(f"[UnitedDH PreAuth] Could not close browser (non-fatal): {close_err}")
|
|
|
|
asyncio.create_task(_remove_session_later(sid, 60))
|
|
return result
|
|
|
|
except Exception as e:
|
|
if s:
|
|
s["status"] = "error"
|
|
s["message"] = f"worker exception: {e}"
|
|
await cleanup_session(sid)
|
|
return {"status": "error", "message": f"worker exception: {e}"}
|
|
|
|
|
|
def submit_otp(sid: str, otp: str) -> Dict[str, Any]:
|
|
s = sessions.get(sid)
|
|
if not s:
|
|
return {"status": "error", "message": "session not found"}
|
|
if s.get("status") != "waiting_for_otp":
|
|
return {"status": "error", "message": f"session not waiting for otp (state={s.get('status')})"}
|
|
s["otp_value"] = otp
|
|
s["last_activity"] = time.time()
|
|
try:
|
|
s["otp_event"].set()
|
|
except Exception:
|
|
pass
|
|
return {"status": "ok", "message": "otp accepted"}
|
|
|
|
|
|
def get_session_status(sid: str) -> Dict[str, Any]:
|
|
s = sessions.get(sid)
|
|
if not s:
|
|
return {"status": "not_found"}
|
|
return {
|
|
"session_id": sid,
|
|
"status": s.get("status"),
|
|
"message": s.get("message"),
|
|
"created_at": s.get("created_at"),
|
|
"last_activity": s.get("last_activity"),
|
|
"result": s.get("result") if s.get("status") in ("completed", "error") else None,
|
|
}
|