Update app.py

This commit is contained in:
Marcos Lohmann
2025-09-22 13:45:39 -03:00
committed by GitHub
parent 10383916a6
commit 5abab02e61

View File

@@ -62,6 +62,23 @@ if not TEST_MODE:
print("Executando em modo de teste...") print("Executando em modo de teste...")
TEST_MODE = True TEST_MODE = True
# ==========================
# Segurança API
# ==========================
def check_api_key():
expected_key = os.environ.get("API_KEY")
if not expected_key:
print("AVISO: API_KEY não configurada nas variáveis de ambiente.")
return
provided_key = request.headers.get("X-API-Key")
if provided_key != expected_key:
abort(401, description="Chave de API inválida ou ausente.")
@app.before_request
def before_all_requests():
check_api_key()
# ========================== # ==========================
# Variáveis de Bucket / Uploads # Variáveis de Bucket / Uploads
# ========================== # ==========================
@@ -86,23 +103,6 @@ if not TEST_MODE:
# Session store para mapear file_id -> object_name (para fallback de download) # Session store para mapear file_id -> object_name (para fallback de download)
FILE_INDEX: Dict[str, str] = {} FILE_INDEX: Dict[str, str] = {}
# ==========================
# Segurança API
# ==========================
def check_api_key():
expected_key = os.environ.get("API_KEY")
if not expected_key:
print("AVISO: API_KEY não configurada nas variáveis de ambiente.")
return
provided_key = request.headers.get("X-API-Key")
if provided_key != expected_key:
abort(401, description="Chave de API inválida ou ausente.")
@app.before_request
def before_all_requests():
check_api_key()
# ========================== # ==========================
# Helpers: Signed URL (PAR) + Upload # Helpers: Signed URL (PAR) + Upload
# ========================== # ==========================