2022-06-14 21:34:27 +02:00
|
|
|
#!/usr/bin/env python3
|
2022-06-17 15:45:20 +02:00
|
|
|
#
|
|
|
|
# Copyright 2022, Mischa Peters <mischa AT alkira DOT net>, Alkira.
|
|
|
|
# get.py
|
|
|
|
# Version 0.1 - 20220617 - initial release
|
2022-06-21 14:58:57 +02:00
|
|
|
# Version 0.2 - 20220621 - simplified structure, added -i for IDs only, added -u for ugly JSON
|
2022-06-17 15:45:20 +02:00
|
|
|
#
|
|
|
|
# Permission to use, copy, modify, and distribute this software for any
|
|
|
|
# purpose with or without fee is hereby granted, provided that the above
|
|
|
|
# copyright notice and this permission notice appear in all copies.
|
|
|
|
#
|
|
|
|
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
|
|
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
|
|
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
|
|
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
|
|
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
|
|
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
|
|
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
#
|
2022-06-14 21:34:27 +02:00
|
|
|
import os
|
|
|
|
import sys
|
|
|
|
import re
|
|
|
|
import json
|
|
|
|
import time
|
|
|
|
import logging
|
|
|
|
import requests
|
|
|
|
import configparser
|
2022-06-17 15:45:20 +02:00
|
|
|
import argparse
|
|
|
|
|
|
|
|
# Parse all arguments
|
2022-06-17 16:16:41 +02:00
|
|
|
parser = argparse.ArgumentParser(description="Get JSON config to AlkiraAPI")
|
2022-06-17 16:01:23 +02:00
|
|
|
parser.add_argument("-t", "--tenant", type=str, default='alkira.cnf', help="location of alikira.cnf (default: alkira.cnf)")
|
2022-06-21 14:58:57 +02:00
|
|
|
parser.add_argument("-i", "--id", help="", action="store_true")
|
|
|
|
parser.add_argument("-u", "--ugly", help="make the JSON ugly!", action="store_true")
|
2022-06-17 16:01:23 +02:00
|
|
|
parser.add_argument("-v", "--verbose", type=int, default=0, help="Verbose level 0 or 1 (default: 0)")
|
2022-06-17 15:45:20 +02:00
|
|
|
|
|
|
|
try:
|
|
|
|
args = parser.parse_args()
|
|
|
|
ALKIRA_CONFIG = args.tenant
|
|
|
|
except argparse.ArgumentError as e:
|
|
|
|
print(str(e))
|
|
|
|
sys.exit()
|
|
|
|
try:
|
|
|
|
loglevel = {
|
|
|
|
0: logging.INFO,
|
|
|
|
1: logging.DEBUG
|
|
|
|
}[args.verbose]
|
|
|
|
except KeyError:
|
|
|
|
loglevel = logging.INFO
|
2022-06-14 21:34:27 +02:00
|
|
|
|
|
|
|
###############################################
|
|
|
|
|
2022-06-17 16:01:23 +02:00
|
|
|
# Set logging.INFO to logging.DEBUG for debug information
|
|
|
|
logging.basicConfig(level=loglevel)
|
|
|
|
logging = logging.getLogger('AlkiraAPI')
|
|
|
|
|
2022-06-17 15:45:20 +02:00
|
|
|
# Tenant config
|
2022-06-17 12:45:57 +02:00
|
|
|
if not os.path.isfile(ALKIRA_CONFIG):
|
|
|
|
logging.error(f"The config file {ALKIRA_CONFIG} doesn't exist")
|
2022-06-14 21:34:27 +02:00
|
|
|
sys.exit(1)
|
2022-06-17 12:45:57 +02:00
|
|
|
alkira = configparser.RawConfigParser()
|
|
|
|
alkira.read(ALKIRA_CONFIG)
|
2022-06-14 21:34:27 +02:00
|
|
|
|
2022-06-17 12:45:57 +02:00
|
|
|
ALKIRA_TENANT = alkira.get('alkira', 'ALKIRA_TENANT')
|
|
|
|
ALKIRA_USERNAME = alkira.get('alkira', 'ALKIRA_USERNAME')
|
|
|
|
ALKIRA_PASSWORD = alkira.get('alkira', 'ALKIRA_PASSWORD')
|
2022-06-14 21:34:27 +02:00
|
|
|
ALKIRA_BASE_URI = f'https://{ALKIRA_TENANT}/api'
|
|
|
|
|
|
|
|
###############################################
|
|
|
|
|
|
|
|
# Set default headers
|
|
|
|
headers = {'Content-Type': "application/json"}
|
|
|
|
|
|
|
|
# Naming exceptions
|
|
|
|
service_exceptions = {
|
|
|
|
"saas": "internet",
|
|
|
|
"pan": "panfw",
|
|
|
|
"ftntfw": "ftnt-fw-",
|
|
|
|
"chkpfw": "chkp-fw-"
|
|
|
|
}
|
|
|
|
|
|
|
|
def alkira_login():
|
|
|
|
body = {'userName': ALKIRA_USERNAME,
|
|
|
|
'password': ALKIRA_PASSWORD}
|
|
|
|
session = requests.session()
|
|
|
|
response = alkira_post(session, '/login', body)
|
|
|
|
return session
|
|
|
|
|
|
|
|
def alkira_post(session, uri, body):
|
|
|
|
url = f'{ALKIRA_BASE_URI}{uri}'
|
|
|
|
try:
|
|
|
|
response = session.post(url, data=json.dumps(body), headers=headers)
|
|
|
|
response.raise_for_status()
|
|
|
|
except Exception as e:
|
|
|
|
logging.error(f'Error: {str(e)}')
|
|
|
|
sys.exit(1)
|
|
|
|
return response
|
|
|
|
|
|
|
|
def alkira_get(session, uri):
|
|
|
|
url = f'{ALKIRA_BASE_URI}{uri}'
|
|
|
|
try:
|
|
|
|
response = session.get(url, headers=headers)
|
|
|
|
response.raise_for_status()
|
|
|
|
except Exception as e:
|
|
|
|
logging.error(f'Error: {str(e)}')
|
|
|
|
sys.exit(1)
|
|
|
|
return response
|
|
|
|
|
|
|
|
def alkira_delete(session, uri):
|
|
|
|
url = f'{ALKIRA_BASE_URI}{uri}'
|
|
|
|
try:
|
|
|
|
response = session.delete(url, headers=headers)
|
|
|
|
response.raise_for_status()
|
|
|
|
except Exception as e:
|
|
|
|
logging.error(f'Error: {str(e)}')
|
|
|
|
sys.exit(1)
|
|
|
|
return response
|
|
|
|
|
|
|
|
# Authenticate
|
2022-06-20 21:59:28 +02:00
|
|
|
logging.info('=== Authenticating')
|
2022-06-14 21:34:27 +02:00
|
|
|
s = alkira_login()
|
|
|
|
logging.debug(s)
|
|
|
|
|
|
|
|
# Get TenantID
|
2022-06-20 21:59:28 +02:00
|
|
|
logging.info('=== Fetching Tenant Info')
|
2022-06-14 21:34:27 +02:00
|
|
|
r = alkira_get(s, '/tenantnetworks')
|
|
|
|
data = r.json()
|
|
|
|
tenantNetworkId = data[0]['id']
|
|
|
|
tenantName = data[0]['name']
|
|
|
|
logging.info(f'Tenant Name: {tenantName}')
|
|
|
|
logging.info(f'Tenant ID: {tenantNetworkId}')
|
|
|
|
|
2022-06-21 14:58:57 +02:00
|
|
|
# Collecting data with base url: /
|
|
|
|
collect = [
|
|
|
|
'tenantenabledfeatures',
|
|
|
|
'credentials'
|
|
|
|
]
|
|
|
|
|
|
|
|
if not args.id:
|
|
|
|
for item in collect:
|
|
|
|
logging.info(f'=== Collecting {item}')
|
|
|
|
r = alkira_get(s, f'/{item}')
|
|
|
|
data = r.json()
|
|
|
|
print(f'# {item}')
|
|
|
|
if args.ugly:
|
|
|
|
print(json.dumps(data))
|
|
|
|
else:
|
|
|
|
print(json.dumps(data, indent=4))
|
|
|
|
|
|
|
|
# Collecting data with base url: /tenantnetworks/{tenantNetworkId}
|
|
|
|
collect = [
|
|
|
|
'connectors',
|
|
|
|
'services',
|
2022-06-27 12:51:23 +02:00
|
|
|
'alkira-remote-access-connectors',
|
2022-06-21 14:58:57 +02:00
|
|
|
'global-cidr-lists'
|
|
|
|
]
|
|
|
|
|
|
|
|
for item in collect:
|
|
|
|
logging.info(f'=== Collecting {item}')
|
|
|
|
r = alkira_get(s, f'/tenantnetworks/{tenantNetworkId}/{item}')
|
|
|
|
data = r.json()
|
|
|
|
print(f'# {item}')
|
|
|
|
if args.id:
|
|
|
|
for i in data:
|
|
|
|
str = f"{i['id']:10}"
|
|
|
|
if 'type' in i:
|
|
|
|
str = str + (f"\t{i['type']:20}")
|
|
|
|
|
|
|
|
if 'name' in i:
|
|
|
|
str = str + (f"\t{i['name']:20}")
|
|
|
|
print(str)
|
|
|
|
else:
|
|
|
|
if args.ugly:
|
|
|
|
print(json.dumps(data))
|
|
|
|
else:
|
|
|
|
print(json.dumps(data, indent=4))
|