61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
|
import json
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def save_merge_json(data_lis, file_path):
|
||
|
|
with open(file_path, 'wt', encoding='utf-8') as file:
|
||
|
|
json.dump(data_lis, file, ensure_ascii=False, separators=(',\n',':'))
|
||
|
|
|
||
|
|
|
||
|
|
def get_all_file_paths(folder_path, file_type='.jsonl'):
|
||
|
|
# 确保传入的是一个目录
|
||
|
|
if not os.path.isdir(folder_path):
|
||
|
|
raise ValueError(f"{folder_path} is not a valid directory")
|
||
|
|
|
||
|
|
# 获取文件夹下所有文件的路径
|
||
|
|
file_paths = [os.path.join(folder_path, file) for file in os.listdir(
|
||
|
|
folder_path) if os.path.isfile(os.path.join(folder_path, file)) and (file_type in file)]
|
||
|
|
return file_paths
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
conversion_lis = []
|
||
|
|
|
||
|
|
folder_path = r'./'
|
||
|
|
# D:\github_repos\EmoLLM\generate_data
|
||
|
|
|
||
|
|
merge_path = folder_path.split('/')[-1]
|
||
|
|
try:
|
||
|
|
merge_last_path = folder_path.split('/')[-2] if folder_path.split('/')[-2]!='.' else ''
|
||
|
|
except:
|
||
|
|
merge_last_path = ''
|
||
|
|
print(f'merge_path={merge_path},merge_last_path={merge_last_path}')
|
||
|
|
|
||
|
|
|
||
|
|
for path in get_all_file_paths(folder_path):
|
||
|
|
print(path)
|
||
|
|
|
||
|
|
with open(path, 'rt', encoding='utf-8') as file:
|
||
|
|
for line in file:
|
||
|
|
# # 移除行尾的换行符
|
||
|
|
# if line == '\n':
|
||
|
|
# line = line.rstrip('\n')
|
||
|
|
line = line.rstrip('\n')
|
||
|
|
# 解析JSON
|
||
|
|
try:
|
||
|
|
data = json.loads(line)
|
||
|
|
conversion_lis.append(data)
|
||
|
|
# conversion_lis.append('\n')
|
||
|
|
except json.JSONDecodeError as e:
|
||
|
|
print(f"Error decoding JSON: {e}")
|
||
|
|
|
||
|
|
if merge_last_path!='':
|
||
|
|
save_merge_json_path = rf'./{merge_last_path}/{merge_path}_merge.jsonl'
|
||
|
|
elif merge_path!='':
|
||
|
|
save_merge_json_path = rf'./{merge_path}_merge.json'
|
||
|
|
else:
|
||
|
|
save_merge_json_path = rf'./curr_merge.json'
|
||
|
|
|
||
|
|
save_merge_json(data_lis=conversion_lis,
|
||
|
|
file_path=save_merge_json_path)
|