# 封装一个响应对象
from common.http_status import HTTPStatus
class ResponseDTO:
    def __init__(self, code, message, data):
        self.code = code
        self.message = message
        self.data = data
    def to_json(self):
        return {
            "code": self.code,
            "message": self.message,
            "data": self.data
        }
    @staticmethod
    def SUCCESS(data):
        return ResponseDTO(HTTPStatus.OK.code, HTTPStatus.OK.message, data)

    @staticmethod
    def SUCCESS_MS_DATA(message, data):
        return ResponseDTO(HTTPStatus.OK.code, message, data)

    @staticmethod
    def ERROR(data):
        return ResponseDTO(HTTPStatus.INTERNAL_SERVER_ERROR.code, HTTPStatus.INTERNAL_SERVER_ERROR.message, data)

    @staticmethod
    def ERROR_MS_DATA(message, data):
        return ResponseDTO(HTTPStatus.INTERNAL_SERVER_ERROR.code, message, data)