testes/app/routes/empresas.py

26 lines
871 B
Python

import logging
from flask import Blueprint, jsonify
from flask_mysqldb import MySQL
from .auth import token_required # Importando o token_required
empresas = Blueprint('empresas', __name__) # Criando o Blueprint
mysql = MySQL() # Criando a instância do MySQL
@empresas.route('/empresas', methods=['GET'])
@token_required
def listar_empresas(data): # Agora aceita o argumento 'data'
try:
cur = mysql.connection.cursor()
cur.execute("SELECT idempresa, nome, dominio FROM empresa")
resultados = cur.fetchall()
# Ajuste aqui: Adicionando o 'id_empresa' no dicionário
empresas_list = [{"idempresa": row['idempresa'], "nome": row['nome'], "dominio": row['dominio']} for row in resultados]
return jsonify(empresas_list), 200
except Exception:
return jsonify({"erro": "Erro interno no servidor"}), 500