Grab all URLs from page like page

deletefb/deletefb.py

@@ -114,7 +114,7 @@ def run_delete():
         )
 
     elif args.mode == "unlike_pages":
-        unlike_pages(driver)
+        unlike_pages(driver, args.profile_url)
     else:
         print("Please enter a valid mode")
         sys.exit(1)

deletefb/tools/likes.py

@@ -2,13 +2,14 @@ from selenium.webdriver.common.by import By
 from selenium.webdriver.common.action_chains import ActionChains
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
+from time import sleep
 
 from .common import SELENIUM_EXCEPTIONS, archiver, logger
 
 LOG = logger(__name__)
 
 
-def load_likes(driver):
+def load_likes(driver, profile_url):
     """
     Loads the page that lists all pages you like
 
@@ -19,24 +20,19 @@ def load_likes(driver):
         None
     """
 
-    driver.refresh()
+    driver.get("{0}/likes_all".format(profile_url))
-    driver.get("https://www.facebook.com/pages/?category=liked")
 
     wait = WebDriverWait(driver, 20)
 
     try:
         wait.until(
-            EC.presence_of_element_located((By.XPATH, "//button/div/div[text()='Liked']"))
+            EC.presence_of_element_located((By.CSS_SELECTOR, ".PageLikeButton"))
-        )
-
-        wait.until(
-            EC.presence_of_element_located((By.XPATH, "//button/div/i[@aria-hidden=\"true\"]"))
         )
     except SELENIUM_EXCEPTIONS:
         LOG.exception("Traceback of load_likes")
         return
 
-def unlike_pages(driver):
+def unlike_pages(driver, profile_url):
     """
     Unlike all pages
 
@@ -49,40 +45,20 @@ def unlike_pages(driver):
 
     like_log, archive_likes = archiver("likes")
 
+    wait = WebDriverWait(driver, 20)
     actions = ActionChains(driver)
 
-    load_likes(driver)
+    load_likes(driver, profile_url)
 
-    pages_list = driver.find_element_by_css_selector("#all_liked_pages")
+    pages = driver.find_elements_by_xpath("//li//div/div/a[contains(@class, 'lfloat')]")
 
-    actions.move_to_element(pages_list).perform()
+    actions = ActionChains(driver)
-
-    unlike_buttons = pages_list.find_elements_by_xpath("//button/div/div[text()='Liked']/../..")
-
-    while unlike_buttons:
-        for button in unlike_buttons:
-            try:
-                if "Liked" in button.text:
-                    page_name = button.find_element_by_xpath("./../..").text.split("\n")[0]
-
-                    driver.execute_script("arguments[0].click();", button)
-
-                    archive_likes(page_name)
 
-                    print("{0} was unliked".format(page_name))
+    page_urls = [page.get_attribute("href").replace("www", "mobile") for page in pages]
 
-            except SELENIUM_EXCEPTIONS:
+    for url in page_urls:
-                continue
+        driver.get(url)
 
-        load_likes(driver)
-        try:
-            pages_list = driver.find_element_by_css_selector("#all_liked_pages")
-            actions.move_to_element(pages_list).perform()
-            unlike_buttons = pages_list.find_elements_by_xpath("//button")
-            if not unlike_buttons:
-                break
-        except SELENIUM_EXCEPTIONS:
-            break
 
     # Explicitly close the log file when we're done with it
     like_log.close()