find_destination.py

from read_config import read_config_multi_environment
from fitconnect import FITConnectClient, Environment
import argparse
import json
import logging
import sys

# parse command line arguments
parser = argparse.ArgumentParser(
                    prog = 'find_destination',
                    description = 'This script uses sender clients to check if a destination exists in any FIT-Connect environment')

parser.add_argument('-c', '--config', help='Path to config file', default='conf/find_destination.yaml')
parser.add_argument('-v', '--verbose', help='Show full server response (destination details or problem details)', action=argparse.BooleanOptionalAction)
parser.add_argument('destination_id', help='The destination that is being searched for')

args = parser.parse_args()

# configure logging. Uncomment to enable logging. Python's default logging level
# is WARN.
logging.basicConfig()
logging.getLogger('fitconnect').level = logging.INFO

# read config_file
config = read_config_multi_environment(args.config)

# Run through all environments read from the configuration file
for environment in config['environments']:
    # initialize SDK for specific environment (insecure=True to allow access to staging/prod)
    fitc = FITConnectClient(Environment[environment['environment']], environment['client_id'], environment['client_secret'], insecure=True)

    # query destinaton and collect response
    r = fitc.get_destination(args.destination_id)

    if r.status_code == 503: # service unavailable
        print(f"Environment {environment['environment']}: Service not available.")
        continue

    if r.ok:
        print(f"Environment {environment['environment']}: Destination {r.json()['destinationId']} found with status `{r.json()['status']}`.")
    else:
        print(f"Environment {environment['environment']}: {r.json()['detail']}.")

    if args.verbose:
        print(json.dumps(r.json(), indent=2) + '\n')