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
        )