diff --git a/deletefb/tools/common.py b/deletefb/tools/common.py index 0bf3df1..4b6cbea 100644 --- a/deletefb/tools/common.py +++ b/deletefb/tools/common.py @@ -30,3 +30,9 @@ def archiver(category): log_file.write("{0}\n".format(dumps(structured_content))) return (log_file, log) + + +no_chrome_driver = """ +You need to install the chromedriver for Selenium\n +Please see this link https://github.com/weskerfoot/DeleteFB#how-to-use-it\n +""" diff --git a/deletefb/tools/login.py b/deletefb/tools/login.py index 2dea088..0a1ddd7 100644 --- a/deletefb/tools/login.py +++ b/deletefb/tools/login.py @@ -1,8 +1,10 @@ import time +from sys import stderr, exit from selenium.webdriver.chrome.options import Options from seleniumrequests import Chrome from selenium.common.exceptions import NoSuchElementException +from .common import no_chrome_driver def login(user_email_address, user_password, @@ -28,7 +30,15 @@ def login(user_email_address, chrome_options.add_argument('--disable-gpu') chrome_options.add_argument('log-level=2') - driver = Chrome(options=chrome_options) + try: + driver = Chrome(options=chrome_options) + except Exception as e: + # The user does not have chromedriver installed + # Tell them to install it + stderr.write(str(e)) + stderr.write(no_chrome_driver) + exit(1) + driver.implicitly_wait(10) driver.get("https://facebook.com")