from flask import Flask, jsonify, request
from flask_cors import CORS
import os
from dotenv import load_dotenv
from db import get_connection

load_dotenv()

PORT = os.getenv("PORT")

app = Flask(__name__)
CORS(app)

@app.route("/health")
def health():
    return jsonify({"message" : "API Flask version 1.0!!!"})

@app.route("/produtos",methods=["GET"])
def listar_produtos():
    conexao = get_connection()
    cursor = conexao.cursor(dictionary=True)

    cursor.execute("Select * from produto")
    produtos = cursor.fetchall()

    cursor.close()
    conexao.close()

    return jsonify(produtos)

@app.route("/produtos/<int:id>", methods=["GET"])
def listar_produto_id(id):
    conexao = get_connection()
    cursor = conexao.cursor(dictionary=True)

    cursor.execute("Select * from produto where id = %s", (id))
    produto = cursor.fetchone()

    cursor.close()
    conexao.close()

    if produto is None:
        return jsonify({"erro" : "Produto nao encontrado"}), 404

    return jsonify(produto)

if __name__ == "__main__":
    app.run(port=PORT, debug=True)