32 lines
1.5 KiB
Python
32 lines
1.5 KiB
Python
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
|
|
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)
|
|
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.info('Influxdb version: %s', version)
|
|
|
|
if 'varken' not in databases:
|
|
self.logger.info("Creating varken database")
|
|
self.influx.create_database('varken')
|
|
|
|
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)
|