Browse Source

Clean up new modules

pull/72/head
Wesley Kerfoot 5 years ago
parent
commit
fbc18058bd
  1. 11
      deletefb/deletefb.py
  2. 4
      deletefb/tools/conversations.py
  3. 2
      deletefb/types.py

11
deletefb/deletefb.py

@ -4,6 +4,8 @@ from .tools.config import settings
from .tools.likes import unlike_pages
from .tools.login import login
from .tools.wall import delete_posts
from .tools.conversations import delete_conversations
from .tools.comments import delete_comments
import argparse
import getpass
@ -21,7 +23,7 @@ def run_delete():
default="wall",
dest="mode",
type=str,
choices=["wall", "unlike_pages"],
choices=["wall", "unlike_pages", "comments", "conversations"],
help="The mode you want to run in. Default is `wall' which deletes wall posts"
)
@ -112,6 +114,13 @@ def run_delete():
elif args.mode == "unlike_pages":
unlike_pages(driver, args.profile_url)
elif args.mode == "comments":
delete_comments(driver, args.profile_url)
elif args.mode == "conversations":
delete_conversations(driver)
else:
print("Please enter a valid mode")
sys.exit(1)

4
deletefb/tools/messages.py → deletefb/tools/conversations.py

@ -7,9 +7,9 @@ from selenium.webdriver.support.ui import WebDriverWait
LOG = logger(__name__)
def delete_comments(driver):
def delete_conversations(driver):
"""
Remove all conversations within a specified range
"""
driver.get("https://www.facebook.com/messages/t/"
driver.get("https://www.facebook.com/messages/t/")

2
deletefb/types.py

@ -26,8 +26,8 @@ class Comment:
@attr.s
class Conversation:
recipient = attr.ib()
last_message_time = attr.ib(factory=timestamp_now)
name = attr.ib()
last_message_time = attr.ib(factory=timestamp_now)
@attr.s
class Page:

Loading…
Cancel
Save