Merge pull request YoongiKim#36 from ranbims/ranbi/fix_web_drive

fix: This version of ChromeDriver only supports Chrome version 87
imsadiku · Jul 1, 2021 · dc0982b · dc0982b
2 parents 9025101 + 90cf47e
commit dc0982b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/collect_links.py b/collect_links.py
@@ -24,6 +24,7 @@
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.chrome.options import Options
+from webdriver_manager.chrome import ChromeDriverManager
 import os.path as osp
 
 
@@ -51,7 +52,7 @@ def __init__(self, no_gui=False):
         chrome_options.add_argument('--disable-dev-shm-usage')
         if no_gui:
             chrome_options.add_argument('--headless')
-        self.browser = webdriver.Chrome(executable, chrome_options=chrome_options)
+        self.browser = webdriver.Chrome(ChromeDriverManager().install(), chrome_options=chrome_options)
 
         browser_version = 'Failed to detect version'
         chromedriver_version = 'Failed to detect version'

diff --git a/requirements.txt b/requirements.txt
@@ -3,3 +3,4 @@ chardet
 idna
 requests
 selenium
+webdriver-manager