From c458d4da9534351f30c479e1e75a004322a695a2 Mon Sep 17 00:00:00 2001 From: zenaku Date: Tue, 30 Jun 2026 12:03:44 -0500 Subject: [PATCH] Update operator command to use new SQLAlchemy ORM --- cli/commands/operator.py | 25 ++++++++++++++++--------- 1 file changed, 16 insertions(+), 9 deletions(-) diff --git a/cli/commands/operator.py b/cli/commands/operator.py index 33453b1..271ec78 100644 --- a/cli/commands/operator.py +++ b/cli/commands/operator.py @@ -1,32 +1,39 @@ -from core.controllers.OperatorController import OperatorController +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': - pprint.pp(OperatorController.get_all()) + 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 = OperatorController.get(arguments.id) + operator = session.get(Operator, arguments.id) if operator is not None: - pprint.pp(operator) + 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') \ No newline at end of file