Update app.py
This commit is contained in:
@@ -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
|
||||||
# ==========================
|
# ==========================
|
||||||
|
|||||||
Reference in New Issue
Block a user