Selenium InvalidSelectorException 오류 해결하기

소개

Selenium을 사용하여 웹 자동화를 수행하다 보면 'InvalidSelectorException' 오류를 자주 경험할 수 있습니다. 이 오류는 주로 잘못된 셀렉터를 사용하여 요소에 접근하려 할 때 발생합니다. 오늘은 이 오류가 어떤 상황에서 발생하는지, 그리고 이를 어떻게 해결할 수 있는지 알아보겠습니다.

에러 발생 예시 코드

먼저, 'InvalidSelectorException' 에러가 발생할 수 있는 간단한 예시 코드를 살펴봅시다.

from selenium import webdriver
from selenium.common.exceptions import InvalidSelectorException

# 웹 드라이버 초기화
driver = webdriver.Chrome()

# 웹사이트 열기
driver.get("http://example.com")

try:
    # 잘못된 XPath 사용
    element = driver.find_element_by_xpath("//div[@class='example'")  # Syntax Error
except InvalidSelectorException as e:
    print("Error: Invalid Selector -", e)
finally:
    driver.quit()

에러 해결 방법

1. XPath 문법 검토하기

가장 일반적인 원인은 XPath 문법 오류입니다. 올바른 문법을 사용하고 있는지 확인하세요. 닫는 괄호가 누락되거나 잘못된 문자가 포함되어 있지 않은지 체크해보세요.

from selenium import webdriver

# 웹 드라이버 초기화
driver = webdriver.Chrome()

# 웹사이트 열기
driver.get("http://example.com")

# 올바른 XPath 사용
element = driver.find_element_by_xpath("//div[@class='example']")  # Syntax Correct
print(element.text)
driver.quit()

2. CSS 셀렉터 사용하기

XPath 외에도 CSS 셀렉터를 사용할 수 있습니다. 경우에 따라 CSS 셀렉터가 간단하고 오류 발생 확률이 낮습니다.

from selenium import webdriver

# 웹 드라이버 초기화
driver = webdriver.Chrome()

# 웹사이트 열기
driver.get("http://example.com")

# CSS 셀렉터 사용
element = driver.find_element_by_css_selector("div.example")  # CSS Selector
print(element.text)
driver.quit()

3. 요소 확인하기

특정 요소를 찾기 전에 DOM이 로드되었는지, 해당 요소가 실제로 페이지에 존재하는지 확인해야 합니다. 페이지가 완전히 로드되는 시간을 기다리는 방법도 좋은 예외 처리가 될 수 있습니다.

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# 웹 드라이버 초기화
driver = webdriver.Chrome()

# 웹사이트 열기
driver.get("http://example.com")

# 요소가 로드되기를 기다림
try:
    element = WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.XPATH, "//div[@class='example']"))
    )
    print(element.text)
finally:
    driver.quit()

마무리

이번 블로그 포스트에서는 Selenium의 'InvalidSelectorException' 오류와 그 해결 방법을 살펴보았습니다. XPath 문법을 점검하고, CSS 셀렉터를 활용하며, 요소의 로드 상태를 체크함으로써 이 오류를 피할 수 있습니다. 웹 자동화가 순간의 실수로 데이터 수집을 막는 일이 없도록 주의하시면 좋겠습니다. 즐거운 코딩 되세요!