|
|
"""
|
|
|
JSON 序列化配置
|
|
|
|
|
|
Flask 3.x 使用 JSONProvider 替代 JSONEncoder
|
|
|
支持通过配置文件控制序列化行为
|
|
|
"""
|
|
|
|
|
|
from datetime import datetime, date
|
|
|
from flask.json.provider import DefaultJSONProvider
|
|
|
|
|
|
|
|
|
class CustomJSONProvider(DefaultJSONProvider):
|
|
|
"""
|
|
|
自定义 JSON Provider (Flask 3.x)
|
|
|
|
|
|
功能：
|
|
|
- 处理 datetime 和 date 类型
|
|
|
- 支持配置文件控制序列化参数
|
|
|
"""
|
|
|
|
|
|
def __init__(self, app):
|
|
|
"""
|
|
|
初始化 JSON Provider
|
|
|
|
|
|
从 app.config 读取配置：
|
|
|
- JSON_ENSURE_ASCII: 是否转义非 ASCII 字符
|
|
|
- JSON_SORT_KEYS: 是否对字典 key 排序
|
|
|
- JSON_INDENT: 缩进空格数（None 为不缩进）
|
|
|
- JSON_SEPARATORS: 分隔符元组
|
|
|
"""
|
|
|
super().__init__(app)
|
|
|
|
|
|
# 从配置读取设置
|
|
|
self._ensure_ascii = app.config.get('JSON_ENSURE_ASCII', False)
|
|
|
self._sort_keys = app.config.get('JSON_SORT_KEYS', False)
|
|
|
self._indent = app.config.get('JSON_INDENT', None)
|
|
|
self._separators = app.config.get('JSON_SEPARATORS', (',', ':'))
|
|
|
|
|
|
def default(self, obj):
|
|
|
"""
|
|
|
序列化自定义对象
|
|
|
|
|
|
Args:
|
|
|
obj: 要序列化的对象
|
|
|
|
|
|
Returns:
|
|
|
序列化后的 Python 基本类型
|
|
|
"""
|
|
|
# 处理 datetime 类型
|
|
|
if isinstance(obj, datetime):
|
|
|
return obj.strftime("%Y-%m-%d %H:%M:%S")
|
|
|
|
|
|
# 处理 date 类型
|
|
|
elif isinstance(obj, date):
|
|
|
return obj.strftime("%Y-%m-%d")
|
|
|
|
|
|
# 其他类型调用父类处理
|
|
|
return super().default(obj)
|
|
|
|
|
|
def dumps(self, obj, **kwargs):
|
|
|
"""
|
|
|
序列化为 JSON 字符串
|
|
|
|
|
|
合并配置文件和运行时参数
|
|
|
"""
|
|
|
# 使用配置文件的默认值
|
|
|
kwargs.setdefault('ensure_ascii', self._ensure_ascii)
|
|
|
kwargs.setdefault('sort_keys', self._sort_keys)
|
|
|
|
|
|
if self._indent is not None:
|
|
|
kwargs.setdefault('indent', self._indent)
|
|
|
|
|
|
kwargs.setdefault('separators', self._separators)
|
|
|
|
|
|
return super().dumps(obj, **kwargs)
|
|
|
|
|
|
|
|
|
def init_encoder(app):
|
|
|
"""
|
|
|
初始化 JSON 编码器
|
|
|
|
|
|
配置：
|
|
|
- 设置自定义 JSONProvider
|
|
|
- 从配置文件读取序列化参数
|
|
|
- 支持环境差异化配置
|
|
|
"""
|
|
|
# 设置自定义 Provider
|
|
|
app.json_provider_class = CustomJSONProvider
|
|
|
|
|
|
# 直接设置 app.json 的属性（Flask 3.x 推荐方式）
|
|
|
app.json.ensure_ascii = app.config.get('JSON_ENSURE_ASCII', False)
|
|
|
app.json.sort_keys = app.config.get('JSON_SORT_KEYS', False)
|