# -*- coding: utf-8 -*-
"""
Created on Mon Jul  5 10:50:01 2021

@author: haijiao

用于拼接指定目录下所有的json文件
path后填写json文件目录
path_merges后填写拼接好的json文件存放目录
"""

import os
import json
import time

path = './'
path_merges = './results_merges'
file_list = os.listdir(path)
json_list = []
data_list = []
out_time = time.strftime("%m_%d_%H_%M")

for file in file_list:
    if file.endswith(".json"):
        json_list.append(file)
print(json_list)
print(f"json文件数量：{len(json_list)}")

i = 1
for json_file in json_list:
    json_path = os.path.join(path, json_file)
    with open(json_path, 'r') as f:
        locals()['json_data' + str(i)]= f.read()
        locals()['data_' + str(i)] = json.loads(locals()['json_data' + str(i)].strip())
        data_list = data_list + list(locals()['data_' + str(i)].items())
        i += 1

data = dict(data_list)
json_data = json.dumps(data, ensure_ascii=False)

if not os.path.exists(path_merges):
    os.mkdir(path_merges)
with open(os.path.join(path_merges, f'merges_{out_time}.json'),"w") as f1:
    f1.write(json_data)