Pydantic V2 support

datauri/__init__.py

@@ -160,7 +160,7 @@ def __get_validators__(cls):
         yield cls.validate
 
     @classmethod
-    def validate(cls, v: str) -> Self:
+    def validate(cls, v: str, x: Any = None) -> Self:
         if not isinstance(v, str):
             raise TypeError("string required")
 
@@ -169,6 +169,16 @@ def validate(cls, v: str) -> Self:
             raise ValueError("invalid data-uri format")
         return m
 
+    @classmethod
+    def __get_pydantic_json_schema__(cls, core_schema: Any, handler: Any) -> Any:
+        core_schema.update(
+            pattern=DATA_URI_REGEX,
+            examples=[
+                "data:text/plain;charset=utf-8;base64,VGhlIHF1aWNrIGJyb3duIGZveCBqdW1wZWQgb3ZlciB0aGUgbGF6eSBkb2cu"
+            ],
+        )
+        return core_schema
+
     @classmethod
     def __modify_schema__(cls, field_schema: Dict[str, Any]) -> None:
         # __modify_schema__ should mutate the dict it receives in place,

tests/test_pydantic.py

@@ -4,6 +4,13 @@
 
 pydantic = pytest.importorskip("pydantic")
 
+try:
+    pydantic.version.version_info()
+    func_json = "model_dump_json"
+except:
+    pydantic.utils.version_info()
+    func_json = "json"
+
 
 def test_pydantic():
     class Model(pydantic.BaseModel):
@@ -13,8 +20,7 @@ class Model(pydantic.BaseModel):
     instance = Model(content=t)
     assert isinstance(instance.content, DataURI)
     assert (
-        instance.json()
-        == '{"content": "data:text/plain;charset=utf-8;base64,VGhlIHF1aWNrIGJyb3duIGZveCBqdW1wZWQgb3ZlciB0'
-        'aGUgbGF6eSBkb2cu"}'
+        instance.__getattr__(func_json)()
+        == '{"content":"data:text/plain;charset=utf-8;base64,VGhlIHF1aWNrIGJyb3duIGZveCBqdW1wZWQgb3ZlciB0aGUgbGF6eSBkb2cu"}'
     )
     assert instance.dict() == {"content": DataURI(t)}