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