量化交易之python篇 - 读取、解析、存入json文件（新增异常处理功能）

import json
import math


class PositionJsonOperator:
    """
     操作 json文件 的 类
     ps: 使用时, 需要导入 json、math库
    """


    """
        ------------------ public part ------------------
    """
    @classmethod
    def tqz_load_jsonfile(cls, filename=None):
        if filename is None:
            exception = Exception("Error: filename is None")
            raise exception
        else:
            return cls._writeReadFile_except_operation(filename=filename, operationType="r")

    @classmethod
    def tqz_write_jsonfile(cls, content=None, filename=None):
        if filename is None:
            exception = Exception("Error: filename is None")
            raise exception
        else:
            cls._writeReadFile_except_operation(filename=filename, content=content, operationType="w")

    @classmethod
    def tqz_sum_position_all_jsonfile(cls, *jsonfile_list, target_jsonfile):
        """
        加总多个 json文件的 持仓, 并写入新的目json标文件中
        :param jsonfile_list: 字符串数组
        :param target_jsonfile: 要存入的 json文件名
        """
        jsonfile_content_list = []
        [jsonfile_content_list.append(cls.tqz_load_jsonfile(jsonfile)) for jsonfile in jsonfile_list]

        content_dic = cls._sum_position_all_keyValueNotFound_except_operation(dic_list=jsonfile_content_list)

        cls.tqz_write_jsonfile(content=content_dic, filename=target_jsonfile)

    @classmethod
    def tqz_multi_position_all(cls, *jsonfile_list, multi):
        """
        按倍数调整 多个json文件的 持仓
        :param jsonfile_list: 需要调整持仓的 json文件数组
        :param multi: 倍数
        """
        [cls._multi_position(source_jsonfile=jsonfile, multi=multi) for jsonfile in jsonfile_list]

    @classmethod
    def tqz_empty_position_all(cls, *jsonfile_list):
        """
        清空 多个json文件的 持仓
        :param jsonfile_list: 需要清空的 json文件数组
        """
        cls.tqz_multi_position_all(*jsonfile_list, multi=0)


    """
        ------------------ private part ------------------
    """
    @staticmethod
    def _get_newData_with_sumPosition(first_dic=None, second_dic=None):
        new_dic = {}
        for key, value in first_dic.items():
            value['pos'] = first_dic[key]['pos'] + second_dic[key]['pos']
            new_dic[key] = value
        return new_dic

    @classmethod
    def _sum_position(cls, source_filename1, source_filename2, target_filename):
        # 读取json文件数据
        cta_hla_dic = cls.tqz_load_jsonfile(filename=source_filename1)
        cta_hsr_dic = cls.tqz_load_jsonfile(filename=source_filename2)

        # 获取新的要写入json文件的数据
        new_dic = cls._get_newData_with_sumPosition(cta_hla_dic, cta_hsr_dic)

        # 写入目标文件夹
        cls.tqz_write_jsonfile(content=new_dic, filename=target_filename)

    @classmethod
    def _multi_position(cls, source_jsonfile, multi):
        source_content = cls.tqz_load_jsonfile(filename=source_jsonfile)

        source_content = cls._multi_position_keyValueNotFount_except_operation(source_content=source_content, multi=multi)

        cls.tqz_write_jsonfile(content=source_content, filename=source_jsonfile)

    @classmethod
    def _empty_position(cls, source_jsonfile):
        cls._multi_position(source_jsonfile=source_jsonfile, multi=0)


    """
        ------------------ except operation part ------------------
    """
    @classmethod
    def _writeReadFile_except_operation(cls, filename=None, content=None, operationType="r"):
        try:
            if "r" == operationType:
                with open(filename, operationType, encoding="utf-8") as fp:
                    return json.load(fp=fp)
            elif "w" == operationType:
                with open(filename, operationType, encoding="utf-8") as fp:
                    json.dump(content, fp=fp, ensure_ascii=False, indent=4)  # 参数indent: json文件按格式写入, 距行首空4格;
        except FileNotFoundError:  # filename文件 没找到
            print("Error: file %s is not fount" % filename)
        except Exception as result:
            print("unkown Error: %s" % result)

    @classmethod
    def _sum_position_all_keyValueNotFound_except_operation(cls, dic_list=[], new_dic={}):
        try:
            for dic in dic_list:
                for key, value in dic.items():
                    if key not in new_dic:
                        new_dic[key] = dic[key]
                    else:
                        value['pos'] = value['pos'] + new_dic[key]['pos']
                        new_dic[key] = value
        except AttributeError as result:
            print("Error: %s" % result)
        except Exception as result:
            print("unknow Error: %s" % result)
        else:
            return new_dic

    @classmethod
    def _multi_position_keyValueNotFount_except_operation(cls, source_content=None, multi=1):
        try:
            for key, value in source_content.items():
                if value['pos'] > 0:
                    value['pos'] = math.floor(value['pos'] * multi)
                else:
                    value['pos'] = math.floor(-1 * value['pos'] * multi) * -1
                source_content[key] = value
        except AttributeError as result:
            print("Error: %s" % result)
        except Exception as result:
            print("unknow Error: %s" % result)
        else:
            return source_content


def _main_engine():
    list = []
    list.append("symbol_2.json")
    list.append("symbol_3.json")
    list.append("symbol_4.json")
    list.append("symbol_5.json")

    target_jsonfile = "test.json"

    PositionJsonOperator.tqz_sum_position_all_jsonfile(*list, target_jsonfile=target_jsonfile)

    # PositionJsonOperator.tqz_multi_position_all(target_jsonfile, multi=0.5)
    # PositionJsonOperator.tqz_empty_position_all(target_jsonfile)


if __name__ == '__main__':
    _main_engine()