54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
from flask import Flask, jsonify, request
|
|
from flask_cors import CORS
|
|
import os
|
|
from dotenv import load_dotenv
|
|
from db import get_connection
|
|
|
|
#Carrega arquivo de variáveis de ambiente (.env)
|
|
load_dotenv()
|
|
#Carrega variável de ambiente PORT. Essa variável contém a porta TCP utilizada na APIREST
|
|
PORT = os.getenv("PORT")
|
|
|
|
#Cria objeto Flask. Esse objeto é necessário para utilização do FLASK
|
|
#(módulo python para criação de APIs REST).
|
|
app = Flask(__name__)
|
|
#Aplica cors no objeto Flask
|
|
CORS(app)
|
|
|
|
#Criar rota de saúde da aplicação
|
|
@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_produtos_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 não encontrado"}), 404
|
|
|
|
return jsonify(produto)
|
|
|
|
#Inicializa o servidor da APIRest
|
|
if __name__ == "__main__":
|
|
app.run(port=PORT, debug=True) |