88cbf6b44b
In InfluxDB version 0.11.0 the result set for SHOW SERIES changed. See https://github.com/influxdata/influxdb/blob/master/CHANGELOG.md#v0110-2016-03-22 Modifed the code to be compatible with either the old or new result set. In InfluxDB version 1.0.0 the config file changed. See https://github.com/influxdata/influxdb/blob/master/CHANGELOG.md#v100-2016-09-08 Modified the config file deployed by DevStack to be compatible with either the old or new config file. In some other InfluxDB release several database ops, such as CREATE DATABASE, were modified from http GET to POST. Modified how databases are created in the DevStack plugin.sh. Change-Id: I33188f4e64150d2cb24c0a06979c06f03c803da4
131 lines
4.4 KiB
Python
131 lines
4.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
#
|
|
# (C) Copyright 2015,2016 Hewlett Packard Enterprise Development LP
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
# implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
"""A simple script to setup influxdb user and roles. At some point this should
|
|
become a more full featured module. Also this assumes that none of the
|
|
python based influxdb clients are available on this system.
|
|
"""
|
|
|
|
ADMIN = 'root'
|
|
ADMIN_PASS = 'root'
|
|
DBNAME = 'mon'
|
|
USERS = {}
|
|
USERS['mon_api'] = 'password'
|
|
USERS['mon_persister'] = 'password'
|
|
|
|
URL = 'http://127.0.0.1:8086'
|
|
|
|
SHARDSPACE_NAME = 'persister_all'
|
|
REPLICATION = 1
|
|
RETENTION = '90d'
|
|
|
|
import json
|
|
import sys
|
|
import time
|
|
import six.moves.urllib.parse as urlparse
|
|
import urllib2
|
|
|
|
|
|
def format_response(req):
|
|
try:
|
|
json_value = json.loads(req.read())
|
|
if (len(json_value['results'][0]) > 0 and
|
|
'values' in json_value['results'][0]['series'][0]):
|
|
return json_value['results'][0]['series'][0]['values']
|
|
else:
|
|
return []
|
|
except KeyError:
|
|
print "Query returned a non-successful result: {0}".format(json_value['results'])
|
|
raise
|
|
|
|
def influxdb_get(uri, query, db=None):
|
|
"""Runs a query via HTTP GET and returns the response as a Python list."""
|
|
|
|
getparams = {"q": query}
|
|
if db:
|
|
getparams['db'] = db
|
|
|
|
try:
|
|
params = urlparse.urlencode(getparams)
|
|
uri = "{}&{}".format(uri,params)
|
|
req = urllib2.urlopen(uri)
|
|
return format_response(req)
|
|
|
|
except KeyError:
|
|
sys.exit(1)
|
|
|
|
def influxdb_get_post(uri, query, db=None):
|
|
"""Runs a query using HTTP GET or POST and returns the response as a Python list.
|
|
At some InfluxDB release several ops changed from using GET to POST. For example,
|
|
CREATE DATABASE. To maintain backward compatibility, this function first trys the
|
|
query using POST and if that fails it retries again using GET."""
|
|
|
|
query_params = {"q": query}
|
|
if db:
|
|
query_params['db'] = db
|
|
|
|
try:
|
|
encoded_params = urlparse.urlencode(query_params)
|
|
|
|
try:
|
|
req = urllib2.urlopen(uri, encoded_params)
|
|
return format_response(req)
|
|
|
|
except urllib2.HTTPError:
|
|
uri = "{}&{}".format(uri, encoded_params)
|
|
req = urllib2.urlopen(uri)
|
|
return format_response(req)
|
|
|
|
except KeyError:
|
|
sys.exit(1)
|
|
|
|
def main(argv=None):
|
|
"""If necessary, create the database, retention policy, and users"""
|
|
auth_str = '?u=%s&p=%s' % (ADMIN, ADMIN_PASS)
|
|
api_uri = "{0}/query{1}".format(URL, auth_str)
|
|
|
|
# List current databases
|
|
dbs = influxdb_get(uri=api_uri, query="SHOW DATABASES")
|
|
if [DBNAME] not in dbs:
|
|
print "Creating database '{}'".format(DBNAME)
|
|
influxdb_get_post(uri=api_uri, query="CREATE DATABASE {0}".format(DBNAME))
|
|
print "...created!"
|
|
|
|
# Check retention policy
|
|
policies = influxdb_get(uri=api_uri,
|
|
query="SHOW RETENTION POLICIES ON {0}".format(DBNAME))
|
|
if not any(pol[0] == SHARDSPACE_NAME for pol in policies):
|
|
# Set retention policy
|
|
policy = "CREATE RETENTION POLICY {0} ON {1} DURATION {2} REPLICATION {3} DEFAULT".format(SHARDSPACE_NAME,
|
|
DBNAME,
|
|
RETENTION,
|
|
REPLICATION)
|
|
influxdb_get_post(uri=api_uri, db=DBNAME, query=policy)
|
|
|
|
# Create the users
|
|
users = influxdb_get(uri=api_uri, query="SHOW USERS", db=DBNAME)
|
|
for name, password in USERS.iteritems():
|
|
if not any(user[0] == name for user in users):
|
|
influxdb_get_post(uri=api_uri,
|
|
query=unicode("CREATE USER {0} WITH PASSWORD '{1}'".format(name, password)),
|
|
db=DBNAME)
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|