python
This commit is contained in:
parent
af79066ca0
commit
bb69fea283
|
|
@ -13,7 +13,9 @@ from flask import Flask, request, jsonify, send_from_directory
|
||||||
load_dotenv()
|
load_dotenv()
|
||||||
|
|
||||||
app = Flask(__name__)
|
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
|
# Configuração do MySQL usando variáveis de ambiente
|
||||||
app.config['MYSQL_HOST'] = os.getenv('MYSQL_HOST')
|
app.config['MYSQL_HOST'] = os.getenv('MYSQL_HOST')
|
||||||
|
|
@ -126,6 +128,9 @@ def mouting():
|
||||||
# Extrair o domínio do email do usuário autenticado
|
# Extrair o domínio do email do usuário autenticado
|
||||||
domain = username_full.split('@')[1]
|
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
|
# Consultar o banco de dados para obter o nome da empresa com base no domínio
|
||||||
cur = mysql.connection.cursor()
|
cur = mysql.connection.cursor()
|
||||||
cur.execute("SELECT nome FROM empresa WHERE dominio = %s", (domain,))
|
cur.execute("SELECT nome FROM empresa WHERE dominio = %s", (domain,))
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue