61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
import re
|
||
import json
|
||
from typing import Optional, Union, Any
|
||
from loguru import logger
|
||
|
||
|
||
def clean_json_markdown(text: str) -> str:
|
||
"""
|
||
清理文本中的markdown格式，提取纯JSON内容
|
||
|
||
Args:
|
||
text: 包含markdown格式的文本
|
||
|
||
Returns:
|
||
清理后的JSON字符串
|
||
"""
|
||
if not text:
|
||
return text
|
||
|
||
# 移除markdown代码块格式 ```json ... ``` 或 ``` ... ```
|
||
# 支持多种情况：```json、```JSON、```
|
||
pattern = r'```(?:json|JSON)?\s*(.*?)\s*```'
|
||
match = re.search(pattern, text, re.DOTALL | re.IGNORECASE)
|
||
|
||
if match:
|
||
# 如果找到markdown格式，提取其中的内容
|
||
cleaned_text = match.group(1).strip()
|
||
else:
|
||
# 如果没有找到markdown格式，直接使用原文本
|
||
cleaned_text = text.strip()
|
||
|
||
return cleaned_text
|
||
|
||
|
||
def safe_json_loads(text: str) -> Optional[Union[dict, list, Any]]:
|
||
"""
|
||
安全地解析JSON字符串，自动清理markdown格式
|
||
|
||
Args:
|
||
text: 包含JSON的文本字符串
|
||
|
||
Returns:
|
||
解析后的Python对象，失败返回None
|
||
"""
|
||
if not text:
|
||
return None
|
||
|
||
try:
|
||
# 先清理markdown格式
|
||
cleaned_text = clean_json_markdown(text)
|
||
|
||
# 尝试解析JSON
|
||
return json.loads(cleaned_text)
|
||
|
||
except json.JSONDecodeError as e:
|
||
logger.error(f"JSON解析失败: {e}, 原文本: {text[:200]}...")
|
||
return None
|
||
except Exception as e:
|
||
logger.error(f"JSON处理异常: {e}")
|
||
return None
|