slackdeleter.py

from datetime import datetime, timedelta
import time
import sys

import requests
import click

import slack


def fetch_files(token, days, only_created):
    threshold = datetime.now() - timedelta(days=days)
    unixtime = time.mktime(threshold.timetuple())
    params = {'token': token, 'ts_to': unixtime, 'page': 1}

    if only_created:
        user = slack.get_user(token)
        # print "User ID: %s" % user['user_id']
        params['user'] = user['user_id']

    print '.',

    json = slack.slack_call(token, 'https://slack.com/api/files.list', params=params)
    pages = json['paging']['pages']
    files = json['files']

    for i in range(2, pages + 1):
        params['page'] = i
        # print "Fetching Files with %s" % params
        print '.',
        json = slack.slack_call(token, 'https://slack.com/api/files.list', params=params)
        files += json['files']

    return files


def human_size(b):
    if b > 1000000:
        return "%0.2f MB" % (b / 1000000.0)

    if b > 1000:
        return "%0.2f KB" % (b / 1000.0)

    else:
        return "%s Bytes" % b


def delete_files(queue, token):
    params = {'token': token}
    bytes_deleted = 0
    files_deleted = 0
    for f in queue:
        print "  Deleting %s (%s) ... " % (f['name'].encode("utf-8"), human_size(f['size'])),

        params['file'] = f['id']
        a = requests.get('https://slack.com/api/files.delete', params=params)
        if a.status_code == 200 and a.json()['ok']:
            bytes_deleted += f['size']
            files_deleted += 1
            print "Deleted"
        else:
            print " - ERROR: HTTP:%s (%s)" % (a.status_code, a.json()['error'])

    return files_deleted, bytes_deleted


# Age From Timestamp
def aft(timestamp):
    d = datetime.fromtimestamp(int(timestamp))

    td = datetime.now() - d

    if td.total_seconds() > 60 * 60 * 24:
        return "%0.0f days ago" % (td.total_seconds() / 86400)

    elif td.total_seconds() > 60 * 60:
        return "%0.0f hours ago" % (td.total_seconds() / 3600)

    else:
        return "%0.0f minutes ago" % (td.total_seconds() / 60)


def get_real_name(users, user_id):
    n = slack.where(users, 'id', user_id)

    if 'profile' in n:  # Caters for deleted profiles.
        return n['profile']['real_name']

    else:
        return n['real_name']


@click.command()
@click.option('--token', prompt='Slack Test Token',
              help='Your Slack test token (https://api.slack.com/docs/oauth-test-tokens)',
              type=click.STRING)

@click.option('--days', default=31, prompt='Age of files to delete.',
              help='How old a file must be in order to delete it.')

@click.option('--only_created', default=True, prompt='Only delete files you created?',
              help="Only delete files you created?", type=click.BOOL)

@click.option('--sort', default='size', prompt='Sort by',
              help="Sort by 'size', 'date' or 'user'.", type=click.Choice(['size', 'date', 'user']))

@click.option('--min_kb', default='1000', prompt='Minimum filesize to delete (in kilobytes)',
              help="Minimum number of Kilobytes for file to qualify.", type=click.INT)

@click.option('--no-prompt', default=False, type=click.BOOL)

def query_slack(token, days, only_created, sort, min_kb, no_prompt):

    print "Querying Slack's Servers",

    users = slack.get_users(token)

    files = fetch_files(token, days, only_created)

    if sort == 'size':
        files = sorted(files, key=lambda k: k['size'], reverse=True)

    elif sort == 'date':
        files = sorted(files, key=lambda k: k['created'])

    elif sort == 'user':
        files = sorted(files, key=lambda k: k['user'])

    min_bytes = min_kb * 1000

    delete_queue = []
    total_bytes = 0

    file_name_lengths = []

    for f in files:
        if f['size'] >= min_bytes:
            file_name_lengths.append(len(f['name']))
            total_bytes += f['size']
            delete_queue.append(f)

    if delete_queue:
        np = max(file_name_lengths)

        print "\nFiles:"

        for f in delete_queue:
            real_name = get_real_name(users, f['user']).encode("utf-8")

            print "  %s %s\t%s\t%s" % (f['name'].encode("utf-8").ljust(np + 4), human_size(f['size']).ljust(12),
                                       real_name.ljust(24),
                                       aft(f['created']))

        print "\n%s files match your criteria. (%s)\n" % (len(delete_queue), human_size(total_bytes))

        if no_prompt:
            files_deleted, bytes_deleted = delete_files(delete_queue, token)
            print '\n%s files deleted (%s)' % (files_deleted, human_size(bytes_deleted))
            return
        else:
            delete = click.confirm('Do you want to delete all of these files now?')

        if delete:
            really = click.prompt("Type \"YES\" to really delete the files listed above")
            if really != "YES":
                print "You didn't type \"YES\" so I'm quitting to be safe."
                return

            print "\nDELETING FILES:"
            files_deleted, bytes_deleted = delete_files(delete_queue, token)
            print '\n%s files deleted (%s)' % (files_deleted, human_size(bytes_deleted))

    else:
        print "\nNo files matched your criteria."

    print ""


if __name__ == '__main__':
    try:
        query_slack()
    except Exception as e:
        print "\nFATAL ERROR: %s\n" % e
        sys.exit(99)