[Python] Selenium操作Firefox

安裝selenium

pip install selenium==3.14.0

安裝geckodriver

# 取得合適的版本
wget https://github.com/mozilla/geckodriver/releases/download/v0.21.0/geckodriver-v0.21.0-linux64.tar.gz
# 解壓縮
tar zxvf geckodriver-v0.21.0-linux64.tar.gz
sudo mv geckodriver /usr/bin

設定firefox環境參數

export DISPLAY=:0.0

現在可以來測試selenium效果，打開瀏覽器到google，截圖後存檔。若執行成功應該會在同一目錄下產生google擷取畫面。

from selenium import webdriver
browser = webdriver.Firefox()
browser.get('https://www.google.com')
browser.save_screenshot('screenshot_google.png')
browser.close()