45 lines
2.1 KiB
Python
45 lines
2.1 KiB
Python
from sys import exit
|
|
from logging import getLogger
|
|
from influxdb import InfluxDBClient
|
|
from requests.exceptions import ConnectionError
|
|
from influxdb.exceptions import InfluxDBServerError
|
|
|
|
|
|
class DBManager(object):
|
|
def __init__(self, server):
|
|
self.server = server
|
|
if self.server.url == "influxdb.domain.tld":
|
|
self.logger.critical("You have not configured your varken.ini. Please read Wiki page for configuration")
|
|
exit()
|
|
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',
|
|
verify_ssl=self.server.verify_ssl)
|
|
self.logger = getLogger()
|
|
try:
|
|
version = self.influx.request('ping', expected_response_code=204).headers['X-Influxdb-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:
|
|
self.logger.info("Creating varken database")
|
|
self.influx.create_database('varken')
|
|
|
|
retention_policies = [policy['name'] for policy in self.influx.get_list_retention_policies(database='varken')]
|
|
|
|
# Fix name bug by adding retention policy retroactively
|
|
if 'varken 30d-1h' not in retention_policies:
|
|
self.logger.info("Creating varken retention policy (30d-1h)")
|
|
self.influx.create_retention_policy('varken 30d-1h', '30d', '1', 'varken', False, '1h')
|
|
|
|
def write_points(self, data):
|
|
d = data
|
|
self.logger.debug('Writing Data to InfluxDB %s', d)
|
|
try:
|
|
self.influx.write_points(d)
|
|
except (InfluxDBServerError, ConnectionError) as e:
|
|
self.logger.error('Error writing data to influxdb. Dropping this set of data. '
|
|
'Check your database! Error: %s', e)
|