d_mcknight/Varken
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix uncaught influxdb connection error

Browse source
This commit is contained in:
Nicholas St. Germain 2019-04-19 11:29:27 -05:00
parent 1b1a977ba6
commit 663b399a97
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
varken/dbmanager.py
View file

@ -1,3 +1,4 @@
from sys import exit
from logging import getLogger from logging import getLogger
from influxdb import InfluxDBClient from influxdb import InfluxDBClient
from requests.exceptions import ConnectionError from requests.exceptions import ConnectionError
@ -10,10 +11,15 @@ class DBManager(object):
self.influx = InfluxDBClient(host=self.server.url, port=self.server.port, username=self.server.username, self.influx = InfluxDBClient(host=self.server.url, port=self.server.port, username=self.server.username,
password=self.server.password, ssl=self.server.ssl, database='varken', password=self.server.password, ssl=self.server.ssl, database='varken',
verify_ssl=self.server.verify_ssl) verify_ssl=self.server.verify_ssl)
version = self.influx.request('ping', expected_response_code=204).headers['X-Influxdb-Version']
databases = [db['name'] for db in self.influx.get_list_database()]
self.logger = getLogger() self.logger = getLogger()
try:
version = self.influx.request('ping', expected_response_code=204).headers['X-Influxdb-Version']
self.logger.info('Influxdb version: %s', version) self.logger.info('Influxdb version: %s', version)
except ConnectionError:
self.logger.critical("Error testing connection to InfluxDB. Please check your url/hostname")
exit()
databases = [db['name'] for db in self.influx.get_list_database()]
if 'varken' not in databases: if 'varken' not in databases:
self.logger.info("Creating varken database") self.logger.info("Creating varken database")