Browse Source

Bug the user if they do not have chromedriver installed

pull/31/head
Wesley Kerfoot 5 years ago
parent
commit
fddd373d14
  1. 6
      deletefb/tools/common.py
  2. 12
      deletefb/tools/login.py

6
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
"""

12
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")

Loading…
Cancel
Save