from sklearn.linear_model import LinearRegression
import pandas as pd 
import numpy as np
import requests
import logging


# --------------- Fonction -----------------  #
def generateJsonModel(model:LinearRegression):
    listCoef = []
    listIntercept = []
    for i in range(0,len(model.coef_)):
        listCoef.append(model.coef_[i][0])
        listIntercept.append(model.intercept_[i])
    json = {"coef":listCoef,"intercept":listIntercept}
    return json

def generateModele(dataJson:dict[str,str]):
    # -- Préparation des données 
    arrayBpm = []
    arrayStartTime = []
    arrayTimeOfActivity = []

    for data in dataJson["Data"]:
        arrayBpm.append(data["BpmAvg"])
        arrayTimeOfActivity.append(data["TimeOfActivity"])

        arrayStartTime.append(data["StartTime"])
    # -- DataFrame 
    data = pd.DataFrame({
        "Bpm": arrayBpm,
        "TimeOfActivity": arrayTimeOfActivity
    })
    # -- Régression linéaire 
    model = LinearRegression()
    model.fit(np.array(arrayStartTime).reshape(-1,1),data)
    return model


def getUserWithData(url:str):
    response = requests.get(urlGetAllData)
    if ( response.status_code != 200):
        print('problème lors de l extraction des données avec l api !! ->  "getUserWithData" (status_code != 200)')
        exit()
    return response.json

def sendJsonToApi(url,json):
    response = requests.post(url,json)
    if ( response.status_code != 200):
        print('Problème lors de l envoi des données avec l api !! -> "sendJsonToApi" (status_code != 200)')
        exit()
    return

# ---------------- Main ------------------- #
logging.error("RUNNNNNNNN !")

urlGetAllData = "https://codefirst.iut.uca.fr/containers/SmartFit-smartfit_api/ia/data"

jsonBack = { "Users" : []}
# --- Call Api 
dataUser = getUserWithData(url=urlGetAllData)
for user in dataUser["Users"]:
    jsonTmp = {}

    jsonTmp["Identifiant"] = user["Identifiant"]
    jsonTmp["Info"] = []

    for category in user["Info"]:
        #Mettre la condition longueur ici
        model = generateModele(category)
        jsonTmp["Info"].append({"Category": category["Category"],"Model" : generateJsonModel(model)})
    # Add User
    jsonBack["Users"].append(jsonTmp)

# -- Send Api 
sendJsonToApi(urlGetAllData,jsonBack)