참고소스 수정본

참고/guardrails-main/guardrails/classes/generic/serializeable.py

@@ -0,0 +1,52 @@
+import inspect
+import json
+import sys
+from dataclasses import InitVar, asdict, dataclass, field, is_dataclass
+from json import JSONEncoder
+from typing import Any, Dict
+
+from pydash.strings import snake_case
+
+
+def get_annotations(obj):
+    if sys.version_info.minor >= 10 and hasattr(inspect, "get_annotations"):
+        return inspect.get_annotations(obj)  # type: ignore
+    else:
+        return obj.__annotations__
+
+
+class SerializeableJSONEncoder(JSONEncoder):
+    def default(self, o):
+        if is_dataclass(o):
+            return asdict(o)
+        return super().default(o)
+
+
+encoder_kwargs = {}
+if sys.version_info.minor >= 10:
+    encoder_kwargs["kw_only"] = True
+    encoder_kwargs["default"] = SerializeableJSONEncoder
+
+
+@dataclass
+class Serializeable:
+    encoder: InitVar[JSONEncoder] = field(**encoder_kwargs)
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]):
+        annotations = get_annotations(cls)
+        attributes = dict.keys(annotations)
+        snake_case_kwargs = {
+            snake_case(k): data.get(k) for k in data if snake_case(k) in attributes
+        }
+        snake_case_kwargs["encoder"] = snake_case_kwargs.get(
+            "encoder", SerializeableJSONEncoder
+        )
+        return cls(**snake_case_kwargs)  # type: ignore
+
+    @property
+    def __dict__(self) -> Dict[str, Any]:
+        return asdict(self)
+
+    def to_json(self):
+        return json.dumps(self, cls=self.encoder)  # type: ignore