Initial commit

1 files changed, 113 insertions, 0 deletions

diff --git a/daemon.py b/daemon.py
new file mode 100755
index 0000000..dfa9bfa
--- /dev/null
+++ b/daemon.py
@@ -0,0 +1,113 @@
+#!/usr/bin/env python
+
+import ubus
+import sys
+import os
+import time
+from periphery import I2C
+import sqlite3
+import commands
+
+# 60mV per step
+power_supply_cut_off = 212
+power_supply_reaquare = 215
+battery_low = 202 
+battery_not_low = 206
+battery_high = 260
+battery_not_high = 250
+
+# TODO use __main__
+
+# Initialize I2C
+i2c = I2C("/dev/i2c-7")
+
+# Initialize sqlite
+db_connection = sqlite3.connect('/tmp/i2c-collect.db')
+db = db_connection.cursor()
+
+# Initialize ubus
+ubus.connect()
+
+def read_chunk(handler, data):
+    result = []
+    for row in db.execute('SELECT date, %s FROM collected_data WHERE date > %s' %  \
+            (data['field'], (int(time.time()) - data['range']))):
+        result.append((row[0], row[1]))
+    handler.reply({"data": result})
+
+ubus.add(
+    "i2c-collect", {
+        "read_data": {
+            "method": read_chunk,
+            "signature": {"range": ubus.BLOBMSG_TYPE_INT32, "field": ubus.BLOBMSG_TYPE_STRING},
+        },
+    }
+)
+
+db.execute('''CREATE TABLE IF NOT EXISTS collected_data (
+    date INTEGER NOT NULL,
+    temperature INTEGER NOT NULL,
+    light INTEGER NOT NULL,
+    voltage INTEGER NOT NULL,
+    trimmer INTEGER NOT NULL
+)''')
+
+def i2c_get(addr):
+    msg = [I2C.Message([addr]), I2C.Message([0x00], read=True)]
+    i2c.transfer(0x48, msg)
+    return msg[1].data[0]
+
+def load_data():
+    return {
+            'trimmer': i2c_get(0x40),
+            'temperature': i2c_get(0x41),
+            'light': i2c_get(0x42),
+            'voltage': i2c_get(0x43)
+        }
+
+def db_data(data):
+    db.execute('''INSERT INTO collected_data (date, temperature, light, voltage, trimmer) VALUES
+            (%s, %s, %s, %s, %s)''' % (str(int(time.time())), data['temperature'], data['light'], data['voltage'], data['trimmer']))
+    db_connection.commit()
+
+def ubus_data(data):
+    date = str(int(time.time()))
+    for k in iter(data.keys()):
+        ubus.send("i2c-collect", {k: (date, data[k])})
+
+notified_power_supply = False
+notified_low_battery = False
+notified_high_battery = False
+
+def check_data(data):
+    global notified_low_battery, notified_power_supply, notified_high_battery
+    for k in data:
+        if k == "voltage":
+            if data['voltage'] > power_supply_reaquare and notified_power_supply:
+                commands.getoutput('create_notification -s news "Sitove napajeni bylo obnoveno." "Power supply was reaquired."')
+                notified_power_supply = False
+            elif data['voltage'] <= power_supply_cut_off and not notified_power_supply: # 12.8V
+                commands.getoutput('create_notification -s error "Vypadlo sitove napajeni!" "Power supply was interrupted!"')
+                notified_power_supply = True
+            if data['voltage'] > battery_not_low and notified_low_battery:
+                notified_low_battery = False
+            elif data['voltage'] <= battery_low and not notified_low_battery: # 12V
+                volts = float(data['voltage']) * 0.0593
+                commands.getoutput('create_notification -s error "Nizke napetu akumulatoru! %.2f V" "Low voltage of the acumulator! %.2f V"' % (volts, volts))
+                notified_low_battery = True
+            if data['voltage'] <= battery_not_high and notified_high_battery:
+                notified_high_battery = False
+            elif data['voltage'] > battery_high and not notified_high_battery:
+                commands.getoutput('create_notification -s error "Vysoke napetu akumulatoru!" "High voltage of the acumulator!"')
+                notified_high_battery = True
+
+# TODO add to exit handler
+# i2c.close()
+
+while True:
+    ubus.loop(60000)
+    data = load_data()
+    #print(data)
+    db_data(data)
+    ubus_data(data)
+    check_data(data)