cli.py

from __future__ import print_function
import sys
import argparse
import li_privacy
import signal

def signal_handler(sig, frame):
    print('Interrupted by Ctrl+C!')
    sys.exit(0)

def main(name=None):
    signal.signal(signal.SIGINT, signal_handler)
    parser = argparse.ArgumentParser(
            description="Interact with the LiveIntent Privacy API", \
            epilog="For API documentation, see https://link.liveintent.com/privacy-api")
    parser.add_argument('--version', action='version', \
            version='%(prog)s v.{version}'.format(version=li_privacy.__version__))

    subparsers = parser.add_subparsers(title="actions", dest='command')

    actions = {
        "init": li_privacy.InitProcessor(),
        "delete":  li_privacy.DeleteProcessor(),
        "optout": li_privacy.OptoutProcessor(),
        "ping": li_privacy.PingProcessor(),
        "hash": li_privacy.HashProcessor()
    }
    for action in actions:
        processor = actions[action]
        sub_parser = subparsers.add_parser(action, help=processor.description)
        processor.setup_argparse(sub_parser)

    args = parser.parse_args(sys.argv[1:])
    try:
        func = actions[args.command]
    except (AttributeError, KeyError):
        parser.print_help()
        return 1
    func.execute(args)

if __name__ == "__main__":
    main()