File size: 781 Bytes
b6c2bbd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import http
from typing import Optional
from fastapi import HTTPException, status
class HfApiException(Exception):
def __init__(
self,
status_code: int,
detail: Optional[str] = None,
) -> None:
if detail is None:
self.detail = http.HTTPStatus(status_code).phrase
else:
self.detail = detail
self.status_code = status_code
def __repr__(self) -> str:
class_name = self.__class__.__name__
return f"{class_name}(status_code={self.status_code!r}, detail={self.detail!r})"
def __str__(self) -> str:
return self.__repr__()
INVALID_API_KEY_ERROR = HfApiException(
status_code=status.HTTP_403_FORBIDDEN,
detail="Invalid API Key",
)
|