1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
import sys
from cliff import app
from cliff import commandmanager
from testapiclient.utils import clientmanager
class TestAPIClient(app.App):
def __init__(self):
super(TestAPIClient, self).__init__(
description='TestAPI Client',
version='0.1',
command_manager=commandmanager.CommandManager('testapi'),
deferred_help=True,
)
def build_option_parser(self, description, version, argparse_kwargs=None):
self.LOG.debug('build_option_parser')
parser = super(TestAPIClient, self).build_option_parser(
description,
version,
argparse_kwargs)
parser.add_argument('-u',
type=str,
help='Username for authentication')
parser.add_argument('-p',
type=str,
help='Password for authentication')
return parser
def initialize_app(self, argv):
self.LOG.debug('initialize_app')
self.client_manager = clientmanager.ClientManager(self.options)
def prepare_to_run_command(self, cmd):
self.LOG.debug('prepare_to_run_command %s', cmd.__class__.__name__)
if self.client_manager.auth_required:
self.client_manager.auth()
def clean_up(self, cmd, result, err):
self.LOG.debug('clean_up %s', cmd.__class__.__name__)
if err:
self.LOG.debug('got an error: %s', err)
def main(argv=sys.argv[1:]):
client = TestAPIClient()
return client.run(argv)
if __name__ == '__main__':
sys.exit(main(sys.argv[1:]))
|
