mcmanage.py

#!/usr/bin/env python
"""MediaCrush manage.

Usage:
    mcmanage.py database clear
    mcmanage.py database sync
    mcmanage.py admin list
    mcmanage.py admin add <pwhash>
    mcmanage.py admin delete <pwhash>
    mcmanage.py report show
    mcmanage.py report email
    mcmanage.py files delete <hash>
    mcmanage.py files nsfw <hash>
"""

from docopt import docopt

from mediacrush.mcmanage.database import database_clear, database_sync
from mediacrush.mcmanage.report import report
from mediacrush.mcmanage.files import files_delete, files_nsfw

from mediacrush.email import send_report

def show_report(args):
    print(report())

database_commands = {
    'clear': database_clear,
    'sync': database_sync,
}

report_commands = {
    'show': show_report,
    'email': lambda args: send_report(report())
}

files_commands = {
    'delete': files_delete,
    'nsfw': files_nsfw
}

mapping = {
    'database': database_commands,
    'report': report_commands,
    'files': files_commands,
    'admin': None,
}

def find_true(arguments, mapping_dict):
    return filter(lambda x: x is not None, [item if arguments[item] else None for item in mapping_dict])[0]

if __name__ == '__main__':
    arguments = docopt(__doc__, version='1.0')
    module = find_true(arguments, mapping)
    commands = mapping[module]
    command = find_true(arguments, commands)
    command = commands[command]

    command(arguments)