json_utils.py 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. import orjson
  2. from loguru import logger
  3. import ast
  4. def parse_general_json(response_text, key=None):
  5. """
  6. 通用的 JSON 解析方法,兼容单引号和双引号的 JSON 字符串。
  7. 优先尝试标准 JSON 解析,失败后尝试处理单引号情况。
  8. :param response_text: 待解析的 JSON 字符串
  9. :param key: 若指定,则从解析结果中提取该键对应的值,默认为 None
  10. :return: 解析结果,如果指定了 key 则返回对应的值,解析失败返回 None
  11. """
  12. try:
  13. # 优先尝试使用 orjson.loads 解析标准 JSON 格式
  14. data = orjson.loads(response_text.strip())
  15. except orjson.JSONDecodeError:
  16. try:
  17. # 若 orjson.loads 解析失败,尝试使用 ast.literal_eval 处理单引号情况
  18. data = ast.literal_eval(response_text.strip())
  19. except (SyntaxError, ValueError) as e:
  20. # 记录解析错误信息
  21. logger.error(f"JSON 解析错误, 响应内容: {response_text}, 错误信息: {e}")
  22. return None
  23. if key is not None:
  24. try:
  25. return data[key]
  26. except KeyError:
  27. logger.error(f"键 {key} 不存在于解析结果中,响应内容: {response_text}")
  28. return None
  29. return data
  30. if __name__ == '__main__':
  31. # 测试正常解析情况 - 标准 JSON(双引号)
  32. standard_json = '{"text": "这是标准 JSON 测试内容"}'
  33. result_standard = parse_general_json(standard_json, key='text')
  34. print("标准 JSON 解析结果:", result_standard)
  35. # 测试正常解析情况 - 非标准 JSON(单引号)
  36. non_standard_json = "{'text': '这是非标准 JSON 测试内容'}"
  37. result_non_standard = parse_general_json(non_standard_json, key='text')
  38. print("非标准 JSON 解析结果:", result_non_standard)