vroClientScripts/runWorkflow.py at master · burkeazbill/vroClientScripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#!/usr/bin/env python

"""

	Invoke a vRO workflow via the REST API. Added capability to specify arguments at the command line.

	If you omit the username and password arguments it will prompt for them. The password will be masked.

"""

# TODO needs testing by someone other than me.
# TODO build a script to retrieve a list of available workflows


import argparse
import requests
import json
import getpass
from base64 import b64encode

def getargs():
    parser = argparse.ArgumentParser()
    parser.add_argument('-s', '--server',
                        required=True,
                        action='store',
                        help='vRealize Orchestrator server in format: FQDN:port or IP:port')
    parser.add_argument('-u', '--username',
                        required=False,
                        action='store',
                        help='Username to access vRealize Orchestrator')
    parser.add_argument('-p', '--password',
                        required=False,
                        action='store',
                        help='The password used to access vRealize Orchestrator')
    parser.add_argument('-w', '--workflow',
                        required=True,
                        action='store',
                        help='Workflow ID')
    parser.add_argument('-j', '--json',
                        required=True,
                        action='store',
                        dest='input_json',
                        help='Path to JSON post body')
    parser.add_argument('--nosslverify',
    					required=False,
    					action='store_true',
    					help='Stop SSL certificate validation')
    args = parser.parse_args()
    return args

def run_workflow(server, username, password, workflow, input_json, ssl_verify):
    try:

        authentication = b64encode(username + ':' + password).decode("ascii")

        with open(input_json, 'r') as f:
            postdata = json.load(f)

        r = requests.post(url='https://' + server + '/vco/api/workflows/' + workflow + '/executions/',
                          verify=ssl_verify,
                          headers= {'Authorization': 'Basic ' + authentication,
                                    'Content-Type': 'application/json',
                                    'Accept': 'application/json'},
                          data = json.dumps(postdata)
                          )

        print r.status_code
        print r.content

    except requests.RequestException as e:
        print e


def main():
    args = getargs()
    server = args.server
    username = args.username
    password = args.password
    workflow = args.workflow
    input_json = args.input_json
    nosslverify = args.nosslverify

    if not username:
        username = raw_input("target host username:")

    if not password:
        password_request = "Target host '%s' Password: " % username
        password = getpass.getpass(password_request)

    if nosslverify:
        ssl_verify = False
        run_workflow(server, username, password, workflow, input_json, ssl_verify)
    else:
        run_workflow(server, username, password, workflow, input_json, ssl_verify=True)

if __name__ == '__main__':
    main()