|  | 
|  | 1 | +import csv, json | 
|  | 2 | +from valispace import API | 
|  | 3 | + | 
|  | 4 | +import warnings | 
|  | 5 | +import os | 
|  | 6 | +import urllib.request | 
|  | 7 | +import time | 
|  | 8 | + | 
|  | 9 | +from ast import Str | 
|  | 10 | +from collections.abc import Callable | 
|  | 11 | +from xmlrpc.client import Boolean | 
|  | 12 | + | 
|  | 13 | +VALISPACE = { | 
|  | 14 | +    'domain': 'https://demonstration.valispace.com/', | 
|  | 15 | +    'username': 'AutomationsAPI', | 
|  | 16 | +    'password': 'AutomationsAPI' | 
|  | 17 | +} | 
|  | 18 | + | 
|  | 19 | +DEFAULT_VALUES = { | 
|  | 20 | +    "project": 24,  | 
|  | 21 | +} | 
|  | 22 | + | 
|  | 23 | +def get_map(api: API, endpoint: Str = "/", name: Str = "id", name_map_func: Callable[[Str], Boolean] = None, filter_func: Callable[[Str], Boolean]= None): | 
|  | 24 | +    """ | 
|  | 25 | +    Function that given an endpoint returns a dict with specific keys. | 
|  | 26 | +    If function is provided it generates the key. name_map_func must receive an object instance. | 
|  | 27 | +    Otherwise key will be the property of each object specified in name. | 
|  | 28 | +    """ | 
|  | 29 | +    mapping = {} | 
|  | 30 | +    if not name_map_func: | 
|  | 31 | +        name_map_func = lambda x: x[name] | 
|  | 32 | +    for inst in api.get(endpoint): | 
|  | 33 | +        if filter_func and not filter_func(inst): | 
|  | 34 | +            # Filtered out | 
|  | 35 | +            continue | 
|  | 36 | + | 
|  | 37 | +        key = name_map_func(inst) | 
|  | 38 | +        if not mapping.get(key): | 
|  | 39 | +            mapping[key] = inst | 
|  | 40 | +        else: | 
|  | 41 | +            warnings.warn(f"Warning ---> Key: {key} already has an object. Some data may be lost in mapping.") | 
|  | 42 | +    return mapping | 
|  | 43 | + | 
|  | 44 | +def main(**kwargs): | 
|  | 45 | + | 
|  | 46 | +    api = API( | 
|  | 47 | +            url = VALISPACE.get('domain'), | 
|  | 48 | +            username = VALISPACE.get('username'), | 
|  | 49 | +            password = VALISPACE.get('password') | 
|  | 50 | +        ) | 
|  | 51 | +     | 
|  | 52 | +    all_project_requirements = get_map(api, f"requirements/complete/?project="+str(DEFAULT_VALUES["project"]), "id") | 
|  | 53 | + | 
|  | 54 | +    nr_reqs = 0 #15093 | 
|  | 55 | +    nr_req_w_children = 0 #15094 | 
|  | 56 | +    nr_req_w_vm = 0 #15095 | 
|  | 57 | +    nr_req_vm_analysis = 0 #15099 | 
|  | 58 | +    nr_req_vm_review = 0 #15100 | 
|  | 59 | +    nr_req_vm_inspection = 0 #15101 | 
|  | 60 | +    nr_req_vm_rules = 0 #15102 | 
|  | 61 | +    nr_req_vm_tests = 0 #15103 | 
|  | 62 | +    nr_req_state_draft = 0 #15096 | 
|  | 63 | +    nr_req_state_in_review = 0 #15097 | 
|  | 64 | +    nr_req_state_final = 0 #15098 | 
|  | 65 | +    nr_req_verified = 0 #15104 | 
|  | 66 | +    nr_req_not_verified = 0 #15105 | 
|  | 67 | + | 
|  | 68 | +    nr_reqs = len(all_project_requirements) | 
|  | 69 | + | 
|  | 70 | +    for requirement in all_project_requirements: | 
|  | 71 | +        req_data=all_project_requirements[requirement] | 
|  | 72 | +         | 
|  | 73 | +        if req_data['verified'] == True: | 
|  | 74 | +            nr_req_verified += 1 | 
|  | 75 | +        elif req_data['verified'] == False: | 
|  | 76 | +            nr_req_not_verified += 1 | 
|  | 77 | + | 
|  | 78 | +        if req_data['total_children'] != 0: | 
|  | 79 | +            nr_req_w_children += 1 | 
|  | 80 | +         | 
|  | 81 | +        if req_data['state'] == 7: #but for other projects we need to get by state name | 
|  | 82 | +            nr_req_state_draft += 1 | 
|  | 83 | +        elif req_data['state'] == 8: | 
|  | 84 | +            nr_req_state_in_review += 1 | 
|  | 85 | +        elif req_data['state'] == 9: | 
|  | 86 | +            nr_req_state_final += 1 | 
|  | 87 | + | 
|  | 88 | +        if req_data['verification_methods'] != None: | 
|  | 89 | +            nr_req_w_vm += 1 | 
|  | 90 | +            for vm in req_data['verification_methods'] :                 | 
|  | 91 | +                if vm['method'] != None: | 
|  | 92 | +                    vm_name = vm['method']['name'] | 
|  | 93 | +                    if vm_name == "Analysis": | 
|  | 94 | +                        nr_req_vm_analysis += 1 | 
|  | 95 | +                    elif vm_name == "Inspection": | 
|  | 96 | +                        nr_req_vm_inspection += 1 | 
|  | 97 | +                    elif vm_name == "Review": | 
|  | 98 | +                        nr_req_vm_review += 1 | 
|  | 99 | +                    elif vm_name == "Rules": | 
|  | 100 | +                        nr_req_vm_rules += 1 | 
|  | 101 | +                    elif vm_name == "Test": | 
|  | 102 | +                        nr_req_vm_tests += 1 | 
|  | 103 | + | 
|  | 104 | +    print("going to patch") | 
|  | 105 | +    time.sleep(15) | 
|  | 106 | + | 
|  | 107 | +    api.request("patch", "valis/15093/", data={"formula":nr_reqs}) | 
|  | 108 | +    api.request("patch", "valis/15094/", data={"formula":nr_req_w_children}) | 
|  | 109 | +    api.request("patch", "valis/15095/", data={"formula":nr_req_w_vm})     | 
|  | 110 | +    api.request("patch", "valis/15096/", data={"formula":nr_req_state_draft}) | 
|  | 111 | +    api.request("patch", "valis/15097/", data={"formula":nr_req_state_in_review}) | 
|  | 112 | +    api.request("patch", "valis/15098/", data={"formula":nr_req_state_final})     | 
|  | 113 | +    api.request("patch", "valis/15099/", data={"formula":nr_req_vm_analysis}) | 
|  | 114 | +    api.request("patch", "valis/15100/", data={"formula":nr_req_vm_review}) | 
|  | 115 | +    api.request("patch", "valis/15101/", data={"formula":nr_req_vm_inspection}) | 
|  | 116 | +    api.request("patch", "valis/15102/", data={"formula":nr_req_vm_rules}) | 
|  | 117 | +    api.request("patch", "valis/15103/", data={"formula":nr_req_vm_tests}) | 
|  | 118 | +    api.request("patch", "valis/15104/", data={"formula":nr_req_verified}) | 
|  | 119 | +    api.request("patch", "valis/15105/", data={"formula":nr_req_not_verified}) | 
|  | 120 | + | 
|  | 121 | +    pass | 
|  | 122 | + | 
|  | 123 | +if __name__=='__main__': | 
|  | 124 | +    main() | 
0 commit comments