diff --git a/deletefb/deletefb.py b/deletefb/deletefb.py index 30e8f52..42ab985 100755 --- a/deletefb/deletefb.py +++ b/deletefb/deletefb.py @@ -6,7 +6,7 @@ import time from seleniumrequests import Chrome from selenium.webdriver.common.action_chains import ActionChains from selenium.webdriver.chrome.options import Options - +from selenium.common.exceptions import NoSuchElementException, StaleElementReferenceException MAX_POSTS = 5000 @@ -103,30 +103,31 @@ def delete_posts(user_email_address, driver.get(user_profile_url) for _ in range(MAX_POSTS): - try: - post_button_sel = "_4xev" - timeline_element = driver.find_element_by_class_name(post_button_sel) - actions = ActionChains(driver) - actions.move_to_element(timeline_element).click().perform() - - menu = driver.find_element_by_css_selector("#globalContainer > div.uiContextualLayerPositioner.uiLayer > div") - actions.move_to_element(menu).perform() + post_button_sel = "_4xev" + while True: try: - delete_button = menu.find_element_by_xpath("//a[@data-feed-option-name=\"FeedDeleteOption\"]") - - # FIXME Using a bare except here to avoid having to handle all possible exceptions - except: - delete_button = menu.find_element_by_xpath("//a[@data-feed-option-name=\"HIDE_FROM_TIMELINE\"]") - - actions.move_to_element(delete_button).click().perform() - - confirmation_button = driver.find_element_by_class_name("layerConfirm") - - # Facebook would not let me get focus on this button without some custom JS - driver.execute_script("arguments[0].click();", confirmation_button) - except: - pass + timeline_element = driver.find_element_by_class_name(post_button_sel) + actions = ActionChains(driver) + actions.move_to_element(timeline_element).click().perform() + + menu = driver.find_element_by_css_selector("#globalContainer > div.uiContextualLayerPositioner.uiLayer > div") + actions.move_to_element(menu).perform() + + try: + delete_button = menu.find_element_by_xpath("//a[@data-feed-option-name=\"FeedDeleteOption\"]") + except (NoSuchElementException, StaleElementReferenceException): + delete_button = menu.find_element_by_xpath("//a[@data-feed-option-name=\"HIDE_FROM_TIMELINE\"]") + + actions.move_to_element(delete_button).click().perform() + confirmation_button = driver.find_element_by_class_name("layerConfirm") + + # Facebook would not let me get focus on this button without some custom JS + driver.execute_script("arguments[0].click();", confirmation_button) + except (NoSuchElementException, StaleElementReferenceException): + continue + else: + break # Required to sleep the thread for a bit after using JS to click this button time.sleep(5)