diff --git a/deletefb/deletefb.py b/deletefb/deletefb.py index bd552cc..8cd0f49 100755 --- a/deletefb/deletefb.py +++ b/deletefb/deletefb.py @@ -10,32 +10,55 @@ 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 @@ -50,10 +73,14 @@ def delete_posts(user_email_address=None, chrome_options = Options() prefs = {"profile.default_content_setting_values.notifications" : 2} 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")