pydantic-avro 0.9.2__py3-none-any.whl → 0.9.3__py3-none-any.whl

pydantic_avro/to_avro/base.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Literal, Optional
 
 from pydantic import BaseModel
 
@@ -10,14 +10,30 @@ class AvroBase(BaseModel):
     """This class provides functionality to convert a pydantic model to an Avro schema."""
 
     @classmethod
-    def avro_schema(cls, by_alias: bool = True, namespace: Optional[str] = None) -> dict:
+    def avro_schema(
+        cls,
+        by_alias: bool = True,
+        namespace: Optional[str] = None,
+        mode: Literal["validation", "serialization"] = "serialization",
+    ) -> dict:
         """Returns the avro schema for the pydantic class
 
         :param by_alias: generate the schemas using the aliases defined, if any
         :param namespace: Provide an optional namespace string to use in schema generation
+        :param mode: The mode for generating the schema. Use 'validation' for input validation
+                     or 'serialization' for output serialization. Defaults to 'serialization'.
+                     Only applicable for Pydantic v2.
         :return: dict with the Avro Schema for the model
         """
-        schema = cls.model_json_schema(by_alias=by_alias) if PYDANTIC_V2 else cls.schema(by_alias=by_alias)
+        if PYDANTIC_V2:
+            schema = cls.model_json_schema(by_alias=by_alias, mode=mode)
+        else:
+            if mode != "serialization":
+                raise ValueError(
+                    f"The 'mode' parameter is only supported in Pydantic v2. "
+                    f"Pydantic v1 does not support different schema modes."
+                )
+            schema = cls.schema(by_alias=by_alias)
 
         if namespace is None:
             # Default namespace will be based on title

{pydantic_avro-0.9.2.dist-info → pydantic_avro-0.9.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pydantic-avro
-Version: 0.9.2
+Version: 0.9.3
 Summary: Converting pydantic classes to avro schemas
 Home-page: https://github.com/godatadriven/pydantic-avro
 License: MIT

{pydantic_avro-0.9.2.dist-info → pydantic_avro-0.9.3.dist-info}/RECORD

@@ -7,11 +7,11 @@ pydantic_avro/from_avro/class_registery.py,sha256=n_8yELp-Eux9bAs4CJOV78bQ061VV_
 pydantic_avro/from_avro/types.py,sha256=--aWUMKtR5DUaYXXJ1RrDonLZ7AADcdGGAARopLZexA,5190
 pydantic_avro/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pydantic_avro/to_avro/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pydantic_avro/to_avro/base.py,sha256=Y8QOGTpXO3BwJ6arrRK-k2JniK5kLTvlDpQ0PyZojmg,1409
+pydantic_avro/to_avro/base.py,sha256=QtCgXnqhcQAxgWVihz5idwoY4slmkJ2WDKcGqLld9Ig,2066
 pydantic_avro/to_avro/config.py,sha256=R9HLkWiXs7YZ02te8CQt5Kf2rbsE8Fx-hjqv3tBSFJU,152
 pydantic_avro/to_avro/types.py,sha256=-qemQC_vs-xy7VJDTQ1y1k7K1J6Qv3pTLILqp17MO2A,12731
-pydantic_avro-0.9.2.dist-info/entry_points.txt,sha256=gwHiQfbGLO8Np2sa1bZ_bpxU7sEufx6IachViBE_Fnw,66
-pydantic_avro-0.9.2.dist-info/LICENSE,sha256=gBlYCG1yxb0vGlsmek0lMPVOK5YDxQope4F54jzeqoY,1069
-pydantic_avro-0.9.2.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
-pydantic_avro-0.9.2.dist-info/METADATA,sha256=cSYqoXQORSErCOuZMd_T30e-bHrbRRN0bgHMLM5TXq8,2945
-pydantic_avro-0.9.2.dist-info/RECORD,,
+pydantic_avro-0.9.3.dist-info/entry_points.txt,sha256=gwHiQfbGLO8Np2sa1bZ_bpxU7sEufx6IachViBE_Fnw,66
+pydantic_avro-0.9.3.dist-info/LICENSE,sha256=gBlYCG1yxb0vGlsmek0lMPVOK5YDxQope4F54jzeqoY,1069
+pydantic_avro-0.9.3.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
+pydantic_avro-0.9.3.dist-info/METADATA,sha256=r_czdRovj5dhYm8sfsLTm16EY7HD5LAGEsOSMJnUZ7g,2945
+pydantic_avro-0.9.3.dist-info/RECORD,,