Browse Source

Bug the user if they do not have chromedriver installed

pull/31/head
Wesley Kerfoot 6 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))) log_file.write("{0}\n".format(dumps(structured_content)))
return (log_file, log) 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 import time
from sys import stderr, exit
from selenium.webdriver.chrome.options import Options from selenium.webdriver.chrome.options import Options
from seleniumrequests import Chrome from seleniumrequests import Chrome
from selenium.common.exceptions import NoSuchElementException from selenium.common.exceptions import NoSuchElementException
from .common import no_chrome_driver
def login(user_email_address, def login(user_email_address,
user_password, user_password,
@ -28,7 +30,15 @@ def login(user_email_address,
chrome_options.add_argument('--disable-gpu') chrome_options.add_argument('--disable-gpu')
chrome_options.add_argument('log-level=2') 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.implicitly_wait(10)
driver.get("https://facebook.com") driver.get("https://facebook.com")

Loading…
Cancel
Save