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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: localstack-py-avro-schema
-Version: 3.9.4
+Version: 3.9.6
 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.4.dist-info → localstack_py_avro_schema-3.9.6.dist-info}/RECORD

@@ -1,12 +1,12 @@
-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
+localstack_py_avro_schema-3.9.6.dist-info/licenses/LICENSE,sha256=zE1N4JILDTkSIDtdmqdnKKxKEQh_VdqeoAV2230eNOI,10141
+localstack_py_avro_schema-3.9.6.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=L9_c-I1ihS9BJpXe1-foDMUppxY7PyTT544hMbrS1TM,48589
+py_avro_schema/_schemas.py,sha256=OxSp5EFusu__X1PMA4O02nPN9c6LiAKwr-eavy5A1KY,49819
 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.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,,
+localstack_py_avro_schema-3.9.6.dist-info/METADATA,sha256=ee65BI7ff5JYA9XpIItNjNbOGSDDTtQlQa31WCbe3es,15164
+localstack_py_avro_schema-3.9.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+localstack_py_avro_schema-3.9.6.dist-info/top_level.txt,sha256=0t4oO_5rsdfZGSk8iSzK4TjdyfVz-RE1IZ2TIz6YvI0,15
+localstack_py_avro_schema-3.9.6.dist-info/RECORD,,

py_avro_schema/_schemas.py

@@ -671,6 +671,33 @@ class SequenceSchema(Schema):
         return [self.items_schema.make_default(item) for item in py_default]
 
 
+@register_schema
+class SetSchema(SequenceSchema):
+    """An Avro array schema for a given Python set"""
+
+    @classmethod
+    def handles_type(cls, py_type: type) -> bool:
+        """Whether this schema class can represent a given Python class"""
+        py_type = _type_from_annotated(py_type)
+        origin = get_origin(py_type)
+        return _is_class(origin, collections.abc.MutableSet)
+
+    def __init__(
+        self,
+        py_type: type[collections.abc.MutableSet],
+        namespace: str | None = None,
+        options: Option = Option(0),
+    ):
+        """
+        An Avro array schema for a given Python sequence
+
+        :param py_type:   The Python class to generate a schema for.
+        :param namespace: The Avro namespace to add to schemas.
+        :param options:   Schema generation options.
+        """
+        super().__init__(py_type, namespace=namespace, options=options)  # type: ignore
+
+
 @register_schema
 class DictSchema(Schema):
     """An Avro map schema for a given Python mapping"""
@@ -860,6 +887,13 @@ class EnumSchema(NamedSchema):
             return False
         return True
 
+    def data(self, names: NamesType) -> JSONType:
+        """Return the schema data"""
+        # For invalid enums we don't want to deduplicate the data
+        if not self._is_valid_enum():
+            return self.data_before_deduplication(names=names)
+        return super().data(names)
+
     def data_before_deduplication(self, names: NamesType) -> JSONObj:
         """Return the schema data"""
         if not self._is_valid_enum():
@@ -1141,7 +1175,7 @@ class PlainClassSchema(RecordSchema):
         if field := self.signature_fields.get(name):
             _annotation, _default = field
             if actual_type is _annotation:
-                default = _default
+                default = _default or dataclasses.MISSING
         field_obj = RecordField(
             py_type=actual_type,
             name=name,