|
@@ -37,16 +37,20 @@ def isinteger(value):
|
|
|
return False
|
|
|
|
|
|
|
|
|
-def loadCsv(inputfilename, servername, user, password, dbname, metric, timecolumn, timeformat, tagcolumns, fieldcolumns, usegzip, delimiter, batchsize, datatimezone):
|
|
|
+def loadCsv(inputfilename, servername, user, password, dbname, metric,
|
|
|
+ timecolumn, timeformat, tagcolumns, fieldcolumns, usegzip,
|
|
|
+ delimiter, batchsize, create, datatimezone):
|
|
|
|
|
|
host = servername[0:servername.rfind(':')]
|
|
|
port = int(servername[servername.rfind(':')+1:])
|
|
|
client = InfluxDBClient(host, port, user, password, dbname)
|
|
|
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
+ if(create == True):
|
|
|
+ print('Deleting database %s'%dbname)
|
|
|
+ client.drop_database(dbname)
|
|
|
+ print('Creating database %s'%dbname)
|
|
|
+ client.create_database(dbname)
|
|
|
+
|
|
|
client.switch_user(user, password)
|
|
|
|
|
|
|
|
@@ -137,6 +141,9 @@ if __name__ == "__main__":
|
|
|
parser.add_argument('--dbname', nargs='?', required=True,
|
|
|
help='Database name.')
|
|
|
|
|
|
+ parser.add_argument('--create', action='store_true', default=False,
|
|
|
+ help='Drop database and create a new one.')
|
|
|
+
|
|
|
parser.add_argument('-m', '--metricname', nargs='?', default='value',
|
|
|
help='Metric column name. Default: value')
|
|
|
|
|
@@ -164,4 +171,5 @@ if __name__ == "__main__":
|
|
|
args = parser.parse_args()
|
|
|
loadCsv(args.input, args.server, args.user, args.password, args.dbname,
|
|
|
args.metricname, args.timecolumn, args.timeformat, args.tagcolumns,
|
|
|
- args.fieldcolumns, args.gzip, args.delimiter, args.batchsize, args.timezone)
|
|
|
+ args.fieldcolumns, args.gzip, args.delimiter, args.batchsize, args.create,
|
|
|
+ args.timezone)
|