HydroBot/hydrobot-software Changeset - 514bbf7bb55e

Changeset - 514bbf7bb55e

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

matthewreed - 9 years ago 2016-07-23 22:31:04

Implemented Influxdb SeriesHelper for logging data

1 file changed with 38 insertions and 69 deletions:

hydrobot.py

0 comments (0 inline, 0 general)

hydrobot.py

➞

Show inline comments

 import sys
 import time
 import _thread
 import ast
 import configparser
 import datetime
 from canard import can, messaging
 from canard.hw import socketcan
 from canard.file import jsondb
 from influxdb import InfluxDBClient
 from influxdb import SeriesHelper
 from apscheduler.schedulers.background import BackgroundScheduler
 #TODO
 #fix temperature offset
 #database series helper
 #database time interval logging
 config = configparser.ConfigParser(allow_no_value = True)
 config.read("hydrobot.conf")
 DEBUG_CAN = config.getboolean("debug", "can")
 DEBUG_CAN_DETAIL = config.getboolean("debug", "can_detail")
 DEBUG_TIMER = config.getboolean("debug", "timer")
 class MySeriesHelper(SeriesHelper):
     # Meta class stores time series helper configuration.
     class Meta:
         # The client should be an instance of InfluxDBClient.
         #client = myclient
         # The series name must be a string. Add dependent fields/tags in curly brackets.
         series_name = '{measurement}'
         # Defines all the fields in this time series.
         fields = ['value']
         # Defines all the tags for the series.
         tags = ['measurement']
         # Defines the number of data points to store prior to writing on the wire.
         bulk_size = 5
         # autocommit must be set to True when using bulk_size
         autocommit = True
 class Database:
     def __init__(self):
         host = config.get("database", "host")
         port = config.get("database", "port")
         username = config.get("database", "username")
         password = config.get("database", "password")
         database = config.get("database", "database")
         self.name = config.get("system", "name")
         self.client = InfluxDBClient(host, port, username, password, database)
         MySeriesHelper.Meta.client = self.client
         MySeriesHelper.Meta.series_name = self.name + '.{measurement}'
         # To manually submit data points which are not yet written, call commit:
         #MySeriesHelper.commit()
     def log_data(self, msgdb, message):
         if message == msgdb.AirSense:
             json_body = [
+                {
                     "measurement": self.name + "_air_temp",
                     "fields": {
                         "value": (float)(message.Temperature.value)
+                    }
                 },
+                {
                     "measurement": self.name + "_air_humidity",
                     "fields": {
                         "value": (float)(message.Humidity.value)
+                    }
                 },
+                {
                     "measurement": self.name + "_air_pressure",
                     "fields": {
                         "value": (float)(message.Pressure.value)
+                    }
+                }
+            ]
             self.client.write_points(json_body)
             MySeriesHelper(measurement='air_temp', value=(float)(message.Temperature.value))
             MySeriesHelper(measurement='air_humidity', value=(float)(message.Humidity.value))
             MySeriesHelper(measurement='air_pressure', value=(float)(message.Pressure.value))
         if message == msgdb.RelayDriveIn:
             json_body = [
+                {
                     "measurement": self.name + "_input_1",
                     "fields": {
                         "value": (float)(message.Input1.value)
+                    }
                 },
+                {
                     "measurement": self.name + "_input_2",
                     "fields": {
                         "value": (float)(message.Input2.value)
+                    }
                 },
+                {
                     "measurement": self.name + "_input_3",
                     "fields": {
                         "value": (float)(message.Input3.value)
+                    }
                 },
+                {
                     "measurement": self.name + "_input_4",
                     "fields": {
                         "value": (float)(message.Input4.value)
+                    }
+                }
+            ]
             self.client.write_points(json_body)
             MySeriesHelper(measurement='input_1', value=(float)(message.Input1.value))
             MySeriesHelper(measurement='input_2', value=(float)(message.Input2.value))
             MySeriesHelper(measurement='input_3', value=(float)(message.Input3.value))
             MySeriesHelper(measurement='input_4', value=(float)(message.Input4.value))
         if message == msgdb.WaterSense:
             json_body = [
+                {
                     "measurement": self.name + "_water_level",
                     "fields": {
                         "value": (float)(message.PercentFull.value)
+                    }
                 },
+                {
                     "measurement": self.name + "_water_temp",
                     "fields": {
                         "value": (float)(message.Temperature.value)
+                    }
                 },
+                {
                     "measurement": self.name + "_water_raw_debug",
                     "fields": {
                         "value": (float)(message.RawDebug.value)
+                    }
+                }
+            ]
             self.client.write_points(json_body)
             MySeriesHelper(measurement='water_level', value=(float)(message.PercentFull.value))
             MySeriesHelper(measurement='water_temp', value=(float)(message.Temperature.value))
             MySeriesHelper(measurement='water_raw_debug', value=(float)(message.RawDebug.value))
 class CanBus:
     def __init__(self, database):
         self.database = database
         self.dev = socketcan.SocketCanDev("can0")
         parser = jsondb.JsonDbParser()
         self.msgdb = parser.parse('hydrobot_can.json')
@@ @@ -152,24 +119,25 @@ class CanBus: @@
         self.relay_send_msg.Output4.value = 1
         if DEBUG_CAN:
             print("Send CAN message! ID: " + hex(self.relay_send_msg.id))
         if DEBUG_CAN_DETAIL:
             print(self.relay_send_msg)
         self.dev.send(self.relay_send_msg.encode())
     def set_output(self, module, output, state):
         msg = self.msgdb.lookup_message(module)
         msg.lookup_signal(output).value = state
         self.dev.send(msg.encode())
 class Scheduler:
     def __init__(self, canbus):
         self.canbus = canbus
         self.apscheduler = BackgroundScheduler()
     def start(self):
         self.apscheduler.start()
         for section in config.sections():
             if "timer" in section:
                 items = config.items(section)
@@ @@ -203,24 +171,25 @@ class Scheduler: @@
         self.apscheduler.get_job(timer_off).pause()
         if DEBUG_TIMER:
             print("Turning " + timer_on + "! " + str(datetime.datetime.now().time()))
     def interval_on(self, module, output, timer_off, timer_on, off_duration):
         self.canbus.set_output(module, output, 0)
         self.apscheduler.get_job(timer_off).reschedule("interval", weeks = off_duration[0], days = off_duration[1], hours = off_duration[2], minutes = off_duration[3], seconds = off_duration[4])
         self.apscheduler.get_job(timer_off).resume()
         self.apscheduler.get_job(timer_on).pause()
         if DEBUG_TIMER:
             print("Turning " + timer_off + "! " + str(datetime.datetime.now().time()))
 def func():
     print(1)
 def main():
     database = Database()
     canbus = CanBus(database)
     canbus.start()
     canbus.start_receive()
     scheduler = Scheduler(canbus)

0 comments (0 inline, 0 general)