파이썬 Selenium 파일 다운로드 폴더 지정과 완료 대기 방법

⚙️ Selenium에서 다운로드 폴더 지정하기

Selenium을 사용해 자동화된 다운로드 작업을 수행할 때 가장 먼저 해야 할 일은 다운로드 경로를 미리 지정하는 것입니다.
기본 설정 그대로라면 다운로드되는 파일이 운영체제의 기본 다운로드 폴더에 저장되기 때문에, 프로젝트 단위로 파일을 관리하기 어렵고 자동화된 후처리 과정에도 혼란이 생깁니다.

예를 들어 크롬 드라이버를 사용하는 경우에는 ChromeOptions를 통해 prefs 속성을 지정하여 다운로드 위치와 동작을 제어할 수 있습니다.
‘download.default_directory’를 원하는 경로로 설정하면, 다운로드된 파일이 항상 해당 위치로 저장됩니다.
또한 ‘download.prompt_for_download’를 False로 지정하면 다운로드 시 확인창이 뜨지 않아 완전히 무인화된 자동화를 구현할 수 있습니다.

from selenium import webdriver

options = webdriver.ChromeOptions()
prefs = {
    "download.default_directory": r"C:\\my_downloads",
    "download.prompt_for_download": False,
    "directory_upgrade": True
}
options.add_experimental_option("prefs", prefs)
driver = webdriver.Chrome(options=options)

위와 같은 방식으로 설정을 적용하면, 파일이 항상 C:\my_downloads 경로로 다운로드됩니다.
이 설정은 크롬뿐 아니라 파이어폭스에서도 프로필 옵션을 활용해 비슷하게 지정할 수 있으며, 브라우저마다 약간의 설정 차이만 있을 뿐 기본적인 원리는 동일합니다.

📂 임시 확장자 파일로 다운로드 완료 확인하기

폴더 경로를 지정했다면 이제 남은 문제는 다운로드가 끝났는지 확인하는 것입니다.
대부분의 브라우저는 다운로드 중인 파일에 임시 확장자를 붙입니다.
예를 들어 크롬은 .crdownload, 파이어폭스는 .part 확장자를 사용합니다.
이러한 임시 파일이 사라져야 다운로드가 완료된 것으로 간주할 수 있습니다.

따라서 자동화 코드에서는 특정 폴더를 주기적으로 확인(polling)하면서 임시 확장자가 사라질 때까지 기다린 뒤 후속 로직을 실행하면 안전합니다.
이는 단순히 time.sleep()을 사용하는 방식보다 훨씬 안정적이며, 네트워크 속도나 파일 크기에 따라 유동적으로 대기 시간을 조절할 수 있습니다.

import os, time

def wait_for_downloads(download_dir, timeout=60):
    start_time = time.time()
    while True:
        files = os.listdir(download_dir)
        # 임시 확장자가 남아있다면 아직 다운로드 중
        if not any(f.endswith((".crdownload", ".part")) for f in files):
            break
        if time.time() - start_time > timeout:
            raise TimeoutError("다운로드가 시간 내에 완료되지 않았습니다.")
        time.sleep(1)

wait_for_downloads(r"C:\\my_downloads")

이 방식은 특정 파일 이름을 지정하지 않고 폴더 내 전체 상태를 체크하기 때문에, 여러 개의 파일을 동시에 다운로드할 때에도 유용합니다.
다만, 대용량 파일의 경우 다운로드 시간이 길어질 수 있으므로 timeout 값을 충분히 설정하는 것이 좋습니다.

💻 파이썬 예제 코드로 구현해보기

앞에서 설명한 폴더 지정과 다운로드 완료 대기 방법을 실제로 하나의 코드로 합쳐보면 훨씬 이해하기 쉽습니다.
아래 예제는 크롬 브라우저를 이용해 특정 파일을 다운로드하고, 다운로드가 끝날 때까지 기다리는 과정을 모두 포함하고 있습니다.

from selenium import webdriver
import os, time

DOWNLOAD_DIR = r"C:\\my_downloads"

def wait_for_downloads(download_dir, timeout=60):
    start_time = time.time()
    while True:
        files = os.listdir(download_dir)
        if not any(f.endswith((".crdownload", ".part")) for f in files):
            break
        if time.time() - start_time > timeout:
            raise TimeoutError("다운로드가 시간 내에 완료되지 않았습니다.")
        time.sleep(1)

options = webdriver.ChromeOptions()
prefs = {
    "download.default_directory": DOWNLOAD_DIR,
    "download.prompt_for_download": False,
    "directory_upgrade": True
}
options.add_experimental_option("prefs", prefs)
driver = webdriver.Chrome(options=options)

driver.get("https://example.com/sample.pdf")
wait_for_downloads(DOWNLOAD_DIR)
print("다운로드 완료!")
driver.quit()

