python_examp.py

# Imports
import requests
import json
import jsonpickle
import sys
import os.path

# Deserialize Function
def show_issues (response):
    #frozen = jsonpickle.encode(response)
    jsonObject = jsonpickle.decode(response)

    for check in jsonObject['data']['getIssues']['issues']:
        id = check['id']
        desc = check['mainTitle']
        owners = check['owners']
        print("Issue ID:", id)
        print("Issue Description:", desc)
        print("Issue Owners:", owners)

def writeJSON (fileN,response):
    fileN = sys.argv[1]+'_response' 
    if os.path.exists(fileN+'.temp') == True:
        os.remove(fileN+'.temp')
    writefile = open(fileN + '.temp', 'w')
    writefile.write(passresult)
    writefile.close()
    if os.path.exists(fileN+'.json') == True:
        os.remove(fileN+'.json')
    os.rename(fileN+'.temp',fileN+'.json')

# OX GraphQL Info
# USE pip install python-dotenv to enable .env file
from dotenv import load_dotenv
load_dotenv()
apiurl = os.environ.get('oxUrl')
key = os.environ.get('oxKey')
# apiurl = 'https://api.cloud.ox.security/api/apollo-gateway'
# key = 'api_key'

if (len(sys.argv) < 2):
    print('You must enter the name of the query to submit. Query filenames should contain .query.json and .variables.json suffixes.')
    exit()

if sys.argv[1].lower() == 'help':
    print('Use any of the following queries as an argument: python_examp.py queryname')
    print('Commands are case sensitive if not Windows/MacOS environment')
    print('getIssues')
    print('getSingleIssue')
    print('getApps')
    print('getAppInventory')
    print('getAppInfo')
    print('getSbomLibs')
    print('getSbomLibStats')
    print('getSbomShort')
    print('getIssuesShort')
    print('getAppsShort')
    print('getAppsTags')
    print('getSbomFilters')
    exit()    

# Reading Query and Variables files for GraphQL API
qFilename = "./" +sys.argv[1]+ '.query.json'
vFilename = "./" +sys.argv[1]+ '.variables.json'

if os.path.exists(qFilename) == False:
    print('Query filename '+qFilename+' does not exist. This file should contain the GraphQL query. The Variables should be located at '+vFilename+'.')
    exit()

with open(qFilename, 'r') as query_file: 
    query = query_file.read()

with open(vFilename, 'r') as variables_file:
    variables = json.load(variables_file)

# Setting Post Params

headers = {
    'Content-Type': 'application/json',
    'Authorization': f'{key}',
}

body = {
    'query': query,
    'variables': variables,
}

# Post Request
attempts = 3

while attempts:
    try:
        response = requests.post(apiurl, headers=headers, json=body, timeout=300)
        if response.status_code == 200:
            result = response.json()
            passresult = json.dumps(result, indent=2)
            fileN = sys.argv[1]+'_response' 
            writeJSON(fileN,passresult)
            # if sys.argv[1].lower() == 'getissues':
            #   show_issues(passresult)
            break
        else:
            print(f'GraphQL request failed with status code: {response.status_code}')
            break
    except TimeoutError:
        attempts -= 1
    except requests.exceptions.RequestException as error:
        print(f'Error: {error}')