合并到远程
This commit is contained in:
@@ -1,54 +1,6 @@
|
||||
import copy
|
||||
from typing import Any
|
||||
|
||||
|
||||
class BaseDataModel:
|
||||
def deepcopy(self):
|
||||
return copy.deepcopy(self)
|
||||
|
||||
|
||||
def transform_class_to_dict(obj: Any) -> Any:
|
||||
# sourcery skip: assign-if-exp, reintroduce-else
|
||||
"""
|
||||
将对象或容器中的 BaseDataModel 子类(类对象)或 BaseDataModel 实例
|
||||
递归转换为普通 dict,不修改原对象。
|
||||
- 对于类对象(isinstance(value, type) 且 issubclass(..., BaseDataModel)),
|
||||
读取类的 __dict__ 中非 dunder 项并递归转换。
|
||||
- 对于实例(isinstance(value, BaseDataModel)),读取 vars(instance) 并递归转换。
|
||||
"""
|
||||
|
||||
def _transform(value: Any) -> Any:
|
||||
# 值是类对象且为 BaseDataModel 的子类
|
||||
if isinstance(value, type) and issubclass(value, BaseDataModel):
|
||||
return {k: _transform(v) for k, v in value.__dict__.items() if not k.startswith("__") and not callable(v)}
|
||||
|
||||
# 值是 BaseDataModel 的实例
|
||||
if isinstance(value, BaseDataModel):
|
||||
return {k: _transform(v) for k, v in vars(value).items()}
|
||||
|
||||
# 常见容器类型,递归处理
|
||||
if isinstance(value, dict):
|
||||
return {k: _transform(v) for k, v in value.items()}
|
||||
if isinstance(value, list):
|
||||
return [_transform(v) for v in value]
|
||||
if isinstance(value, tuple):
|
||||
return tuple(_transform(v) for v in value)
|
||||
if isinstance(value, set):
|
||||
return {_transform(v) for v in value}
|
||||
# 基本类型,直接返回
|
||||
return value
|
||||
|
||||
result = _transform(obj)
|
||||
|
||||
def flatten(target_dict: dict):
|
||||
flat_dict = {}
|
||||
for k, v in target_dict.items():
|
||||
if isinstance(v, dict):
|
||||
# 递归扁平化子字典
|
||||
sub_flat = flatten(v)
|
||||
flat_dict.update(sub_flat)
|
||||
else:
|
||||
flat_dict[k] = v
|
||||
return flat_dict
|
||||
|
||||
return flatten(result) if isinstance(result, dict) else result
|
||||
|
||||
Reference in New Issue
Block a user