import json


class R():
    """
    响应实体
    """

    def __init__(self, code=0):
        self.d = {'code': code, 'msg': ''}

    def json(self):
        """
        返回json数据
        :return:
        """
        return json.dumps(self.d, ensure_ascii=False)   # 不转成ASCII码

    def add(self, key, value):
        """
        链式调用支持
        :param key:
        :param value:
        :return:
        """
        self.d[key] = value
        return self

    def __str__(self):
        """
        string处理
        """
        self.json()


def ok(msg='success.'):
    return R().add('msg', msg)


def error(msg='failure!'):
    return R(400).add('msg', msg)


def expire(msg='token expired.'):
    return R(401).add('msg', msg)


if __name__ == '__main__':
    print(ok('操作成功').json())