python

python/server.py

@@ -13,7 +13,9 @@ from flask import Flask, request, jsonify, send_from_directory
 load_dotenv()
 
 app = Flask(__name__)
-CORS(app)  # Habilita CORS para todas as rotas
+
+# Permitir apenas uma origem específica
+CORS(app, resources={r"/*": {"origins": "https://dev.itguys.com.br"}})
 
 # Configuração do MySQL usando variáveis de ambiente
 app.config['MYSQL_HOST'] = os.getenv('MYSQL_HOST')
@@ -126,6 +128,9 @@ def mouting():
     # Extrair o domínio do email do usuário autenticado
     domain = username_full.split('@')[1]
 
+    if not username_full:
+        return jsonify({'msg': 'Token inválido!'}), 401
+
     # Consultar o banco de dados para obter o nome da empresa com base no domínio
     cur = mysql.connection.cursor()
     cur.execute("SELECT nome FROM empresa WHERE dominio = %s", (domain,))