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.
Shreddit/shreddit

95 lines
2.6 KiB

#!/usr/bin/env python2
import praw
from praw.objects import Comment, Submission
import ConfigParser
import argparse
from datetime import datetime, timedelta
from re import sub
try:
from loremipsum import get_sentence
except:
get_sentence = lambda: '''I have been Shreddited for privacy!\n\nhttps://github.com/x89/Shreddit/'''
parser = argparse.ArgumentParser()
parser.add_argument('-c', '--config', help="config file to use instead of shreddit.cfg")
args = parser.parse_args()
config = ConfigParser.RawConfigParser()
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 2.1")
if _user and _pass:
r.login(_user, _pass)
else:
r.login()
if verbose:
print "Logged in as {0}".format(r.user)
if verbose:
print "Deleting messages before {0}.".format(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:
print "Keeping messages from subreddits {0}".format(', '.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)
before_time = datetime.utcnow() - timedelta(hours=hours)
if thing.id in whitelist_ids:
continue
if thing_time > before_time:
continue
if str(thing.subreddit).lower() in whitelist:
continue
if not trial_run:
if clear_vote:
thing.clear_vote()
if isinstance(thing, Submission):
if verbose:
print u'Deleting submission: #{0} {1}'.format(thing.id, thing.url)
elif isinstance(thing, Comment):
replacement_text = get_sentence()
if verbose:
print 'Editing and deleting /r/{3}/ #{0} with:\n\t"{1}" to\n\t"{2}"'.format(
thing.id, sub(r'[\t\r\n]', ' ', thing.body.encode('ascii', 'ignore')[:78]),
replacement_text[:78], thing.subreddit)
thing.edit(replacement_text)
if not edit_only:
thing.delete()