python-nitro/config.py at master · slauger/python-nitro · GitHub

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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#!/usr/bin/env python

"""config.py: bulk configuration for NetScaler trough the NITRO API."""

__author__ = "Simon Lauger"
__copyright__ = "Copyright 2018, IT Consulting Simon Lauger"
__license__ = "Apache"
__version__ = "0.0.1"
__maintainer__ = "Simon Lauger"
__email__ = "simon@lauger.de"

import argparse
import nitro
import os.path
import json
from ConfigParser import SafeConfigParser

add_args = {
  '--config': {
    'metavar': '<file>',
    'help': 'Path to the ini config file (default: ~/.netscaler.ini)',
    'type': str,
    'default': '~/.netscaler.ini',
    'required': False,
  },
  '--section': {
    'metavar': '<section>',
    'help': 'Section in the ini config file (default: netscaler)',
    'type': str,
    'default': 'netscaler',
    'required': False,
  },
}

def add_argument(parser, arg, params):
    parser.add_argument(
        arg,
        metavar=params['metavar'],
        type=params['type'],
        help=params['help'],
        default=params['default'],
        required=params['required']
    )

# add cli arguments
parser = argparse.ArgumentParser(description='command line arguments.')

parser.add_argument('method',
    metavar='<method>',
    help='http method (get/post/put/delete)',
    type=str,
    default=None,
)

parser.add_argument('file',
    metavar='<file>',
    help='command file for the macro api',
    type=str,
    default=None,
)

# optional arguments
for key in add_args:
  add_argument(parser, key, add_args[key])

# parse arguments
args = parser.parse_args()

# read configuration from ~/.netscaler.ini
cfgfile = os.path.expanduser(args.config)
section = args.section

if not os.path.isfile(cfgfile):
   raise Exception('required config does not exist')

config = SafeConfigParser()
config.read(cfgfile)

if config.has_option(section, 'url'):
    url = config.get(section, 'url')
else:
    raise Exception('no system url defined in config file')

if config.has_option(section, 'username'):
    username = config.get(section, 'username')
else:
    username = 'nsroot'

if config.has_option(section, 'password'):
    password = config.get(section, 'password')
else:
    password = 'nsroot'

if config.has_option(section, 'verify_ssl'):
   verify_ssl = config.getboolean(section, 'verify_ssl')
else:
    verify_ssl = True

with open(args.file, 'r') as socket:
  payload = file.read(socket)

# start nitro client
nitro_client = nitro.NitroClient(url, username, password)

# disable ssl verification if needed
nitro_client.set_verify(verify_ssl)

# rollback is not supported on update
#if args.method != 'post':
#  nitro_client.on_error('exit')

nitro_client.on_error('continue')

# do the request to the NITRO API
result = nitro_client.request(
  args.method,
  endpoint='config',
  objecttype='macroapi',
  data=payload
)

# print result
try:
    print(json.dumps(result.json(), sort_keys=True, indent=4))
except:
    print(result.__dict__)