126 lines
4.0 KiB
Python
126 lines
4.0 KiB
Python
from typing import Optional
|
|
from fastapi import status
|
|
from fastapi.responses import JSONResponse
|
|
|
|
|
|
class BaseEndpointResponse:
|
|
|
|
def __init__(self, lang: str, code: str):
|
|
self.lang = lang
|
|
self.code = code
|
|
|
|
@property
|
|
def response(self) -> Optional[dict]:
|
|
from Services.Redis import RedisActions
|
|
from ApiLayers.AllConfigs.Redis.configs import RedisValidationKeys
|
|
|
|
language_model_key = f"{RedisValidationKeys.LANGUAGE_MODELS}:{RedisValidationKeys.RESPONSES}"
|
|
language_model = RedisActions.get_json(list_keys=[language_model_key, self.code , self.lang])
|
|
if language_model.status:
|
|
return language_model.first.as_dict
|
|
raise ValueError("Language model not found")
|
|
|
|
|
|
class EndpointSuccessResponse(BaseEndpointResponse): # 1. 200 OK
|
|
|
|
def as_dict(self, data: Optional[dict] = None):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_200_OK,
|
|
content=dict(completed=True, lang=self.lang, data=data, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointCreatedResponse(BaseEndpointResponse): # 2. 201 Created
|
|
|
|
def as_dict(self, data: Optional[dict] = None):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_201_CREATED,
|
|
content=dict(completed=True, lang=self.lang, data=data, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointAcceptedResponse(BaseEndpointResponse): # 3. 202 Accepted
|
|
|
|
def as_dict(self, data: Optional[dict] = None):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_202_ACCEPTED,
|
|
content=dict(completed=True, lang=self.lang, data=data, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointBadRequestResponse(BaseEndpointResponse): # 4. 400 Bad Request
|
|
|
|
def as_dict(self, data: Optional[dict] = None):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
content=dict(completed=False, lang=self.lang, data=data, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointUnauthorizedResponse(BaseEndpointResponse): # 5. 401 Unauthorized
|
|
|
|
def as_dict(self):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
content=dict(completed=False, lang=self.lang, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointNotFoundResponse(BaseEndpointResponse): # 6. 404 Not Found
|
|
|
|
def as_dict(self):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
content=dict(completed=False, lang=self.lang, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointForbiddenResponse(BaseEndpointResponse): # 3. 403 Forbidden
|
|
|
|
def as_dict(self):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
content=dict(completed=False, lang=self.lang, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointConflictResponse(BaseEndpointResponse): # 6. 409 Conflict
|
|
|
|
def as_dict(self):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_409_CONFLICT,
|
|
content=dict(completed=False, lang=self.lang, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointTooManyRequestsResponse(BaseEndpointResponse): # 7. 429 Too Many Requests
|
|
|
|
def __init__(self, retry_after: int, lang: str, code: str):
|
|
super().__init__(lang=lang, code=code)
|
|
self.retry_after = retry_after
|
|
|
|
def as_dict(self):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_429_TOO_MANY_REQUESTS,
|
|
headers={"Retry-After": str(self.retry_after)},
|
|
content=dict(completed=False, lang=self.lang, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointInternalErrorResponse(BaseEndpointResponse): # 7. 500 Internal Server Error
|
|
|
|
def as_dict(self):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
content=dict(completed=False, lang=self.lang, **self.response)
|
|
)
|
|
|
|
|
|
class EndpointErrorResponse(BaseEndpointResponse):
|
|
|
|
def as_dict(self):
|
|
return JSONResponse(
|
|
status_code=status.HTTP_304_NOT_MODIFIED,
|
|
content=dict(completed=False, lang=self.lang, **self.response)
|
|
)
|