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 shard init
    mcmanage.py shard migrate
"""

from docopt import docopt

from mediacrush.email import send_report
from mediacrush.mcmanage.database import database_clear, database_sync
from mediacrush.mcmanage.files import files_delete
from mediacrush.mcmanage.report import report
from mediacrush.mcmanage.shard import init, migrate


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}

shard_commands = {"init": init, "migrate": migrate}

mapping = {
    "database": database_commands,
    "report": report_commands,
    "files": files_commands,
    "shard": shard_commands,
    "admin": None,
}


def find_true(arguments, mapping_dict):
    return list(
        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)