Spaces:
Build error
Build error
| import hmac | |
| import os | |
| from typing import Union | |
| from fastapi import Depends, HTTPException, status | |
| from fastapi.security import APIKeyQuery | |
| api_key = APIKeyQuery(name="token", auto_error=False) | |
| client_token: str = os.getenv("CLIENT_TOKEN", "") | |
| async def validate_token( | |
| token: Union[str, None] = Depends(api_key), | |
| ): | |
| if token is None: | |
| raise HTTPException(status.HTTP_401_UNAUTHORIZED, "No token provided") | |
| if not hmac.compare_digest(token, client_token): | |
| raise HTTPException(status.HTTP_401_UNAUTHORIZED, "Invalid token") | |
| return token | |