catalog-be/src/main/resources/scripts/import/tosca/upgradeNfvTypes.py

   1 from importNormativeTypes import *
   2
   3
   4 ################################################################################################################################################################################################
   5 #                                                                                                                                                                                                                                                                                                                                      #
   6 # Upgrades all Heat and Normative types confiugred in "typesToUpgrade.json" file                                                                                                                                                                                                                       #
   7 #                                                                                                                                                                                                                                                                                                                                          #
   8 # activation                                                                                                                                                                                   #
   9 #       python upgradeHeatAndNormativeTypes.py [-s <scheme> | --scheme=<scheme> ] [-i <be host> | --ip=<be host>] [-p <be port> | --port=<be port> ] [-f <input file> | --ifile=<input file> ] #
  10 #                                                                                                                                                                                                                                                                                                                                          #
  11 # shortest activation (be host = localhost, be port = 8080):                                                                                                                                                                                                               #
  12 #               python upgradeHeatAndNormativeTypes.py [-f <input file> | --ifile=<input file> ]                                                                                                                                                                       #
  13 #                                                                                                                                                                                                                                                                                                                                      #
  14 ################################################################################################################################################################################################
  15
  16
  17 def upgradeNfvTypesPerConfigFile(scheme, be_host, be_port, admin_user, base_dir, update_version):
  18     response_codes = [200, 201]
  19     results = []
  20
  21     if update_version == 'false':
  22         response_codes = [200, 201, 409]
  23     with open("onapTypesToUpgrade.json", 'r') as stream:
  24         try:
  25             types = json.load(stream)
  26             nfv_types = types["nfv"]
  27             debug(nfv_types)
  28             nfv_file_dir = base_dir + "nfv-types/"
  29             debug(nfv_file_dir)
  30             for nfvType in nfv_types:
  31                 result = createNormativeType(scheme, be_host, be_port, admin_user, nfv_file_dir,
  32                                              nfvType.encode('ascii', 'ignore'), update_version)
  33                 results.append(result)
  34                 if result[1] is None or result[1] not in response_codes:
  35                     print "Failed creating nfv type " + nfvType + ". " + str(result[1])
  36         except Exception as exc:
  37             print(exc)
  38
  39     return results
  40
  41
  42 def main(argv):
  43     print 'Number of arguments:', len(sys.argv), 'arguments.'
  44
  45     be_host = 'localhost'
  46     be_port = '8080'
  47     admin_user = 'jh0003'
  48     update_version = 'true'
  49     scheme = 'http'
  50
  51     try:
  52         opts, args = getopt.getopt(argv, "i:p:u:v:h:s:", ["ip=", "port=", "user=", "updateversion=", "scheme="])
  53     except getopt.GetoptError:
  54         usage()
  55         error_and_exit(2, 'Invalid input')
  56
  57     for opt, arg in opts:
  58         # print opt, arg
  59         if opt == '-h':
  60             usage()
  61             sys.exit(3)
  62         elif opt in ("-i", "--ip"):
  63             be_host = arg
  64         elif opt in ("-p", "--port"):
  65             be_port = arg
  66         elif opt in ("-u", "--user"):
  67             admin_user = arg
  68         elif opt in ("-s", "--scheme"):
  69             scheme = arg
  70         elif opt in ("-v", "--updateversion"):
  71             if arg.lower() == "false" or arg.lower() == "no":
  72                 update_version = 'false'
  73
  74     print 'scheme =', scheme, ', be host =', be_host, ', be port =', be_port, ', user =', admin_user
  75
  76     if be_host is None:
  77         usage()
  78         sys.exit(3)
  79
  80     results = upgradeNfvTypesPerConfigFile(scheme, be_host, be_port, admin_user, "../../../import/tosca/",
  81                                            update_version)
  82
  83     print "-----------------------------"
  84     for result in results:
  85         print "{0:20} | {1:6}".format(result[0], result[1])
  86     print "-----------------------------"
  87
  88     response_codes = [200, 201]
  89
  90     if update_version == 'false':
  91         response_codes = [200, 201, 409]
  92
  93     failed_normatives = filter(lambda x: x[1] is None or x[1] not in response_codes, results)
  94     if len(list(failed_normatives)) > 0:
  95         error_and_exit(1, None)
  96     else:
  97         error_and_exit(0, None)
  98
  99
 100 if __name__ == "__main__":
 101     main(sys.argv[1:])