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)