35 lines
838 B
Python
35 lines
838 B
Python
|
from typing import Union, Tuple, Dict, List
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class Response(BaseModel):
|
||
|
|
code: int
|
||
|
|
message: str
|
||
|
|
data: Union[Dict, List]
|
||
|
|
|
||
|
|
|
||
|
|
def build_success_response(data: Union[Dict, List]):
|
||
|
|
return {
|
||
|
|
"code": RESPONSE_SUCCESS[0],
|
||
|
|
"message": RESPONSE_SUCCESS[1],
|
||
|
|
"data": data
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def build_failure_response(result_tuple: Tuple, data: Union[Dict, List]):
|
||
|
|
return build_failure_response(result_tuple[0], result_tuple[1], data)
|
||
|
|
|
||
|
|
|
||
|
|
def build_failure_response(code: int, message: str, data: Union[Dict, List]):
|
||
|
|
return {
|
||
|
|
"code": code,
|
||
|
|
"message": message,
|
||
|
|
"data": data
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
RESPONSE_SUCCESS = 200, "SUCCESS"
|
||
|
|
RESPONSE_UNKNOWN_ERROR = 100001, "UNKNOWN ERROR"
|
||
|
|
RESPONSE_UNAUTHORIZED = 100002, "UNAUTHORIZED"
|
||
|
|
RESPONSE_TASK_NOT_FOUND = 100003, "TASK NOT FOUND"
|