Initial commit

proba.py

@@ -0,0 +1,47 @@
+import psycopg2
+from psycopg2 import sql
+from datetime import datetime
+import requests
+from huey import SqliteHuey, crontab
+
+# Initialize scheduler
+huey = SqliteHuey(filename='demo.db')
+
+# Database connection
+CONNECTION = "postgres://postgres:postgres*@localhost:5432/example"
+conn = psycopg2.connect(CONNECTION)
+
+# Devices
+devices = {"livingroom": "192.168.1.56","bedroom":"192.168.1.57"}
+
+
+# Func for fetching data from device using REST API
+def fetch_data_from_device(device):
+    data = dict()
+    data["time"] = datetime.now()
+    data["device"] = device
+    r = requests.get("http://" + devices[device] + "/sensor/temperature")
+    data["temperature"] = r.json()['value']
+    r = requests.get("http://" + devices[device] + "/sensor/humidity")
+    data["humidity"] = r.json()['value']
+    return (data["time"], data["device"], data["temperature"], data["humidity"])
+
+
+# Func for inserting data to database
+def insert_data(conn, device):
+    data = fetch_data_from_device(device)
+    cursor = conn.cursor()
+    insert_query = sql.SQL(
+        "INSERT INTO conditions (time, device, temperature, humidity) "
+        "VALUES (%s, %s, %s, %s)"
+    )
+    cursor.execute(insert_query, data)
+    conn.commit()
+    cursor.close()
+    print("Done")
+
+
+@huey.periodic_task(crontab(minute='*/1'))
+def test():
+    for device in devices:
+        insert_data(conn, device)