wes
/
DeleteFB
1
0
Fork 0
Code Issues 26 Pull Requests 3 Projects Releases 12 Wiki Activity
Browse Source

cleanly shut down selenium when an exception happens

pull/120/head
Wesley Kerfoot 5 years ago
parent
486c4687d9
commit
b564aa564b
3 changed files with 90 additions and 71 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 37
      deletefb/deletefb.py
  2. 10
      deletefb/quit_driver.py
  3. 114
      deletefb/tools/login.py

37
deletefb/deletefb.py
View File

@ -6,6 +6,7 @@ from .tools.login import login
from .tools.wall import delete_posts from .tools.wall import delete_posts
from .tools.conversations import traverse_conversations from .tools.conversations import traverse_conversations
from .tools.comments import delete_comments from .tools.comments import delete_comments
from .quit_driver import quit_driver_and_reap_children
import argparse import argparse
import getpass import getpass
@ -116,22 +117,26 @@ def run_delete():
chrome_binary_path=args.chromebin chrome_binary_path=args.chromebin
) )
if args.mode == "wall": try:
delete_posts( if args.mode == "wall":
driver, delete_posts(
args.profile_url, driver,
year=args.year args.profile_url,
) year=args.year
)
elif args.mode == "unlike_pages":
unlike_pages(driver, args.profile_url) elif args.mode == "unlike_pages":
unlike_pages(driver, args.profile_url)
elif args.mode == "conversations":
traverse_conversations(driver, year=args.year) elif args.mode == "conversations":
traverse_conversations(driver, year=args.year)
else:
print("Please enter a valid mode") else:
sys.exit(1) print("Please enter a valid mode")
sys.exit(1)
except:
if driver:
quit_driver_and_reap_children(driver)
if __name__ == "__main__": if __name__ == "__main__":
run_delete() run_delete()

10
deletefb/quit_driver.py
View File

@ -0,0 +1,10 @@
import os
def quit_driver_and_reap_children(driver):
driver.quit()
try:
pid = True
while pid:
pid = os.waitpid(-1, os.WNOHANG)
except ChildProcessError:
pass

114
deletefb/tools/login.py
View File

@ -1,6 +1,7 @@
from .chrome_driver import get_webdriver, setup_selenium from .chrome_driver import get_webdriver, setup_selenium
from selenium.common.exceptions import NoSuchElementException from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.chrome.options import Options from selenium.webdriver.chrome.options import Options
from ..quit_driver import quit_driver_and_reap_children
import time import time
@ -42,59 +43,62 @@ def login(user_email_address,
driver_path = get_webdriver(chrome_binary_path) driver_path = get_webdriver(chrome_binary_path)
driver = setup_selenium(driver_path, chrome_options) driver = setup_selenium(driver_path, chrome_options)
driver.implicitly_wait(10)
driver.get("https://www.facebook.com/login/device-based/regular/login/?login_attempt=1&lwv=110")
email = "email"
password = "pass"
login_button = "loginbutton"
approvals_code = "approvals_code"
driver.find_element_by_name(email).send_keys(user_email_address)
driver.find_element_by_name(password).send_keys(user_password)
driver.find_element_by_id(login_button).click()
# Defaults to no 2fa
has_2fa = False
try: try:
# If this element exists, we've reached a 2FA page driver.implicitly_wait(10)
driver.find_element_by_xpath("//form[@class=\"checkpoint\"]")
driver.find_element_by_xpath("//input[@name=\"approvals_code\"]") driver.get("https://www.facebook.com/login/device-based/regular/login/?login_attempt=1&lwv=110")
has_2fa = True
except NoSuchElementException: email = "email"
has_2fa = "two-factor authentication" in driver.page_source.lower() or has_2fa password = "pass"
login_button = "loginbutton"
if has_2fa: approvals_code = "approvals_code"
print("""
Two-Factor Auth is enabled. driver.find_element_by_name(email).send_keys(user_email_address)
Please file an issue at https://github.com/weskerfoot/DeleteFB/issues if you run into any problems driver.find_element_by_name(password).send_keys(user_password)
""") driver.find_element_by_id(login_button).click()
if two_factor_token and has_2fa: # Defaults to no 2fa
twofactorelement = driver.find_element_by_name(approvals_code) has_2fa = False
twofactorelement.send_keys(two_factor_token)
try:
# Submits after the code is passed into the form, does not validate 2FA code. # If this element exists, we've reached a 2FA page
contelement = driver.find_element_by_id("checkpointSubmitButton") driver.find_element_by_xpath("//form[@class=\"checkpoint\"]")
contelement.click() driver.find_element_by_xpath("//input[@name=\"approvals_code\"]")
has_2fa = True
# Defaults to saving this new browser, this occurs on each new automated login. except NoSuchElementException:
save_browser = driver.find_element_by_id("checkpointSubmitButton") has_2fa = "two-factor authentication" in driver.page_source.lower() or has_2fa
save_browser.click()
elif has_2fa: if has_2fa:
# Allow time to enter 2FA code print("""
print("Pausing to enter 2FA code") Two-Factor Auth is enabled.
time.sleep(35) Please file an issue at https://github.com/weskerfoot/DeleteFB/issues if you run into any problems
print("Continuing execution") """)
else:
pass if two_factor_token and has_2fa:
twofactorelement = driver.find_element_by_name(approvals_code)
# block until we have reached the main page twofactorelement.send_keys(two_factor_token)
# print a message warning the user
while driver.current_url != "https://www.facebook.com/": # Submits after the code is passed into the form, does not validate 2FA code.
print("Execution blocked: Please navigate to https://www.facebook.com to continue") contelement = driver.find_element_by_id("checkpointSubmitButton")
time.sleep(5) contelement.click()
return driver # Defaults to saving this new browser, this occurs on each new automated login.
save_browser = driver.find_element_by_id("checkpointSubmitButton")
save_browser.click()
elif has_2fa:
# Allow time to enter 2FA code
print("Pausing to enter 2FA code")
time.sleep(35)
print("Continuing execution")
else:
pass
# block until we have reached the main page
# print a message warning the user
while driver.current_url != "https://www.facebook.com/":
print("Execution blocked: Please navigate to https://www.facebook.com to continue")
time.sleep(5)
return driver
except:
quit_driver_and_reap_children(driver)

Write Preview
Loading…
Cancel
Save