Varken/varken/dbmanager.py
Nicholas St. Germain 61fcf3b80c major pep8 cleanup.
funneled scope of imports where possible
fixed #59
Added fstrings where possible
added logging both temporarily and in secondary places
2018-12-17 19:12:37 -06:00

23 lines
895 B
Python

from logging import getLogger
from influxdb import InfluxDBClient
class DBManager(object):
def __init__(self, server):
self.server = server
self.influx = InfluxDBClient(self.server.url, self.server.port, self.server.username, self.server.password,
'varken')
databases = [db['name'] for db in self.influx.get_list_database()]
self.logger = getLogger()
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)
self.influx.write_points(d)