✨ pydantic config model support

ArcletProject · Feb 14, 2025 · d921f71 · d921f71
1 parent 002cc12
commit d921f71
Show file tree

Hide file tree

Showing 3 changed files with 158 additions and 3 deletions.
diff --git a/arclet/entari/builtins/pyd.py b/arclet/entari/builtins/pyd.py
@@ -0,0 +1,23 @@
+from typing import Any, TypeVar
+
+from arclet.entari.config import config_validator_register
+from arclet.entari.plugin import metadata, declare_static
+from pydantic import BaseModel, TypeAdapter
+
+declare_static()
+metadata(
+    "pydantic",
+    author=["RF-Tar-Railt <rf_tar_railt@qq.com>"],
+    description="Pydantic Config Model support for entari",
+)
+
+
+TB = TypeVar("TB", bound=BaseModel)
+
+
+@config_validator_register(BaseModel)
+def _pydantic_validate(data: dict[str, Any], base: type[TB]) -> TB:
+    return TypeAdapter(base).validate_python(data)
+
+
+__all__ = ["BaseModel"]
diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -41,6 +41,9 @@ full = [
 dotenv = [
     "python-dotenv>=1.0.1",
 ]
+pydantic = [
+    "pydantic>=2.10.6",
+]
 
 [project.scripts]
 entari = "arclet.entari.__main__:main"