cronpy/core/influxdb.py

27 lines
790 B
Python

import influxdb
import setting
class InfluxDB:
def __init__(self, host=setting.INFLUX_HOST, port=setting.INFLUX_PORT, base=setting.INFLUX_BASE):
self.host = host
self.port = port
self.base = base
self._client = influxdb.InfluxDBClient(host=self.host, port=self.port, database=self.base)
def save_stats(self, module, start_time, nb_tasks, tasks_done, quantity, errors, total_time):
points = [{
'measurement': module,
'time': start_time,
'fields': {
'nbt': nb_tasks,
'tdn': tasks_done,
'qty': quantity,
'err': errors,
'tti': total_time
}
}]
self._client.write_points(points, time_precision='s')