39 lines
No EOL
1.2 KiB
Python
39 lines
No EOL
1.2 KiB
Python
from core.models.orm_models.Operator import Operator
|
|
from core.models.manage.session_management import get_session
|
|
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
|
|
|
|
session = get_session()
|
|
|
|
if arguments.subcommand == 'list':
|
|
operators = session.query(Operator).all()
|
|
pprint.pp([{
|
|
'id': o.id,
|
|
'name': o.name,
|
|
'public_key': o.public_key,
|
|
} for o in operators])
|
|
|
|
elif arguments.subcommand == 'show':
|
|
operator = session.get(Operator, arguments.id)
|
|
if operator is not None:
|
|
pprint.pp({
|
|
'id': operator.id,
|
|
'name': operator.name,
|
|
'public_key': operator.public_key,
|
|
})
|
|
else:
|
|
main_parser.error('the following argument should be a valid reference: --id/-i') |