42 lines
963 B
Python
42 lines
963 B
Python
from pydantic import BaseModel
|
|
from typing import Any, Optional
|
|
|
|
|
|
class BaseWmsApiResponse(BaseModel):
|
|
code: int
|
|
message: str
|
|
|
|
@staticmethod
|
|
def success(message: str = "操作成功"):
|
|
return BaseWmsApiResponse(
|
|
code=200,
|
|
message=message
|
|
)
|
|
|
|
@staticmethod
|
|
def error(message: str = "操作失败", code: int = 500):
|
|
return BaseWmsApiResponse(
|
|
code=code,
|
|
message=message
|
|
)
|
|
|
|
|
|
class WmsApiResponse(BaseWmsApiResponse):
|
|
data: Optional[Any] = None
|
|
|
|
@staticmethod
|
|
def success(data: Any = None, message: str = "操作成功"):
|
|
return WmsApiResponse(
|
|
code=200,
|
|
message=message,
|
|
data=data
|
|
)
|
|
|
|
@staticmethod
|
|
def error(message: str = "操作失败", code: int = 500, data: Any = None):
|
|
return WmsApiResponse(
|
|
code=code,
|
|
message=message,
|
|
data=data
|
|
)
|