localstack-py-avro-schema 3.9.3__py3-none-any.whl → 3.9.4__py3-none-any.whl

{localstack_py_avro_schema-3.9.3.dist-info → localstack_py_avro_schema-3.9.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: localstack-py-avro-schema
-Version: 3.9.3
+Version: 3.9.4
 Summary: Generate Apache Avro schemas for Python types including standard library data-classes and Pydantic data models.
 Author-email: LocalStack Contributors <info@localstack.cloud>, "J.P. Morgan Chase & Co." <open_source@jpmorgan.com>
 License:                                  Apache License

{localstack_py_avro_schema-3.9.3.dist-info → localstack_py_avro_schema-3.9.4.dist-info}/RECORD

@@ -1,12 +1,12 @@
-localstack_py_avro_schema-3.9.3.dist-info/licenses/LICENSE,sha256=zE1N4JILDTkSIDtdmqdnKKxKEQh_VdqeoAV2230eNOI,10141
-localstack_py_avro_schema-3.9.3.dist-info/licenses/LICENSE_HEADER.txt,sha256=Nx3RGmYbJKjIKM8Y2yMrCPJgCw4oBt_H62PxDKlAsto,564
+localstack_py_avro_schema-3.9.4.dist-info/licenses/LICENSE,sha256=zE1N4JILDTkSIDtdmqdnKKxKEQh_VdqeoAV2230eNOI,10141
+localstack_py_avro_schema-3.9.4.dist-info/licenses/LICENSE_HEADER.txt,sha256=Nx3RGmYbJKjIKM8Y2yMrCPJgCw4oBt_H62PxDKlAsto,564
 py_avro_schema/__init__.py,sha256=bK6hTw4XJy5bAP4lgIvJR5LHtEiyUWq4GNJ5w-uSEpc,2414
 py_avro_schema/_alias.py,sha256=nCQqtn7IbpjV0ibpJf9aMX9FvwVx2EDC1iKzwjQ7CqI,3412
-py_avro_schema/_schemas.py,sha256=uUyztznCjmIescXDWEFetwAgBZ0mAIdGHpdea1EEY9k,48182
+py_avro_schema/_schemas.py,sha256=L9_c-I1ihS9BJpXe1-foDMUppxY7PyTT544hMbrS1TM,48589
 py_avro_schema/_testing.py,sha256=3aSfMbNDDeatl3H7GEUcynxK81HFiF-WCLOZ3FCCLiw,2261
 py_avro_schema/_typing.py,sha256=9kBlPA7C33zZP3nRhCvSJA_0m54uvUo0AQ4wJJmSxMk,3291
 py_avro_schema/py.typed,sha256=HnEDkaznpgfRW07Qfogy4tFLu_4dcQ5YcOsI7pmU5rQ,52
-localstack_py_avro_schema-3.9.3.dist-info/METADATA,sha256=hykDGJcbbKVESlOoQ3zQFWlSQ8KAv9JA18_y_AIhv6o,15164
-localstack_py_avro_schema-3.9.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-localstack_py_avro_schema-3.9.3.dist-info/top_level.txt,sha256=0t4oO_5rsdfZGSk8iSzK4TjdyfVz-RE1IZ2TIz6YvI0,15
-localstack_py_avro_schema-3.9.3.dist-info/RECORD,,
+localstack_py_avro_schema-3.9.4.dist-info/METADATA,sha256=2VBFpx5j-KsRbwOmtIICNMNLSgB34kwkCra0Jh_WO5U,15164
+localstack_py_avro_schema-3.9.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+localstack_py_avro_schema-3.9.4.dist-info/top_level.txt,sha256=0t4oO_5rsdfZGSk8iSzK4TjdyfVz-RE1IZ2TIz6YvI0,15
+localstack_py_avro_schema-3.9.4.dist-info/RECORD,,

py_avro_schema/_schemas.py

@@ -137,9 +137,10 @@ JSON_OPTIONS = [opt for opt in Option if opt.name and opt.name.startswith("JSON_
 _SCHEMA_CLASSES = []
 
 
-def register_schema(cls):
+def register_schema(cls: type | None = None, *, priority: int = 0):
     """
     Decorator to register a class as a known ``Schema``
+    It also accept a priority value to sort the list of schemas. Default schemas have priority 0.
 
     Schema classes are instantiated when calling ``schema``. Example use::
 
@@ -149,10 +150,15 @@ def register_schema(cls):
           def handles_type(cls, py_type: Type) -> bool:
               ...
           ...
-
     """
-    _SCHEMA_CLASSES.append(cls)
-    return cls
+
+    def _wrapper(_cls):
+        """Wrapper function to attach priority and sort the list of schemas."""
+        _cls.__py_avro_priority = priority
+        _SCHEMA_CLASSES.append(_cls)
+        return _cls
+
+    return _wrapper if not cls else _wrapper(cls)
 
 
 def schema(
@@ -188,7 +194,7 @@ def _schema_obj(py_type: Type, namespace: Optional[str] = None, options: Option
     :param options:   Schema generation options.
     """
     # Find concrete Schema subclasses defined in the current module
-    for schema_class in _SCHEMA_CLASSES:
+    for schema_class in sorted(_SCHEMA_CLASSES, key=lambda c: getattr(c, "__py_avro_priority", 0)):
         # Find the first schema class that handles py_type
         schema_obj = schema_class(py_type, namespace=namespace, options=options)  # type: ignore
         if schema_obj: