Add explicit exceptions

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)