From e4dd3ab6c12fe861eaab2394556b15816f51785f Mon Sep 17 00:00:00 2001
From: Matt Pryor <matt@stackhpc.com>
Date: Mon, 6 Nov 2023 15:50:21 +0000
Subject: [PATCH] Changes to support Pydantic v2

---
 pyhelm3/models.py | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/pyhelm3/models.py b/pyhelm3/models.py
index 93321e1..ef915e8 100644
--- a/pyhelm3/models.py
+++ b/pyhelm3/models.py
@@ -7,15 +7,17 @@
 
 from pydantic import (
     BaseModel,
+    TypeAdapter,
     Field,
     PrivateAttr,
     DirectoryPath,
     FilePath,
-    AnyUrl,
-    HttpUrl,
+    AnyUrl as PydanticAnyUrl,
+    HttpUrl as PydanticHttpUrl,
     constr,
-    validator
+    field_validator
 )
+from pydantic.functional_validators import AfterValidator
 
 from .command import Command, SafeLoader
 
@@ -37,11 +39,11 @@ def __init__(self, _command: Command, **kwargs):
 
 
 #: Type for a name (chart or release)
-Name = constr(regex = r"^[a-z0-9-]+$")
+Name = constr(pattern = r"^[a-z0-9-]+$")
 
 
 #: Type for a SemVer version
-SemVerVersion = constr(regex = r"^v?\d+\.\d+\.\d+(-[a-zA-Z0-9\.\-]+)?(\+[a-zA-Z0-9\.\-]+)?$")
+SemVerVersion = constr(pattern = r"^v?\d+\.\d+\.\d+(-[a-zA-Z0-9\.\-]+)?(\+[a-zA-Z0-9\.\-]+)?$")
 
 
 #: Type variables for forward references to the chart and release types
@@ -50,6 +52,17 @@ def __init__(self, _command: Command, **kwargs):
 ReleaseRevisionType = t.TypeVar("ReleaseRevisionType", bound = "ReleaseRevision")
 
 
+#: Type annotation for validating a string using a Pydantic type
+def validate_str_as(validate_type):
+    adapter = TypeAdapter(validate_type)
+    return lambda v: str(adapter.validate_python(v))
+
+
+#: Annotated string types for URLs
+AnyUrl = t.Annotated[str, AfterValidator(validate_str_as(PydanticAnyUrl))]
+HttpUrl = t.Annotated[str, AfterValidator(validate_str_as(PydanticHttpUrl))]
+
+
 class ChartDependency(BaseModel):
     """
     Model for a chart dependency.
@@ -200,7 +213,7 @@ class Chart(ModelWithCommand):
     _crds: t.List[t.Dict[str, t.Any]] = PrivateAttr(None)
     _values: t.Dict[str, t.Any] = PrivateAttr(None)
 
-    @validator("ref")
+    @field_validator("ref")
     def ref_is_abspath(cls, v):
         """
         If the ref is a path on the filesystem, make sure it is absolute.