위 코드를 실행하면 지정한 URL에서 PDF 파일을 다운로드한 후, 해당 파일이 완전히 내려받아질 때까지 기다렸다가 “다운로드 완료!”라는 메시지를 출력합니다.
이 방식은 확장자가 임시 상태에서 정상 파일로 바뀌는 시점을 기준으로 하기 때문에 매우 안정적입니다.

💬 실제 환경에서는 네트워크 상황, 브라우저 버전, 파일 크기에 따라 다운로드 시간이 달라질 수 있으므로 timeout 값을 상황에 맞게 조정하는 것이 좋습니다.

또한, 여러 개의 파일을 연속적으로 다운로드해야 한다면 각 다운로드마다 wait_for_downloads() 함수를 호출해 주는 것이 안전합니다.
이렇게 하면 예기치 못한 누락이나 중복 실행 문제를 줄일 수 있습니다.

🛠️ 크롬과 파이어폭스 설정 차이

Selenium에서 다운로드 폴더를 지정하는 방법은 브라우저마다 조금씩 다릅니다.
대부분은 비슷한 원리를 따르지만, 옵션 이름이나 접근 방식이 다르기 때문에 반드시 각 브라우저의 특성을 이해해야 합니다.
대표적으로 많이 사용되는 크롬(Chrome)과 파이어폭스(Firefox)의 설정 차이를 살펴보겠습니다.

🌐 크롬에서의 다운로드 설정

크롬은 ChromeOptions의 prefs 속성을 통해 다운로드 경로를 제어합니다.
앞서 살펴본 것처럼 download.default_directory, download.prompt_for_download 등의 옵션을 사용하면 됩니다.

🦊 파이어폭스에서의 다운로드 설정

파이어폭스는 FirefoxProfile을 활용해 다운로드 동작을 제어합니다.
이때 MIME 타입을 미리 지정해야 다운로드 팝업 없이 자동 저장됩니다.
예를 들어 CSV나 PDF를 자동 다운로드하려면 browser.helperApps.neverAsk.saveToDisk 속성에 해당 MIME 타입을 나열해야 합니다.

from selenium import webdriver

profile = webdriver.FirefoxProfile()
profile.set_preference("browser.download.folderList", 2)
profile.set_preference("browser.download.dir", r"C:\\my_downloads")
profile.set_preference("browser.helperApps.neverAsk.saveToDisk",
                       "application/pdf,text/csv,application/octet-stream")
profile.set_preference("pdfjs.disabled", True)

driver = webdriver.Firefox(firefox_profile=profile)
driver.get("https://example.com/sample.csv")

이 설정을 적용하면 CSV, PDF 등 지정한 파일은 별도의 대화창 없이 지정된 폴더로 자동 다운로드됩니다.
특히 파이어폭스는 pdfjs.disabled 속성을 True로 설정해 PDF 뷰어를 비활성화하지 않으면 브라우저에서 열리기 때문에 주의가 필요합니다.

🚀 자동화 프로젝트에서의 활용 팁

파일 다운로드 기능을 제대로 구현하면 단순한 웹 크롤링을 넘어서 보다 실무적인 자동화가 가능해집니다.
예를 들어 기업 내부 보고서, 오픈데이터 포털의 자료, 금융기관의 일별 시세 파일 등을 정해진 시간마다 자동으로 수집해 저장할 수 있습니다.
이때 다운로드 경로 지정과 완료 대기 기능은 프로젝트 안정성을 보장하는 핵심 요소가 됩니다.

또한, 파일 다운로드가 포함된 자동화는 후처리 작업과 연결되기 때문에, 다운로드된 파일을 자동으로 읽어 데이터 분석이나 데이터베이스 적재로 이어지는 경우가 많습니다.
따라서 다운로드 완료를 제대로 확인하지 않으면 불완전한 데이터가 저장되거나 분석이 틀어질 수 있습니다.

📌 안정성을 높이는 체크리스트

• 📂다운로드 전용 폴더 경로를 프로젝트 안에 별도로 설정하기
• ⏳임시 확장자(.crdownload, .part) 감지로 다운로드 완료 여부 확인하기
• 🛠️timeout 값은 네트워크 환경에 맞춰 충분히 길게 설정하기
• 🔄여러 파일을 연속 다운로드할 경우 각 파일마다 대기 함수 실행하기
• 🗑️테스트 과정에서 생성된 파일은 주기적으로 삭제해 폴더를 깔끔하게 유지하기

이러한 원칙을 지켜가며 작업을 진행하면 자동화의 완성도를 높일 수 있습니다.
특히 기업 환경에서는 다운로드된 파일을 다른 시스템으로 넘겨야 하는 경우가 많기 때문에, 파일 무결성 확보가 무엇보다 중요합니다.

❓ 자주 묻는 질문 (FAQ)