from flask import Flask
from flask import request
from datetime import datetime, timezone
import mysql.connector
import json

import os

app = Flask(__name__)
cnx_local = mysql.connector.connect(
    user=os.environ["CLIENTDB_USERNAME"],
    password=os.environ["CLIENTDB_PASSWORD"], 
    host=os.environ["CLIENTDB_HOST"], 
    database=os.environ["CLIENTDB_SCHEMA"]
)
cnx_main = mysql.connector.connect(
    user=os.environ["MAINDB_USERNAME"],
    password=os.environ["MAINDB_PASSWORD"], 
    host=os.environ["MAINDB_HOST"], 
    database=os.environ["MAINDB_SCHEMA"]
)

@app.route("/detect", methods=["POST"])
def detect():
    hyp = json.loads(request.json["hyperparam"])
    model_id = int(hyp["modelid"])
    data_id = int(hyp["datasetid"])
    job_id = int(request.json["jobid"])
    
    cur_time = datetime.now(tz=timezone.utc)
    exc_state = "INSERT INTO job_status (id_job, status, create_time, update_time) VALUES (%s, %s, %s, %s)"
    exc_val = (job_id, "working", cur_time, cur_time)

    cursor = cnx_main.cursor()
    cursor.execute(exc_state, exc_val)
    cnx_main.commit()
    cursor.close()
    
    os.system(f"nohup python -u main_predict.py --jobid {job_id} --modelid {model_id} --datasetid {data_id} >> client_out.log 2>&1 &")
    
    return "Job started!"