[python] Selenium - 웹페이지 Alert창 종류와 관리

[python] Selenium - 웹페이지 Alert창 종류 와 관리

#UnexpectedAlertPresentException #NoAlertPresentException 관련 정리

웹페이지 크롤링이나 웹페이지 이벤트 테스트 작업을 진행하다보면  Aler창이 나옵니다.

Alert창이 뜨면 Selenium 작업이 중단되고 대기 상태가 되어 멈추고 다음으로 진행이 안되는 경우가 있습니다. Exception이 발생하면 프로그램이 중단 되기도 합니다.
이때 Alert 창을 닫고 프로그램이 진행 되도록  해야 합니다.

Alert창을 닫거나 처리하는 Selenium 커멘드 들을 알아 보고자 합니다.

웹페이지 Alert창의 종류

 

1. 경고(alert) 창: 중요한 정보를 전달하거나 사용자의 주의를 요구합니다. 일반적으로 제목과 메시지가 포함되어 있으며, 사용자는 확인 버튼을 클릭하여 알림을 닫습니다.

2. 확인(confirm) 창: 사용자에게 작업 수행 여부를 확인하는 메시지를 표시합니다. 제목, 메시지, 확인 버튼 및 취소 버튼이 있으며, 사용자는 확인 또는 취소 버튼을 클릭하여 작업을 결정합니다.

3. 프롬프트(prompt) 창: 사용자로부터 추가 정보를 입력받기 위해 사용됩니다. 제목, 메시지, 입력 필드, 확인 버튼 및 취소 버튼이 있으며, 사용자는 입력 필드에 정보를 입력한 후 확인 또는 취소 버튼을 클릭하여 작업을 수행하거나 취소합니다.

4. 에러(error) 창: 오류 메시지를 표시하여 사용자에게 문제가 발생했음을 알립니다. 일반적으로 오류 메시지와 확인 버튼이 있으며, 사용자는 확인 버튼을 클릭하여 알림을 닫습니다.

 

Selenium CommandType

Selenium에서는 alert 창에 대하여 다음과 같이 "webdriver.CommandType" 을 제공하고 Alert창에 필요한 동작을 구현할 수 있습니다.

# webdriver.CommandType 의 종류
SendKeysToAlert: 경고창(alert)에 키를 보내는 동작
AcceptAlert: 경고창을 수락(확인)하는 동작
DismissAlert: 경고창을 거부(취소)하는 동작
SetAlertCredentials: 경고창의 인증 정보를 설정하는 동작
GetCurrentWindowHandle: 현재 창의 핸들을 가져오는 동작
GetWindowHandles: 모든 창 핸들을 가져오는 동작
SwitchToWindow: 다른 창으로 전환하는 동작

 

Selenium Alert 다루기

Alert 창에 대하여 다음과 같은 방법을 취할 수 있습니다.

Alert창으로 포커스를 전환하고 Alert 창의 내용을 "확인" 또는"취소" 를 할 수 있습니다.

# AcceptAlert : 경고창을 수락(확인)하는 동작 수행
    alert = driver.switch_to.alert
    alert.accept()
    
# DismissAlert: 경고창을 거부(취소)하는 동작 수행
    alert = driver.switch_to.alert
    alert.dismiss()


# SendKeysToAlert: 경고창에 키를 보내는 동작 수행
    alert = driver.switch_to.alert
    alert.send_keys("Hello, World!")
    
# GetCurrentWindowHandle: 현재 창의 핸들을 가져오는 동작 수행
    current_handle = driver.current_window_handle
    print("Current Window Handle:", current_handle)
    
# SwitchToWindow: 다른 창으로 전환하는 동작 수행
    window_handle = "window_handle_to_switch"  # 전환할 창의 핸들
    driver.switch_to.window(window_handle)

 
 

Selenium Alert 창을 만들고 닫는 예제

webdriver.CommandType.AcceptAlert 또는  webdriver.CommandType.DismissAlert  으로  Alert 창을 닫는 예시를 작성 했습니다.

from selenium import webdriver
from selenium.webdriver.common.alert import Alert
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import NoAlertPresentException

driver = webdriver.Chrome()

try:
    driver.get('https://example.com/')
    
    # 경고 창을 띄우는 동작 수행 : Javascript를 실행합니다.
    driver.execute_script("alert('This is an alert!')")
    
    # 경고 창이 있는지 확인
    # alert_present = EC.alert_is_present()
    alert_present = WebDriverWait(driver, 5).until(EC.alert_is_present())
    
    if alert_present :
        # 경고 창에 대해 switch_to.alert를 통해 액세스
        alert = driver.switch_to.alert
        
        # 경고 창을 거부(취소)하는 동작 수행
        alert.dismiss()
        print("Alert dismissed successfully!")
    else:
        print("Alert is not presented")
    
except NoAlertPresentException:
    print("No alert found. Continuing with the script.")
except Exception as err:
    print(" Exception {err=}, {type(err)=}")
finally:
    driver.quit()

 
위의 예시는 Chrome 웹드라이버를 사용하여 웹 페이지에서 경고 창을 띄우고, webdriver.CommandType.DismissAlert 명령 유형을 사용하여 경고 창을 거부(취소)하는 동작을 수행하는 예시입니다.

 코드를 실행하면 경고 창이 뜨고 이후에 자동으로 창이 닫히는 것을 확인할 수 있습니다.

- driver.execute_script("alert('This is an alert!')")  : 열려있는 웹페이지에 JavaScrip를 실행해서 Alert 창을 엽니다.
- alert_present = EC.alert_is_present() 보다
- alert_present  = WebDriverWait(driver, 5).until(EC.alert_is_present()) 가 유용하게 쓰이는 문장이 될 것 같습니다.
-  WebDriverWait(driver, 5) 은 5초 동안 alert이 출력되지 않으면 다음으로 넘어 가도록 합니다.