3 from StringIO import StringIO
6 from importCommon import *
8 #########################################################################################################################################################################################
10 # Import all users from a given file #
13 # python importNormativeTypes.py [-s <scheme> | --scheme=<scheme> ] [-i <be host> | --ip=<be host>] [-p <be port> | --port=<be port> ] [-f <input file> | --ifile=<input file> ] #
14 # [-v <true|false> | --updateversion=<true|false>] #
15 # shortest activation (be host = localhost, be port = 8080): #
16 # python importNormativeTypes.py [-f <input file> | --ifile=<input file> ] #
18 #########################################################################################################################################################################################
20 def createNormativeType(scheme, beHost, bePort, adminUser, fileDir, ELEMENT_NAME, updateversion):
23 log("in create normative type ", ELEMENT_NAME)
24 debug("userId", adminUser)
25 debug("fileDir", fileDir)
30 url = scheme + '://' + beHost + ':' + bePort + '/sdc2/rest/v1/catalog/upload/multipart'
31 if updateversion != None:
32 url += '?createNewVersion=' + updateversion
36 adminHeader = 'USER_ID: ' + adminUser
37 #c.setopt(pycurl.HTTPHEADER, ['Content-Type: application/json', 'Accept: application/json', adminHeader])
38 c.setopt(pycurl.HTTPHEADER, [adminHeader])
41 path = fileDir + ELEMENT_NAME + "/" + ELEMENT_NAME + ".zip"
43 CURRENT_JSON_FILE=fileDir + ELEMENT_NAME + "/" + ELEMENT_NAME + ".json"
44 #sed -i 's/"userId": ".*",/"userId": "'${USER_ID}'",/' ${CURRENT_JSON_FILE}
46 jsonFile = open(CURRENT_JSON_FILE)
48 debug("before load json")
49 json_data = json.load(jsonFile, strict=False)
52 jsonAsStr = json.dumps(json_data)
54 send = [('resourceMetadata', jsonAsStr), ('resourceZip', (pycurl.FORM_FILE, path))]
56 c.setopt(pycurl.HTTPPOST, send)
58 #data = json.dumps(user)
59 #c.setopt(c.POSTFIELDS, data)
62 c.setopt(c.SSL_VERIFYPEER, 0)
64 #c.setopt(c.WRITEFUNCTION, lambda x: None)
65 c.setopt(c.WRITEFUNCTION, buffer.write)
66 #print("before perform")
69 #print("Before get response code")
70 httpRes = c.getinfo(c.RESPONSE_CODE)
72 debug("http response=", httpRes)
73 #print('Status: ' + str(responseCode))
74 debug(buffer.getvalue())
77 return (ELEMENT_NAME, httpRes, buffer.getvalue())
79 except Exception as inst:
80 print("ERROR=" + str(inst))
81 return (ELEMENT_NAME, None, None)
85 print sys.argv[0], '[optional -s <scheme> | --scheme=<scheme>, default http] [-i <be host> | --ip=<be host>] [-p <be port> | --port=<be port> ] [-u <user userId> | --user=<user userId> ] [-v <true|false> | --updateversion=<true|false>]'
88 def importNormativeTypes(scheme, beHost, bePort, adminUser, fileDir, updateversion):
90 normativeTypes = [ "root", "compute", "softwareComponent", "webServer", "webApplication", "DBMS", "database", "objectStorage", "blockStorage", "containerRuntime", "containerApplication", "loadBalancer", "port", "network", "allottedResource"]
91 #normativeTypes = [ "root" ]
92 responseCodes = [200, 201]
94 if(updateversion == 'false'):
95 responseCodes = [200, 201, 409]
98 for normativeType in normativeTypes:
99 result = createNormativeType(scheme, beHost, bePort, adminUser, fileDir, normativeType, updateversion)
100 results.append(result)
101 if ( result[1] == None or result[1] not in responseCodes ):
102 print "Failed creating normative type " + normativeType + ". " + str(result[1])
107 print 'Number of arguments:', len(sys.argv), 'arguments.'
112 updateversion = 'true'
116 opts, args = getopt.getopt(argv,"i:p:u:v:h:s:",["ip=","port=","user=","updateversion=","scheme="])
117 except getopt.GetoptError:
119 errorAndExit(2, 'Invalid input')
121 for opt, arg in opts:
126 elif opt in ("-i", "--ip"):
128 elif opt in ("-p", "--port"):
130 elif opt in ("-u", "--user"):
132 elif opt in ("-s", "--scheme"):
134 elif opt in ("-v", "--updateversion"):
135 if (arg.lower() == "false" or arg.lower() == "no"):
136 updateversion = 'false'
138 print 'scheme =',scheme,', be host =',beHost,', be port =', bePort,', user =', adminUser, ', updateversion =', updateversion
140 if ( beHost == None ):
144 results = importNormativeTypes(scheme, beHost, bePort, adminUser, "../../../import/tosca/normative-types/", updateversion)
146 print "-----------------------------"
147 for result in results:
148 print "{0:20} | {1:6}".format(result[0], result[1])
149 print "-----------------------------"
151 responseCodes = [200, 201]
153 if(updateversion == 'false'):
154 responseCodes = [200, 201, 409]
156 failedNormatives = filter(lambda x: x[1] == None or x[1] not in responseCodes, results)
157 if (len(failedNormatives) > 0):
158 errorAndExit(1, None)
160 errorAndExit(0, None)
163 if __name__ == "__main__":