Skip to content

Commit 4ca8bf8

Browse files
committed
Update pki.server.cli.listener to use argparse
1 parent e7289d1 commit 4ca8bf8

File tree

1 file changed

+27
-29
lines changed

1 file changed

+27
-29
lines changed

base/server/python/pki/server/cli/listener.py

+27-29
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,8 @@
1818
# All rights reserved.
1919
#
2020

21-
from __future__ import absolute_import
22-
from __future__ import print_function
23-
import getopt
21+
import argparse
2422
import logging
25-
import sys
2623

2724
import pki.cli
2825

@@ -48,6 +45,24 @@ class ListenerFindCLI(pki.cli.CLI):
4845
def __init__(self):
4946
super().__init__('find', 'Find listeners')
5047

48+
self.parser = argparse.ArgumentParser(
49+
prog=self.name,
50+
add_help=False)
51+
self.parser.add_argument(
52+
'-i',
53+
'--instance',
54+
default='pki-tomcat')
55+
self.parser.add_argument(
56+
'-v',
57+
'--verbose',
58+
action='store_true')
59+
self.parser.add_argument(
60+
'--debug',
61+
action='store_true')
62+
self.parser.add_argument(
63+
'--help',
64+
action='store_true')
65+
5166
def print_help(self):
5267
print('Usage: pki-server listener-find [OPTIONS]')
5368
print()
@@ -59,36 +74,19 @@ def print_help(self):
5974

6075
def execute(self, argv):
6176

62-
try:
63-
opts, _ = getopt.gnu_getopt(argv, 'i:v', [
64-
'instance=',
65-
'verbose', 'debug', 'help'])
77+
args = self.parser.parse_args(args=argv)
6678

67-
except getopt.GetoptError as e:
68-
print('ERROR: %s' % e)
79+
if args.help:
6980
self.print_help()
70-
sys.exit(1)
71-
72-
instance_name = 'pki-tomcat'
73-
74-
for o, a in opts:
75-
if o in ('-i', '--instance'):
76-
instance_name = a
81+
return
7782

78-
elif o in ('-v', '--verbose'):
79-
logging.getLogger().setLevel(logging.INFO)
83+
if args.debug:
84+
logging.getLogger().setLevel(logging.DEBUG)
8085

81-
elif o == '--debug':
82-
logging.getLogger().setLevel(logging.DEBUG)
86+
elif args.verbose:
87+
logging.getLogger().setLevel(logging.INFO)
8388

84-
elif o == '--help':
85-
self.print_help()
86-
sys.exit()
87-
88-
else:
89-
print('ERROR: unknown option: %s' % o)
90-
self.print_help()
91-
sys.exit(1)
89+
instance_name = args.instance
9290

9391
instance = pki.server.PKIServerFactory.create(instance_name)
9492

0 commit comments

Comments
 (0)