본문으로 바로가기

[Python] Selenium 태그 체크 함수

category 개발/파이썬 2019. 1. 31. 21:57

원하는 태그가 없을때 다른 작업을 수행하기 위해 만든 함수입니다. 

태그 체크 함수


# 텍스트 링크 체크 함수
def check_exists_by_link_text(text):
try:
driver.find_element_by_link_text(text)
except NoSuchElementException:
return False
return True

# xpath 체크 함수
def check_exists_by_xpath(xpath):
try:
driver.find_element_by_xpath(xpath)
except NoSuchElementException:
return False
return True

사용 예시


if check_exists_by_xpath("//iframe"):
self.alertMessage("보안 프로그램을 해제하고 클릭하세요.")



if check_exists_by_link_text("로그아웃"):
driver.find_element_by_link_text("로그아웃").click()