Selenium: Python으로 시작하기

셀레니움은 웹 크롤링을 하거나, 작업을 자동화하기 위해 자주 쓰이곤 한다. 또한, 이 기능을 활용하여 구현한 웹의 테스트 케이스를 자동으로 실행 가능하도록 만들 수 있다. 이를 위해 파이썬으로 셀레니움을 사용하기 위한 방법에 대해 정리하고자 한다.

 

설치

1. selenium

 파이썬의 경우 pip가 있으므로 간단하게 설치할 수 있다. 설치 환경은 Pycharm을 사용하고 venv를 사용하는 프로젝트를 기준으로 설명하고자 한다.

 Pycharm 하단에 Terminal을 클릭 한 후에, pip install selenium을 입력하면 간단히 설치할 수 있다.

 

2. web-driver

 각자 사용하고자 하는 웹에 따라, 웹 드라이버 파일을 프로젝트 폴더 내에 위치시켜야 한다. 또한 웹 드라이버와 현재 로컬에 사용중인 웹의 버전이 일치하여야 한다.

 크롬의 경우 우측 상단의 점 세개 버튼을 클릭 후, 도움말 → Chrome 정보를 통해 버전을 확인할 수 있다. 사용 중인 웹 브라우저의 버전을 알았다면, 일치하는 버전의 드라이버를 다운하면 된다.

 다운이 된 드라이버를 프로젝트 폴더 내에 위치시키면 된다. 보통 Pycharm 프로젝트의 경우 윈도우를 기준으로, C:\Users\[UserName]\PychamProcjests\[ProjectName]이 경로이다. 따라서 해당 경로에 웹 드라이버를 위치시키면 된다.

 

3. 설치된 크롬 활용하기

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager

driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)

 언제부터 생겼는지 모르겠지만, 위와 같이 webdriver_manager를 활용하면 번거롭게 web-driver를 다운받지 않아도 호스트상에 설치된 크롬을 활용하여 셀레니움을 실행할 수 있다. built-in 라이브러리가 아니기에, pip install webdriver_manager를 통해 별도로 설치하여야 한다.

 

간단한 테스트

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager


class DefaultTest:
    def __init__(self):
        self.driver = webdriver.Chrome()
        # webdriver_manager를 활용하는 경우.
		# self.driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
        
    def teardown(self):
        self.driver.quit()

    def auto_test(self):
        self.driver.get("https://www.google.com/")
        self.driver.set_window_size(974, 1040)
        self.driver.find_element(By.NAME, "q").send_keys("test")
        self.driver.find_element(By.NAME, "q").send_keys(Keys.ENTER)


if __name__ == '__main__':
    test = DefaultTest()
    test.auto_test()

 위의 코드를 실행하였을 때, 구글에 test라는 키워드를 검색한다면 정상적으로 사용 가능한 상태이다.