147 lines
5.0 KiB
Python
147 lines
5.0 KiB
Python
|
||
import json
|
||
import re
|
||
|
||
def extract_content_from_json(string):
|
||
"""
|
||
输入字符串，提取 { 和 } 之间的内容，并将其解析为字典。
|
||
如果使用 insert_missing_commas 修复后仍然失败，则尝试返回原始的解析结果（如果可能）。
|
||
"""
|
||
if not string.strip():
|
||
return {}
|
||
|
||
# 提取第一个匹配的 JSON 对象
|
||
match = re.search(r'\{[\s\S]*\}', string)
|
||
if match:
|
||
json_data = match.group(0)
|
||
try:
|
||
# 尝试直接解析原始 JSON 数据
|
||
return json.loads(json_data)
|
||
except json.JSONDecodeError as original_error:
|
||
print(f"原始 JSON 解析失败: {original_error}")
|
||
try:
|
||
# 尝试修复缺失的逗号
|
||
fixed_json = insert_missing_commas(json_data)
|
||
return json.loads(fixed_json)
|
||
except json.JSONDecodeError as fixed_error:
|
||
print(f"修复后的 JSON 解析失败: {fixed_error}")
|
||
# 可选：返回空字典或其他默认值
|
||
return {}
|
||
else:
|
||
print("json_utils: extract_content_from_json: 未找到有效的 JSON 内容。")
|
||
return {}
|
||
|
||
def insert_missing_commas(json_str):
|
||
"""
|
||
使用正则表达式在缺失逗号的位置插入逗号。
|
||
具体来说，寻找一个值的结束引号后紧跟着下一个键的开始引号，并在中间插入逗号。
|
||
"""
|
||
# 这个正则匹配一个字符串结尾的引号，可能有空白字符，然后是另一个键的引号
|
||
pattern = r'(":\s*"[^"]*)"\s*(")'
|
||
replacement = r'\1", \2'
|
||
|
||
previous_str = None
|
||
while previous_str != json_str:
|
||
previous_str = json_str
|
||
json_str = re.sub(pattern, replacement, json_str)
|
||
|
||
return json_str
|
||
def clean_json_string(json_string):
|
||
"""清理JSON字符串，移除多余的反引号并解析为字典"""
|
||
return extract_content_from_json(json_string)
|
||
|
||
def combine_json_results(json_lists):
|
||
"""
|
||
将类json格式的列表整合成json数据（即大括号{}包裹）
|
||
"""
|
||
combined_result = {}
|
||
for json_str in json_lists:
|
||
if json_str.strip():
|
||
json_data = clean_json_string(json_str)
|
||
combined_result.update(json_data)
|
||
return combined_result
|
||
|
||
|
||
def nest_json_under_key(data, key):
|
||
"""
|
||
将给定的字典 data 嵌套在一个新的字典层级下，该层级由 key 指定，并返回 JSON 格式的字符串。
|
||
|
||
参数:
|
||
- data: dict, 要嵌套的原始字典。
|
||
- key: str, 新层级的键名。
|
||
|
||
返回:
|
||
- 嵌套后的 JSON 字符串。
|
||
"""
|
||
# 创建一个新字典，其中包含一个键，该键的值是原始字典
|
||
nested_dict = {key: data}
|
||
# 将字典转换成 JSON 字符串
|
||
nested_json = json.dumps(nested_dict, ensure_ascii=False, indent=4)
|
||
return nested_json
|
||
|
||
|
||
def add_keys_to_json(target_dict, source_dict):
|
||
"""
|
||
将 source_dict 的内容添加到 target_dict 中的唯一外层键下的字典中。
|
||
|
||
参数:
|
||
target_dict (dict): 要更新的目标字典，假定只有一个外层键。
|
||
source_dict (dict): 源字典，其内容将被添加到目标字典。
|
||
|
||
返回:
|
||
dict: 更新后的字典。
|
||
"""
|
||
if not target_dict:
|
||
print("json_utils: Error: Target dictionary is empty.")
|
||
return {}
|
||
|
||
if len(target_dict) != 1:
|
||
print("json_utils: Error: Target dictionary must contain exactly one top-level key.")
|
||
return target_dict
|
||
|
||
# 获取唯一的外层键
|
||
target_key, existing_dict = next(iter(target_dict.items()))
|
||
|
||
if not isinstance(existing_dict, dict):
|
||
print(f"json_utils: Error: The value under the key '{target_key}' is not a dictionary.")
|
||
return target_dict
|
||
|
||
# 合并字典
|
||
existing_dict.update(source_dict)
|
||
|
||
# 更新原字典
|
||
target_dict[target_key] = existing_dict
|
||
|
||
return target_dict
|
||
|
||
|
||
def rename_outer_key(original_data,new_key):
|
||
# 定义新的键名
|
||
# new_key = "重新招标, 不再招标和终止招标"
|
||
|
||
# 提取原始数据中的唯一外层值（假设只有一个外层键）
|
||
if not original_data or not isinstance(original_data, dict):
|
||
print("json_utils: Error: Invalid input or input is not a dictionary.") # 如果输入无效或不是字典，则返回空字典
|
||
return {}
|
||
|
||
# 使用 next(iter(...)) 提取第一个键的值
|
||
original_value = next(iter(original_data.values()), {})
|
||
|
||
# 创建一个新的字典，使用新的键名
|
||
new_data = {new_key: original_value}
|
||
|
||
return new_data
|
||
def transform_json_values(data):
|
||
if isinstance(data, dict):
|
||
return {key.replace(' ', ''): transform_json_values(value) for key, value in data.items()}
|
||
elif isinstance(data, list):
|
||
return [transform_json_values(item) for item in data]
|
||
elif isinstance(data, bool):
|
||
return '是' if data else '否'
|
||
elif isinstance(data, (int, float)):
|
||
return str(data)
|
||
elif isinstance(data, str):
|
||
return data.replace('\n', '<br>')
|
||
else:
|
||
return data
|