Mireya Cueto Garrido
28 lines
804 B
Python
28 lines
804 B
Python
import os
|
|
from dotenv import load_dotenv
|
|
from fastapi import Depends, HTTPException, status
|
|
from fastapi.security import APIKeyHeader
|
|
|
|
# Cargar variables del .env
|
|
load_dotenv()
|
|
|
|
API_KEY_NAME = os.getenv("API_KEY_NAME")
|
|
API_KEY_VALUE = os.getenv("API_KEY_VALUE")
|
|
|
|
if not API_KEY_NAME:
|
|
raise RuntimeError("ERROR: La variable API_KEY_NAME no está definida en el .env")
|
|
|
|
if not API_KEY_VALUE:
|
|
raise RuntimeError("ERROR: La variable API_KEY_VALUE no está definida en el .env")
|
|
|
|
api_key_header = APIKeyHeader(name=API_KEY_NAME, auto_error=False)
|
|
|
|
|
|
def get_api_key(api_key: str = Depends(api_key_header)):
|
|
if api_key != API_KEY_VALUE:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="API key inválida o ausente."
|
|
)
|
|
return api_key
|