new registry system backwards to yaml

unilabos/utils/tools.py

@@ -1,4 +1,39 @@
+import json
+
+from unilabos.utils.type_check import TypeEncoder, json_default
+
+try:
+    import orjson
+
+    def fast_dumps(obj, **kwargs) -> bytes:
+        """JSON 序列化为 bytes，优先使用 orjson。"""
+        return orjson.dumps(obj, option=orjson.OPT_NON_STR_KEYS, default=json_default)
+
+    def fast_dumps_pretty(obj, **kwargs) -> bytes:
+        """JSON 序列化为 bytes（带缩进），优先使用 orjson。"""
+        return orjson.dumps(
+            obj,
+            option=orjson.OPT_NON_STR_KEYS | orjson.OPT_INDENT_2,
+            default=json_default,
+        )
+
+    def normalize_json(info: dict) -> dict:
+        """经 JSON 序列化/反序列化一轮来清理非标准类型。"""
+        return orjson.loads(orjson.dumps(info, default=json_default))
+
+except ImportError:
+
+    def fast_dumps(obj, **kwargs) -> bytes:  # type: ignore[misc]
+        return json.dumps(obj, ensure_ascii=False, cls=TypeEncoder).encode("utf-8")
+
+    def fast_dumps_pretty(obj, **kwargs) -> bytes:  # type: ignore[misc]
+        return json.dumps(obj, indent=2, ensure_ascii=False, cls=TypeEncoder).encode("utf-8")
+
+    def normalize_json(info: dict) -> dict:  # type: ignore[misc]
+        return json.loads(json.dumps(info, ensure_ascii=False, cls=TypeEncoder))
+
+
 # 辅助函数：将UUID数组转换为字符串
 def uuid_to_str(uuid_array) -> str:
     """将UUID字节数组转换为十六进制字符串"""
-    return "".join(format(byte, "02x") for byte in uuid_array)
+    return "".join(format(byte, "02x") for byte in uuid_array)