GetScanExecutions()

Complexity

Conditions

9

Size

Total Lines

57

Duplication

Lines	0
Ratio	0 %

# Licensed to the StackStorm, Inc ('StackStorm') under one or more
def GetScanExecutions(config, scan_id):
    """
    The template class for

    Returns: An blank Dict.

    Raises:
    ValueError: On lack of key in config.
    """
    results = {}

    url = "https://{}/api/scan/v1/scans/{}".format(config['api_host'], scan_id)
    headers = {"Accept": "application/json"}

    try:
        r = requests.get(url,
                         headers=headers,
                         auth=(config['api_key'], ''))
        r.raise_for_status()
    except:
        raise ValueError("HTTP error: %s on %s" % (r.status_code, r.url))

    try:
        data = r.json()
    except:
        raise ValueError("Invalid JSON")
    else:
        results = {'latest_complete': None, 'scans': []}
        for item in data:
            cd = datetime.fromtimestamp(item['create_date'])
            create_date = cd.strftime('%Y-%m-%d %H:%M:%S')

            if item['finish_date'] is not None:
                fd = datetime.fromtimestamp(item['finish_date'])
                finish_date = fd.strftime('%Y-%m-%d %H:%M:%S')
                duration = str(fd - cd)
            else:
                finish_date = "-"
                duration = str(datetime.now().replace(microsecond=0) - cd)

            results['scans'].append({"id": item['id'],
                                     "active": item['active'],
                                     "create_date": create_date,
                                     "finish_date": finish_date,
                                     "duration": duration})

    # This list can be very large, limit to the last 10.
    results['scans'].sort(reverse=True)
    results['scans'] = results['scans'][0:10]

    # Find the latest ccmpleted scan..
    for item in results['scans']:
        if item['active'] is False:
            results['latest_complete'] = item['id']
            break

    return results