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,8 +103,10 @@ def delete_posts(user_email_address,
     driver.get(user_profile_url)
 
     for _ in range(MAX_POSTS):
-        try:
         post_button_sel = "_4xev"
+
+        while True:
+            try:
                 timeline_element = driver.find_element_by_class_name(post_button_sel)
                 actions = ActionChains(driver)
                 actions.move_to_element(timeline_element).click().perform()
@@ -114,19 +116,18 @@ def delete_posts(user_email_address,
 
                 try:
                     delete_button = menu.find_element_by_xpath("//a[@data-feed-option-name=\"FeedDeleteOption\"]")
-
+                except (NoSuchElementException, StaleElementReferenceException):
-            # 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:
+            except (NoSuchElementException, StaleElementReferenceException):
-            pass
+                continue
+            else:
+                break
 
         # Required to sleep the thread for a bit after using JS to click this button
         time.sleep(5)