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

   1 from importCommon import *
   2 from importNormativeElements import createNormativeElement
   3
   4
   5 #####################################################################################################################################################################################
   6 #                                                                                                                                                                                                                                                                                                                                                               #
   7 # Import tosca data types                                                                                                                                                                                                                                                                                                                       #
   8 #                                                                                                                                                                                                                                                                                                                                                                       #
   9 # activation :                                                                                                                                                                                                                                                                                                                                          #
  10 #       python importGroupTypes.py [-s <scheme> | --scheme=<scheme> ] [-i <be host> | --ip=<be host>] [-p <be port> | --port=<be port> ] [-f <input file> | --ifile=<input file> ]  #
  11 #                                                                                                                                                                                                                                                                                                                                                                       #
  12 # shortest activation (be host = localhost, be port = 8080):                                                                                                                                                                                                                                            #
  13 #               python importGroupTypes.py [-f <input file> | --ifile=<input file> ]                                                                                                                                                                                                    #
  14 #                                                                                                                                                                                                                                                                                                                                                               #
  15 #####################################################################################################################################################################################
  16
  17 def usage():
  18     print sys.argv[0], \
  19         '[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> ]'
  20
  21
  22 def importGroupTypes(scheme, be_host, be_port, admin_user, exit_on_success, file_dir):
  23     result = createNormativeElement(scheme, be_host, be_port, admin_user, file_dir,
  24                                     "/sdc2/rest/v1/catalog/uploadType/grouptypes", "groupTypes", "groupTypesZip", True)
  25
  26     print_frame_line()
  27     print_name_and_return_code(result[0], result[1])
  28     print_frame_line()
  29
  30     if result[1] is None or result[1] not in [200, 201, 409]:
  31         error_and_exit(1, None)
  32     else:
  33         if exit_on_success:
  34             error_and_exit(0, None)
  35
  36
  37 def main(argv):
  38     print 'Number of arguments:', len(sys.argv), 'arguments.'
  39
  40     be_host = 'localhost'
  41     be_port = '8080'
  42     admin_user = 'jh0003'
  43     scheme = 'http'
  44
  45     try:
  46         opts, args = getopt.getopt(argv, "i:p:u:h:s:", ["ip=", "port=", "user=", "scheme="])
  47     except getopt.GetoptError:
  48         usage()
  49         error_and_exit(2, 'Invalid input')
  50
  51     for opt, arg in opts:
  52         # print opt, arg
  53         if opt == '-h':
  54             usage()
  55             sys.exit(3)
  56         elif opt in ("-i", "--ip"):
  57             be_host = arg
  58         elif opt in ("-p", "--port"):
  59             be_port = arg
  60         elif opt in ("-u", "--user"):
  61             admin_user = arg
  62         elif opt in ("-s", "--scheme"):
  63             scheme = arg
  64
  65     print 'scheme =', scheme, ', be host =', be_host, ', be port =', be_port, ', user =', admin_user
  66
  67     if be_host is None:
  68         usage()
  69         sys.exit(3)
  70
  71     importGroupTypes(scheme, be_host, be_port, admin_user, True, "../../../import/tosca/group-types/")
  72
  73
  74 if __name__ == "__main__":
  75     main(sys.argv[1:])