Selenium: webdriver_manager 활용하기

오랜만에 셀레니움을 활용할 일이 있어 기존처럼, 크롬 드라이버를 로컬에 직접 다운로드하여 사용하였다. 그런데 작성한 코드를 exe로 만들어 다른 사람에게 배포할 일이 생겼다. exe를 만들고, 크롬 드라이버 버전이 달라 다른 환경에서 실행 시 호환되지 않는 것이 있어 어떻게 하지 고민하면서 검색해보다가 좋은 라이브러리가 있어서 다뤄보고자 한다.

 

webdriver_manager?

webdriver-manager

 호스트에 설치 된 `크롬`, `파이어폭스`, `오페라` 등을 활용하여 각 브라우저 별 드라이버를 다운로드하지 않더라도 다운로드해서 프로젝트에 포함한 것과 동일하게 사용 가능한 환경을 만들어 준다. 기본적인 예시는 위 사이트에서 참고해서 볼 수 있다.

 

간단한 예제

import os
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager

options = Options()
# VGA 가속화를 끔.
options.add_argument('disable-gpu')
# 브라우저 창을 숨김.
options.add_argument('--headless')
# --headless 상태에서, agent 정보를 표기하기 위해 필요하다고 함.
options.add_argument('user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Whale/3.12.129.46 Safari/537.36')
# 불필요한 로그를 출력하지 않도록 함.
os.environ['WDM_LOG_LEVEL'] = '0'

# driverManager를 통해 driver초기화.
driver = webdriver.Chrome(options=options, service=Service(ChromeDriverManager().install()))

 위와 같이 적절한 옵션을 준 후에, 실행하면 driver가 정상적으로 초기화 되는 것을 확인할 수 있다. 

 

불필요한 로그 출력 없애기

====== WebDriver manager ======
Current google-chrome version is 96.0.4664
Get LATEST chromedriver version for 96.0.4664 google-chrome
Driver [C:\Users\test\.wdm\drivers\chromedriver\win32\96.0.4664.45\chromedriver.exe] found in cache

 `os.environ['WDM_LOG_LEVEL'] = '0'`을 넣지 않으면 위와 같이 불필요한 로그가 출력된다. 이를 출력하지 않기 위해 여러 곳을 검색하다가, webdriver 자체의 옵션이 아닌, os의 로그 레벨을 설정하니 해결되었다.😀