from selenium import webdriver
from selenium.webdriver.chrome.options import Options
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.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.keys import Keys
import time # 웹 페이지 로딩을 위한 time 모듈 추가

# 브라우저 꺼짐 방지 옵션
chrome_options = Options()
chrome_options.add_experimental_option("detach", True)

# 크롬 드라이버 실행
driver = webdriver.Chrome(options=chrome_options)

# 사이트 접속
driver.get('https://tradedata.go.kr/cts/index.do')

# 최대 10초까지 웹 요소가 나타날 때까지 대기
wait = WebDriverWait(driver, 10)

# '수출입통계' 메뉴가 나타날 때까지 대기
# By.LINK_TEXT를 사용하여 링크 텍스트가 '수출입통계'인 요소를 찾습니다.
menu_export_import = wait.until(EC.presence_of_element_located((By.LINK_TEXT, '수출입통계')))

# ActionChains를 사용하여 마우스 오버로 서브메뉴를 활성화합니다.
actions = ActionChains(driver)
actions.move_to_element(menu_export_import).perform()

# '수출입 실적' 서브메뉴가 클릭 가능할 때까지 대기하고 클릭합니다.
# By.LINK_TEXT를 사용하여 링크 텍스트가 '수출입 실적'인 요소를 찾습니다.
submenu_export_import_performance = wait.until(EC.element_to_be_clickable((By.LINK_TEXT, '수출입 실적')))
submenu_export_import_performance.click()

# 이제 '수출입 실적' 통계 화면에 진입한 상태입니다.

# --- 팝업 처리 로직 ---
# 1. 팝업이 열려 있는지 먼저 확인합니다.
try:
    type_layer = driver.find_element(By.ID, "typeLayer")
    # 'display' CSS 속성이 'block'이면 팝업이 보이는 상태입니다.
    is_layer_visible = type_layer.value_of_css_property("display") == "block"
except:
    # 'typeLayer' 요소를 찾지 못하면 팝업이 열려 있지 않은 것으로 간주합니다.
    is_layer_visible = False

# 2. 팝업이 안 열려 있으면 'selectBtn' 버튼을 클릭합니다.
if not is_layer_visible:
    # 'selectBtn' 클래스를 가진 버튼이 클릭 가능할 때까지 대기하고 클릭합니다.
    select_btn = wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR, "button.selectBtn")))
    select_btn.click()
    # 팝업이 열릴 때까지, 즉 'typeLayer'의 'display' 속성이 'block'이 될 때까지 대기합니다.
    wait.until(lambda d: d.find_element(By.ID, "typeLayer").value_of_css_property("display") == "block")

# 3. '신성질별+국가별' 라디오 버튼 클릭
# XPATH를 사용하여 텍스트 내용이 '신성질별+국가별'인 <label> 요소를 찾습니다.
label = wait.until(
    EC.element_to_be_clickable(
        (By.XPATH, "//label[contains(text(), '신성질별+국가별')]")
    )
)
label.click()

# 세부검색 버튼 클릭
# CSS_SELECTOR를 사용하여 'cate_btn' 클래스와 'ETS0100019Q_btnImexTmprPop' ID를 가진 버튼을 찾습니다.
detail_search_btn = wait.until(
    EC.element_to_be_clickable(
        (By.CSS_SELECTOR, "button.cate_btn#ETS0100019Q_btnImexTmprPop")
    )
)
detail_search_btn.click()

# "신성질명으로 찾기" 링크 클릭
# XPATH를 사용하여 'href' 속성이 '#tabImexTmpr2'이고 텍스트 내용이 '신성질명으로 찾기'인 <a> 요소를 찾습니다.
name_link = wait.until(
    EC.element_to_be_clickable(
        (By.XPATH, "//a[@href='#tabImexTmpr2' and contains(text(), '신성질명으로 찾기')]")
    )
)
name_link.click()

