139 lines
4.5 KiB
Python
139 lines
4.5 KiB
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.
|
|
"""
|
|
|
|
import json
|
|
import sys
|
|
|
|
from oslo_utils.encodeutils import safe_decode
|
|
from oslo_utils.encodeutils import safe_encode
|
|
from six.moves import urllib
|
|
import six.moves.urllib.parse as urlparse
|
|
|
|
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'
|
|
|
|
|
|
def format_response(req):
|
|
try:
|
|
json_value = json.loads(req.read())
|
|
if (len(json_value['results'][0]) > 0 and
|
|
'series' in json_value['results'][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 = urllib.request.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 = safe_encode(urlparse.urlencode(query_params))
|
|
try:
|
|
req = urllib.request.urlopen(uri, encoded_params)
|
|
return format_response(req)
|
|
|
|
except urllib.error.HTTPError:
|
|
uri = "{}&{}".format(uri, encoded_params)
|
|
req = urllib.request.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.items():
|
|
if not any(user[0] == name for user in users):
|
|
influxdb_get_post(uri=api_uri,
|
|
query=safe_decode("CREATE USER {0} WITH PASSWORD '{1}'"
|
|
.format(name, password)),
|
|
db=DBNAME)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|