Minor refactoring and reformatting :)

deletefb/deletefb.py

@@ -1,78 +1,100 @@
 #! /usr/bin/env python
 
+import argparse
+import time
+
 from seleniumrequests import Chrome
 from selenium.webdriver.common.action_chains import ActionChains
 from selenium.webdriver.chrome.options import Options
-import time
-import argparse
+
 
 MAX_POSTS = 5000
 
 def run_delete():
     parser = argparse.ArgumentParser()
-    parser.add_argument("-E",
-                        "--email",
-                        default=None,
-                        help="Your email address associated with the account")
-
-    parser.add_argument("-P",
-                        "--password",
-                        default=None,
-                        help="Your Facebook password")
-
-    parser.add_argument("-U",
-                        "--profile-url",
-                        default=None,
-                        help="""
-                        The link to your Facebook profile, e.g. https://www.facebook.com/your.name
-                        """)
+    parser.add_argument(
+        "-E",
+        "--email",
+        required=True,
+        dest="email",
+        type=str,
+        help="Your email address associated with the account"
+    )
+
+    parser.add_argument(
+        "-P",
+        "--password",
+        required=True,
+        dest="password",
+        type=str,
+        help="Your Facebook password"
+    )
+
+    parser.add_argument(
+        "-U",
+        "--profile-url",
+        required=True,
+        dest="profile_url",
+        type=str,
+        help="The link to your Facebook profile, e.g. https://www.facebook.com/your.name"
+    )
+
+    parser.add_argument(
+        "-H",
+        "--headless",
+        action="store_true",
+        dest="is_headless",
+        default=False,
+        help="Run browser in headless mode (no gui)"
+    )
 
     args = parser.parse_args()
 
-    delete_posts(user_email_address=args.email,
-                 user_password=args.password,
-                 user_profile_url=args.profile_url)
-
-def delete_posts(user_email_address=None,
-                 user_password=None,
-                 user_profile_url=None):
+    delete_posts(
+        user_email_address=args.email,
+        user_password=args.password,
+        user_profile_url=args.profile_url,
+        is_headless=args.is_headless
+    )
+
+def delete_posts(user_email_address,
+                 user_password,
+                 user_profile_url,
+                 is_headless):
     """
-    user_email_address: Your Email
-    user_password: Your password
-    user_profile_url: Your profile URL
+    user_email_address: str Your Email
+    user_password: str Your password
+    user_profile_url: str Your profile URL
     """
-
-    assert all((user_email_address,
-                user_password,
-                user_profile_url)), "Insufficient arguments provided"
-
     # The Chrome driver is required because Gecko was having issues
     chrome_options = Options()
-    prefs = {"profile.default_content_setting_values.notifications" : 2}
+    prefs = {"profile.default_content_setting_values.notifications": 2, 'disk-cache-size': 4096}
     chrome_options.add_experimental_option("prefs", prefs)
-
     chrome_options.add_argument("start-maximized")
 
-    driver = Chrome(chrome_options=chrome_options)
+    if is_headless:
+        chrome_options.add_argument('--headless')
+        chrome_options.add_argument('--disable-gpu')
+        chrome_options.add_argument('log-level=2')
+
+    driver = Chrome(options=chrome_options)
     driver.implicitly_wait(10)
 
     driver.get("https://facebook.com")
 
     email = "email"
     password = "pass"
-    login="loginbutton"
+    login = "loginbutton"
 
     emailelement = driver.find_element_by_name(email)
-
     passwordelement = driver.find_element_by_name(password)
 
     emailelement.send_keys(user_email_address)
-
     passwordelement.send_keys(user_password)
 
     loginelement = driver.find_element_by_id(login)
-
     loginelement.click()
+
     if "Two-factor authentication" in driver.page_source:
         # Allow time to enter 2FA code
         print("Pausing to enter 2FA code")
@@ -106,3 +128,7 @@ def delete_posts(user_email_address=None,
         # Required to sleep the thread for a bit after using JS to click this button
         time.sleep(5)
         driver.refresh()
+
+
+if __name__ == "__main__":
+    run_delete()