from backend_iot import app
from backend_iot.db import getLatestRecord, addRecord, get_aqi
from flask import request, redirect, render_template, abort
from playhouse.shortcuts import model_to_dict
from datetime import datetime

current_latitude, current_longitude = 45.46437891252755, 7.872049153560152

def update_record(record):
    record.update()
    record.latitude = current_latitude
    record.longitude = current_longitude
    record.timestamp = datetime.now()
    record.aqi = get_aqi(current_latitude, current_longitude)
    return record

def latest_or_abort():
    latest = getLatestRecord()
    if latest is None:
        return abort(400)
    return latest

@app.route("/")
def index_route():
    if request.method == 'GET':
        return render_template("index.html")

editable_fields = ["t_target", "power", "fan_speed", "auto", "airflow"]

@app.route("/data", methods = ['GET', 'POST'])
def data_route():

    if request.method == "GET":
        return latest_or_abort()
    
    latest = getLatestRecord()
    form = request.form

    for key in form.keys():
        if key not in latest.keys():
            continue
        if key == "fan_speed" and latest["auto"]:
            continue
        latest[key] = form[key]

    new_record = update_record(latest)
    temp = model_to_dict(new_record)
    del temp["id"]
    addRecord(**temp)
    return temp