mirror of https://github.com/x89/Shreddit
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
132 lines
3.9 KiB
132 lines
3.9 KiB
#!/usr/bin/env python3
|
|
|
|
import praw
|
|
from praw.errors import *
|
|
import argparse
|
|
from simpleconfigparser import simpleconfigparser
|
|
from praw.objects import Comment, Submission
|
|
from datetime import datetime, timedelta
|
|
from re import sub
|
|
from time import sleep
|
|
|
|
try:
|
|
from loremipsum import get_sentence
|
|
except:
|
|
def get_sentence():
|
|
return '''I have been Shreddited for privacy!\n\n\
|
|
https://github.com/x89/Shreddit/'''
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
'-c',
|
|
'--config',
|
|
help="config file to use instead of the default shreddit.cfg"
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
config = simpleconfigparser()
|
|
if args.config:
|
|
config.read(args.config)
|
|
else:
|
|
config.read('shreddit.cfg')
|
|
|
|
hours = config.getint('main', 'hours')
|
|
whitelist = config.get('main', 'whitelist')
|
|
whitelist_ids = config.get('main', 'whitelist_ids')
|
|
sort = config.get('main', 'sort')
|
|
verbose = config.getboolean('main', 'verbose')
|
|
clear_vote = config.getboolean('main', 'clear_vote')
|
|
trial_run = config.getboolean('main', 'trial_run')
|
|
edit_only = config.getboolean('main', 'edit_only')
|
|
item = config.get('main', 'item')
|
|
|
|
_user = config.get('main', 'username')
|
|
_pass = config.get('main', 'password')
|
|
|
|
r = praw.Reddit(user_agent="Shreddit-PRAW 3")
|
|
|
|
|
|
def login(user=None, password=None):
|
|
try:
|
|
if user and password:
|
|
r.login(_user, _pass)
|
|
else:
|
|
r.login() # Let the user supply details
|
|
except InvalidUserPass as e:
|
|
raise InvalidUserPass(e)
|
|
except RateLimitExceeded:
|
|
raise RateLimitExceeded()
|
|
except NonExistentUser:
|
|
raise NonExistentUser("User does not exist")
|
|
|
|
login(user=_user, password=_pass)
|
|
|
|
if verbose:
|
|
print("Logged in as {user}".format(user=r.user))
|
|
|
|
if verbose:
|
|
print("Deleting messages before {time}.".format(
|
|
time=datetime.now() - timedelta(hours=hours))
|
|
)
|
|
|
|
whitelist = [y.strip().lower() for y in whitelist.split(',')]
|
|
whitelist_ids = [y.strip().lower() for y in whitelist_ids.split(',')]
|
|
|
|
if verbose and whitelist:
|
|
print("Keeping messages from subreddits {subs}".format(
|
|
subs=', '.join(whitelist))
|
|
)
|
|
|
|
things = []
|
|
if item == "comments":
|
|
things = r.user.get_comments(limit=None, sort=sort)
|
|
elif item == "submitted":
|
|
things = r.user.get_submitted(limit=None, sort=sort)
|
|
elif item == "overview":
|
|
things = r.user.get_overview(limit=None, sort=sort)
|
|
else:
|
|
raise Exception("Your deletion section is wrong")
|
|
|
|
for thing in things:
|
|
thing_time = datetime.fromtimestamp(thing.created_utc)
|
|
# Delete things after after_time
|
|
after_time = datetime.utcnow() - timedelta(hours=hours)
|
|
if thing_time > after_time:
|
|
continue
|
|
# For edit_only we're assuming that the hours aren't altered.
|
|
# This saves time when deleting (you don't edit already edited posts).
|
|
if edit_only:
|
|
end_time = after_time - timedelta(hours=hours)
|
|
if thing_time < end_time:
|
|
continue
|
|
|
|
if str(thing.subreddit).lower() in whitelist or \
|
|
thing.id in whitelist_ids:
|
|
continue
|
|
|
|
if not trial_run:
|
|
if clear_vote:
|
|
thing.clear_vote()
|
|
if isinstance(thing, Submission):
|
|
if verbose:
|
|
print(u'Deleting submission: #{id} {url}'.format(
|
|
id=thing.id,
|
|
url=thing.url)
|
|
)
|
|
elif isinstance(thing, Comment):
|
|
replacement_text = get_sentence()
|
|
if verbose:
|
|
msg = '/r/{3}/ #{0} with:\n\t"{1}" to\n\t"{2}"'.format(
|
|
thing.id,
|
|
sub(r'\W', ' ', thing.body.encode('ascii', 'ignore')[:78]),
|
|
replacement_text[:78],
|
|
thing.subreddit
|
|
)
|
|
if edit_only:
|
|
print('Editing {msg}'.format(msg=msg))
|
|
else:
|
|
print('Editing and deleting {msg}'.format(msg=msg))
|
|
thing.edit(replacement_text)
|
|
if not edit_only:
|
|
thing.delete()
|