From 1502feef4ba703bca98aa7b7a9d8e2803b3c25ca Mon Sep 17 00:00:00 2001 From: wes Date: Wed, 22 May 2019 21:52:04 -0400 Subject: [PATCH] More cleanup --- deletefb/tools/common.py | 3 +++ deletefb/tools/likes.py | 8 ++++++++ deletefb/tools/login.py | 12 ++++++------ deletefb/tools/wall.py | 3 +-- 4 files changed, 18 insertions(+), 8 deletions(-) create mode 100644 deletefb/tools/common.py create mode 100644 deletefb/tools/likes.py diff --git a/deletefb/tools/common.py b/deletefb/tools/common.py new file mode 100644 index 0000000..92405c4 --- /dev/null +++ b/deletefb/tools/common.py @@ -0,0 +1,3 @@ +from selenium.common.exceptions import NoSuchElementException, StaleElementReferenceException + +SELENIUM_EXCEPTIONS = (NoSuchElementException, StaleElementReferenceException) diff --git a/deletefb/tools/likes.py b/deletefb/tools/likes.py new file mode 100644 index 0000000..80c6b0a --- /dev/null +++ b/deletefb/tools/likes.py @@ -0,0 +1,8 @@ +from selenium.webdriver.common.action_chains import ActionChains +from .common import SELENIUM_EXCEPTIONS + +def unlike_pages(driver): + """ + Unlike all pages + """ + return diff --git a/deletefb/tools/login.py b/deletefb/tools/login.py index 62ad2a3..d3d7663 100644 --- a/deletefb/tools/login.py +++ b/deletefb/tools/login.py @@ -36,14 +36,14 @@ def login(user_email_address, password = "pass" login = "loginbutton" - emailelement = driver.find_element_by_name(email) - passwordelement = driver.find_element_by_name(password) + email_element = driver.find_element_by_name(email) + password_element = driver.find_element_by_name(password) - emailelement.send_keys(user_email_address) - passwordelement.send_keys(user_password) + email_element.send_keys(user_email_address) + password_element.send_keys(user_password) - loginelement = driver.find_element_by_id(login) - loginelement.click() + login_element = driver.find_element_by_id(login) + login_element.click() if "Two-factor authentication" in driver.page_source: # Allow time to enter 2FA code diff --git a/deletefb/tools/wall.py b/deletefb/tools/wall.py index d187bb6..3005dd1 100644 --- a/deletefb/tools/wall.py +++ b/deletefb/tools/wall.py @@ -1,10 +1,9 @@ import time from selenium.webdriver.common.action_chains import ActionChains -from selenium.common.exceptions import NoSuchElementException, StaleElementReferenceException +from .common import SELENIUM_EXCEPTIONS MAX_POSTS = 5000 -SELENIUM_EXCEPTIONS = (NoSuchElementException, StaleElementReferenceException) def delete_posts(driver): """