fix: pydantic 2.10.x breaking change

.github/workflows/github-actions-tests.yml

@@ -1,5 +1,7 @@
 name: Tests
-on: [ pull_request ]
+on:
+  pull_request:
+  push:
 
 jobs:
   pre-commit:

beanie/odm/fields.py

@@ -38,6 +38,7 @@
 from beanie.odm.utils.parsing import parse_obj
 from beanie.odm.utils.pydantic import (
     IS_PYDANTIC_V2,
+    IS_PYDANTIC_V2_10,
     get_field_type,
     get_model_fields,
     parse_object_as,
@@ -147,9 +148,8 @@ def _validate(cls, v):
         def __get_pydantic_core_schema__(
             cls, source_type: Type[Any], handler: GetCoreSchemaHandler
         ) -> CoreSchema:
-            return json_or_python_schema(
-                python_schema=no_info_plain_validator_function(cls._validate),
-                json_schema=no_info_plain_validator_function(
+            if not IS_PYDANTIC_V2_10:
+                return no_info_plain_validator_function(
                     cls._validate,
                     metadata={
                         "pydantic_js_input_core_schema": str_schema(
@@ -158,6 +158,18 @@ def __get_pydantic_core_schema__(
                             max_length=24,
                         )
                     },
+                    serialization=plain_serializer_function_ser_schema(
+                        lambda instance: str(instance),
+                        return_schema=str_schema(),
+                        when_used="json",
+                    ),
+                )
+            return no_info_plain_validator_function(
+                cls._validate,
+                json_schema_input_schema=str_schema(
+                    pattern="^[0-9a-f]{24}$",
+                    min_length=24,
+                    max_length=24,
                 ),
                 serialization=plain_serializer_function_ser_schema(
                     lambda instance: str(instance),

beanie/odm/utils/encoder.py

@@ -24,7 +24,11 @@
 
 import beanie
 from beanie.odm.fields import Link, LinkTypes
-from beanie.odm.utils.pydantic import IS_PYDANTIC_V2, get_model_fields
+from beanie.odm.utils.pydantic import (
+    IS_PYDANTIC_V2,
+    IS_PYDANTIC_V2_10,
+    get_model_fields,
+)
 
 SingleArgCallable = Callable[[Any], Any]
 DEFAULT_CUSTOM_ENCODERS: MutableMapping[type, SingleArgCallable] = {
@@ -51,6 +55,11 @@
 
     DEFAULT_CUSTOM_ENCODERS[Url] = str
 
+if IS_PYDANTIC_V2_10:
+    from pydantic import AnyUrl
+
+    DEFAULT_CUSTOM_ENCODERS[AnyUrl] = str
+
 BSON_SCALAR_TYPES = (
     type(None),
     str,

beanie/odm/utils/pydantic.py

@@ -1,9 +1,22 @@
 from typing import Any, Type
 
 import pydantic
+from packaging.specifiers import SpecifierSet
+from packaging.version import Version
 from pydantic import BaseModel
 
-IS_PYDANTIC_V2 = int(pydantic.VERSION.split(".")[0]) >= 2
+
+def is_version_valid(version, requirement):
+    # Parse the requirement as a SpecifierSet
+    specifiers = SpecifierSet(requirement)
+    # Parse the version as a Version
+    v = Version(version)
+    # Check if the version satisfies the specifiers
+    return v in specifiers
+
+
+IS_PYDANTIC_V2 = is_version_valid(pydantic.VERSION, ">=2")
+IS_PYDANTIC_V2_10 = is_version_valid(pydantic.VERSION, ">=2.10")
 
 if IS_PYDANTIC_V2:
     from pydantic import TypeAdapter