from core.controllers.OperatorController import OperatorController
import pprint

NAME = 'operator'


def register(subparsers):
    parser = subparsers.add_parser(NAME)
    subs = parser.add_subparsers(title='subcommands', dest='subcommand')

    subs.add_parser('list')

    show_parser = subs.add_parser('show')
    show_parser.add_argument('--id', '-i', type=int, required=True)

    return parser


def handle(arguments, main_parser):
    if arguments.subcommand is None:
        main_parser.parse_args(['operator', '--help'])
        return

    if arguments.subcommand == 'list':
        pprint.pp(OperatorController.get_all())

    elif arguments.subcommand == 'show':
        operator = OperatorController.get(arguments.id)
        if operator is not None:
            pprint.pp(operator)
        else:
            main_parser.error('the following argument should be a valid reference: --id/-i')