mirror of
https://github.com/ferdzo/iotDashboard.git
synced 2026-04-05 01:06:24 +00:00
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
import dotenv
|
|
import psycopg2
|
|
from psycopg2 import sql
|
|
from datetime import datetime
|
|
import requests
|
|
from huey import SqliteHuey, crontab
|
|
from dotenv import load_dotenv
|
|
from pathlib import Path
|
|
|
|
|
|
# Initialize scheduler
|
|
huey = SqliteHuey(filename='demo.db')
|
|
dotenv_path = Path("iotDashboard/.env")
|
|
load_dotenv(dotenv_path=dotenv_path)
|
|
CONNECTION = dotenv.dotenv_values(dotenv_path)["CONNECTION_STRING"]
|
|
# Database connection
|
|
conn = psycopg2.connect(CONNECTION)
|
|
|
|
# Devices
|
|
devices = {"livingroom": "192.168.244.131"}
|
|
|
|
|
|
# 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/tempreature")
|
|
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)
|