result_type.py

import json


class ObdiagResult:
    # ObdiagResult is the result of obdiag.
    # It contains the code and result of obdiag.

    # SERVER_ERROR_CODE(5xx) is the code of server error.
    SERVER_ERROR_CODE = 500
    # INPUT_ERROR_CODE(4xx) is the code of input error.
    INPUT_ERROR_CODE = 400
    # SUCCESS_CODE(200) is the code of success.
    SUCCESS_CODE = 200

    def __init__(self, code, data=None, error_data=None):
        self.command = None
        self.trace_id = None
        self.data = data
        self.error_data = error_data
        if code is None:
            raise TypeError("ObdiagResult code is None. Please contact the Oceanbase community. ")
        self.code = code
        if data is not None:
            if isinstance(data, dict):
                self.data = data
            else:
                raise TypeError("ObdiagResult data is not dict. Please contact the Oceanbase community. ")
        if error_data is not None:
            if isinstance(error_data, str):
                self.error_data = error_data
            else:
                raise TypeError("ObdiagResult error_data is not str. Please contact the Oceanbase community. ")

    def set_trace_id(self, trace_id):
        self.trace_id = "{0}".format(trace_id)

    def set_command(self, command):
        self.command = command

    def get_result(self):
        result = {"code": self.code, "data": self.data, "error_data": self.error_data, "trace_id": self.trace_id, "command": self.command}
        return json.dumps(result, ensure_ascii=False)

    def get_code(self):
        return self.code