wss_rm_trans_Depends.py

#!/usr/bin/env python3

import sys
import json
import requests
import time

## Global Vars 
user_key        = sys.argv[1] # Passed via Arg 1
org_token       = sys.argv[2] # Passed via Arg 2
api_url         = "https://app.whitesourcesoftware.com/api/v1.3"
request_headers = {'Content-type': 'application/json'}

def main():
    print("\n\n-- START OF SCRIPT --\n\n")
    
    numberOfIgnoredAlerts = 0
    allProductsTokens = []
    allProjectsTokens = []


    # Get All Products Tokens
    # - - - - - - - - - - - - - - - - - - - - - - - - 
    request_data = {
        "requestType" : "getAllProducts",
        "userKey" : user_key,
        "orgToken" : org_token,
    }

    try:
        response = requests.post(api_url, 
                                headers = request_headers, 
                                data=json.dumps(request_data))
    except requests.exceptions.RequestException as exceptionText:  # This is the correct syntax
        raise SystemExit(exceptionText)

    print("Grabbing all Product Tokens : {} - {}\n".format(response, response.text))

    productJsonGlob = json.loads(response.text)

    # Used for Debuging
    #print(json.dumps(productJsonGlob, indent=2, sort_keys=True))

    for item in productJsonGlob["products"]:
        # Used for Debuging
        #print("{}:{}".format(item["productName"],item["productToken"]))
        allProductsTokens.append(item["productToken"])

    print("-\n\n-")

    # Get All Projects Tokens
    # - - - - - - - - - - - - - - - - - - - - - - - - 
    for product in allProductsTokens:

        request_data = {
            "requestType" : "getAllProjects",
            "userKey" : user_key,
            "productToken" : "{}".format(product),
        }

        try:
            response = requests.post(api_url, 
                                    headers = request_headers, 
                                    data=json.dumps(request_data))
        except requests.exceptions.RequestException as exceptionText:  # This is the correct syntax
            raise SystemExit(exceptionText)
        
        print("Grabbing all Project Tokens : {} - {}\n".format(response, response.text))

        projectJsonGlob = json.loads(response.content)

        # Used for Debuging
        #print(json.dumps(projectJsonGlob, indent=2, sort_keys=True))

        for item in projectJsonGlob["projects"]:
            # Used for Debuging
            #print("{}:{}".format(item["projectName"],item["projectToken"]))
            allProjectsTokens.append(item["projectToken"])


    print("-\n\n-")

    # Parse out Alerts based on each Project
    # - - - - - - - - - - - - - - - - - - - - - - - - 
    for project in allProjectsTokens:
        request_data = {
            "requestType" : "getProjectAlerts",
            "userKey" : user_key,
            "projectToken" : project,
        }

        try:
            responseProject = requests.post(api_url, 
                                    headers = request_headers, 
                                    data=json.dumps(request_data))
        except requests.exceptions.RequestException as exceptionText:  # This is the correct syntax
            raise SystemExit(exceptionText)

        print("\nGrabbing all Alerts Based on Project Tokens : {}\n".format(responseProject))

        projectAlertsGlob = json.loads(responseProject.content)

        print("Number of Alerts:{}".format(len(projectAlertsGlob["alerts"])))

        # reinitialize's the list back to null each loop
        alertsUUIDGlob = []

        # Parse out all ""directDependency": false"
        # - - - - - - - - - - - - - - - - - - - - - - - - 
        for alert in projectAlertsGlob["alerts"]:
            if "{}".format(alert["directDependency"]) == "False":
                # Used for Debuging
                #print("False:{}".format(alert["directDependency"]))
                alertsUUIDGlob.append("{}".format(alert["alertUuid"]))
            #else:
                # Used for Debuging
                #print("True:{}".format(alert["directDependency"]))
        
        print("Ignore Alerts Array:{}".format(alertsUUIDGlob))
        print("Ignore Number of Alerts:{}".format(len(alertsUUIDGlob)))

        numberOfIgnoredAlerts = numberOfIgnoredAlerts + int(len(alertsUUIDGlob))

        # Take Parsed directDependencies and set WhiteSource to Ignore the Alerts
        # - - - - - - - - - - - - - - - - - - - - - - - - 
        request_data = {
            "requestType" : "ignoreAlerts",
            "userKey" : user_key,
            "orgToken" : org_token,
            "alertUuids" : alertsUUIDGlob,
            "comments": "Transitive Dependencies (We do Not Fix Third Party Code)"
        }

        try:
            responseIgnore = requests.post(api_url, 
                                    headers = request_headers, 
                                    data=json.dumps(request_data))
        except requests.exceptions.RequestException as exceptionText:  # This is the correct syntax
            raise SystemExit(exceptionText)

        # Meant to run on a schedules, tells you how many Dependencies were ignored
        print("Removing Transitive Dependencies: {} - {}\n".format(responseIgnore, responseIgnore.text))


    print("\n\nNumber Of Alerts Ignored : {}\n\n".format(numberOfIgnoredAlerts))

    print("\n\n-- END OF SCRIPT --\n\n")


    # Used for Debuging
    # - - - - - - - - - - - - - - - - - - - - - - - - 
    # request_data = {
    #     "requestType" : "getProjectAlerts",
    #     "userKey" : user_key,
    #     "projectToken" : "",
    # }

    # response = requests.post(api_url, 
    #                         headers = request_headers, 
    #                         data=json.dumps(request_data))
    # print("\nGrabbing all Alerts Based on Project Tokens : {} - {}\n".format(response, response.text))

    # projectAlertsGlob = json.loads(response.content)

    # # Used to Create Files to save Json Globs for Development
    # # - - - - - - - - - - - - - - - - - - - - - - - - 
    # with open('t.txt', 'w') as f:
    #     sys.stdout = f # Change the standard output to the file we created.
    #     print(json.dumps(projectAlertsGlob, indent=2, sort_keys=True))


if __name__ == '__main__':
    main()