# 검색어 입력란 선택
# CSS_SELECTOR를 사용하여 'ETS0100010Q_popupImexTmprName' ID를 가진 <input> 요소를 찾습니다.
search_input = wait.until(
    EC.element_to_be_clickable(
        (By.CSS_SELECTOR, "input#ETS0100010Q_popupImexTmprName")
    )
)

# 입력란에 값 입력: "13050102"
search_input.clear()  # 혹시 있을지 모르는 기존 값 제거
search_input.send_keys("13050102")
search_input.send_keys(Keys.RETURN) # 엔터 키를 눌러 검색 실행

# 첫 번째 체크박스 클릭 (검색 결과가 바뀔 때마다 다시 찾아야 함)
# CSS_SELECTOR를 사용하여 'type'이 'checkbox'이고 'name'이 'popupImexTmpr5'인 <input> 요소를 찾습니다.
checkbox_13050102 = wait.until(
    EC.element_to_be_clickable(
        (By.CSS_SELECTOR, "input[type='checkbox'][name='popupImexTmpr5']")
    )
)
checkbox_13050102.click()

# 입력란에 값 입력: "13050103"
search_input.clear()  # 이전 검색어 제거
search_input.send_keys("13050103")
search_input.send_keys(Keys.RETURN) # 엔터 키를 눌러 검색 실행

# 두 번째 체크박스 클릭 (검색 결과가 바뀌면 체크박스도 새로 찾아야 함)
# CSS_SELECTOR를 사용하여 ID 'popupImexTmprNameSearch' 하위의 첫 번째 체크박스를 찾습니다.
# 이 셀렉터는 '13050103' 검색 후 새롭게 로드된 결과를 대상으로 합니다.
checkbox_13050103 = wait.until(
    EC.element_to_be_clickable(
        (By.CSS_SELECTOR, "#popupImexTmprNameSearch > tr > td:nth-child(1) > input[type=checkbox]")
    )
)
checkbox_13050103.click()

# 멈춤 상태로 유지 (브라우저가 자동으로 닫히지 않음)
input("진입 완료. 브라우저를 닫으려면 엔터를 누르세요.")

# 작업 완료 후 드라이버 종료 (선택 사항)
# driver.quit()

세부검색에서 '13050102'를 검색해서 '체크박스'를 선택하고 다시 '13050103'을 검색하고 '체크박스'를 선택하려는데 안되네요. 검색해보니 이런 오류라고 하는데 아무리 해도 수정이 안되서 여기까지 하고 답안을 참고해서 마무리했습니다. 근데 이게 정말 왜 안돼는지 궁금해서요. 체크박스 선택하고 '조회'

`StaleElementReferenceException` 오류 설명

StaleElementReferenceException은 다음과 같은 상황에서 발생합니다.

요소 찾기: Selenium이 웹 페이지에서 특정 요소를 성공적으로 찾습니다.
페이지 변경: 하지만 그 사이에 JavaScript 실행이나 AJAX 호출 등으로 인해 웹 페이지의 DOM(Document Object Model)이 변경됩니다. 이 변경은 해당 요소 자체가 사라지거나, 다시 로드되거나, 또는 해당 요소를 감싸는 부모 요소가 재구성되는 것을 의미할 수 있어요.
다시 요소 사용 시도: 변경된 DOM에서 이전에 찾았던 요소를 다시 사용하려고 할 때, Selenium은 이전에 참조했던 요소가 더 이상 현재 DOM에 존재하지 않음을 감지하고 StaleElementReferenceException을 발생시킵니다. "stale"이라는 단어는 "신선하지 않은, 오래된"이라는 뜻을 가지고 있습니다.

당신의 경우, "13050102"를 검색하고 체크박스를 클릭한 후, 새로운 검색어 "13050103"을 입력하고 엔터를 누르면 검색 결과 목록이 새로 로드되면서 DOM이 변경됩니다. 이 때 이전에 찾았던 checkbox 요소(혹은 그와 관련된 부모 요소)의 참조가 유효하지 않게 되면서 오류가 발생한 것입니다.