aboutsummaryrefslogtreecommitdiffstats
path: root/catalog-be/src/main/resources/scripts/import/tosca/upgradeHeatAndNormativeTypes.py
blob: e35e2d28ebbdca443c28caebaf15a5ead02d62d9 (plain)
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
99
100
101
102
103
104
105
106
107
108
109
110
111
import pycurl
import sys, getopt
from StringIO import StringIO
import json
import copy
from importCommon import *
from importNormativeTypes import *
import importCommon
import json


################################################################################################################################################
#																																		       #
# Upgrades all Heat and Normative types confiugred in "typesToUpgrade.json" file																										   #
# 																																			   #
# activation :																																   #
#       python upgradeHeatAndNormativeTypes.py [-i <be host> | --ip=<be host>] [-p <be port> | --port=<be port> ] [-f <input file> | --ifile=<input file> ]     #
#																																		  	   #
# shortest activation (be host = localhost, be port = 8080): 																				   #
#		python upgradeHeatAndNormativeTypes.py [-f <input file> | --ifile=<input file> ]												 				           #
#																																		       #
################################################################################################################################################
def upgradeTypesPerConfigFile(beHost, bePort, adminUser, baseDir, updateversion):
    responseCodes = [200, 201]
    if (updateversion == 'false'):
        responseCodes = [200, 201, 409]
    with open("typesToUpgrade.json", 'r') as stream:
        try:
            types = json.load(stream)
            heatTypes = types["heat"]
            debug(heatTypes)
            normativeTypes = types["normative"]
            debug(normativeTypes)
            heatFileDir = baseDir + "heat-types/"
            debug(heatFileDir)
            normativeFileDir = baseDir + "normative-types/"
            debug(normativeFileDir)
            results = []
            for heatType in heatTypes:
                result = createNormativeType(beHost, bePort, adminUser, heatFileDir, heatType.encode('ascii', 'ignore'), updateversion)
                results.append(result)
                if (result[1] == None or result[1] not in responseCodes):
                    print "Failed creating heat type " + heatType + ". " + str(result[1])
            for normativeType in normativeTypes:
                result = createNormativeType(beHost, bePort, adminUser, normativeFileDir, normativeType.encode('ascii', 'ignore'), updateversion)
                results.append(result)
                if (result[1] == None or result[1] not in responseCodes):
                    print "Failed creating normative type " + normativeType + ". " + str(result[1])
            return results
        except yaml.YAMLError as exc:
            print(exc)



def main(argv):
    print 'Number of arguments:', len(sys.argv), 'arguments.'

    beHost = 'localhost'
    bePort = '8080'
    adminUser = 'jh0003'
    updateversion = 'true'

    try:
        opts, args = getopt.getopt(argv, "i:p:u:v:h:", ["ip=", "port=", "user=", "updateversion="])
    except getopt.GetoptError:
        usage()
        errorAndExit(2, 'Invalid input')

    for opt, arg in opts:
        # print opt, arg
        if opt == '-h':
            usage()
            sys.exit(3)
        elif opt in ("-i", "--ip"):
            beHost = arg
        elif opt in ("-p", "--port"):
            bePort = arg
        elif opt in ("-u", "--user"):
            adminUser = arg
        elif opt in ("-v", "--updateversion"):
            if (arg.lower() == "false" or arg.lower() == "no"):
                updateversion = 'false'

    print 'be host =', beHost, ', be port =', bePort, ', user =', adminUser

    if (beHost == None):
        usage()
        sys.exit(3)

    results = upgradeTypesPerConfigFile(beHost, bePort, adminUser, "../../../import/tosca/", updateversion)

    print "-----------------------------"
    for result in results:
        print "{0:20} | {1:6}".format(result[0], result[1])
    print "-----------------------------"

    responseCodes = [200, 201]

    if (updateversion == 'false'):
        responseCodes = [200, 201, 409]

    failedNormatives = filter(lambda x: x[1] == None or x[1] not in responseCodes, results)
    if (len(failedNormatives) > 0):
        errorAndExit(1, None)
    else:
        errorAndExit(0, None)


if __name__ == "__main__":
    main(sys.argv[1:])