32 lines
No EOL
903 B
Python
32 lines
No EOL
903 B
Python
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') |