localstack-py-avro-schema 3.9.7__py3-none-any.whl → 3.9.8__py3-none-any.whl

{localstack_py_avro_schema-3.9.7.dist-info → localstack_py_avro_schema-3.9.8.dist-info}/METADATA

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

@@ -1,12 +1,12 @@
-localstack_py_avro_schema-3.9.7.dist-info/licenses/LICENSE,sha256=zE1N4JILDTkSIDtdmqdnKKxKEQh_VdqeoAV2230eNOI,10141
-localstack_py_avro_schema-3.9.7.dist-info/licenses/LICENSE_HEADER.txt,sha256=Nx3RGmYbJKjIKM8Y2yMrCPJgCw4oBt_H62PxDKlAsto,564
+localstack_py_avro_schema-3.9.8.dist-info/licenses/LICENSE,sha256=zE1N4JILDTkSIDtdmqdnKKxKEQh_VdqeoAV2230eNOI,10141
+localstack_py_avro_schema-3.9.8.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=K_7aDBkLGF0PGixRNvV37VMBnmEoSX9-CbzAbj-c9hc,3823
-py_avro_schema/_schemas.py,sha256=OxSp5EFusu__X1PMA4O02nPN9c6LiAKwr-eavy5A1KY,49819
+py_avro_schema/_schemas.py,sha256=MwF-aSPlaZ8M83DxzN6F_H-DMuWC80DrsA2cQpxbJ5E,51121
 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.7.dist-info/METADATA,sha256=aLTpa4yDYNQKJ1ypOTBHFjonme-SRUYAiXJadR55Jyg,15164
-localstack_py_avro_schema-3.9.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-localstack_py_avro_schema-3.9.7.dist-info/top_level.txt,sha256=0t4oO_5rsdfZGSk8iSzK4TjdyfVz-RE1IZ2TIz6YvI0,15
-localstack_py_avro_schema-3.9.7.dist-info/RECORD,,
+localstack_py_avro_schema-3.9.8.dist-info/METADATA,sha256=YrpE6QS35aU-FR7U4oqpej0fn253ZPXuJRaEgOiSVPk,15164
+localstack_py_avro_schema-3.9.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+localstack_py_avro_schema-3.9.8.dist-info/top_level.txt,sha256=0t4oO_5rsdfZGSk8iSzK4TjdyfVz-RE1IZ2TIz6YvI0,15
+localstack_py_avro_schema-3.9.8.dist-info/RECORD,,

py_avro_schema/_schemas.py

@@ -33,6 +33,7 @@ from typing import (
     Annotated,
     Any,
     Dict,
+    Final,
     ForwardRef,
     List,
     Literal,
@@ -378,6 +379,31 @@ class LiteralSchema(Schema):
         return self.literal_value_schema.data(names=names)
 
 
+@register_schema
+class FinalSchema(Schema):
+    """An Avro schema for Python ``typing.Final``"""
+
+    def __init__(self, py_type: Type, namespace: Optional[str] = None, options: Option = Option(0)):
+        """An Avro schema for Python ``typing.Final``"""
+        super().__init__(py_type, namespace, options)
+        py_type = _type_from_annotated(py_type)
+        try:
+            real_type = get_args(py_type)[0]
+        except IndexError:
+            raise TypeError("Can't generate Avro schema from Python typing.Final without a type parameter")
+        self.real_schema = _schema_obj(real_type, namespace=namespace, options=options)
+
+    def data(self, names: NamesType) -> JSONType:
+        """Return the schema data"""
+        return self.real_schema.data(names=names)
+
+    @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)
+        return get_origin(py_type) is Final or py_type is Final
+
+
 @register_schema
 class DictAsJSONSchema(Schema):
     """An Avro string schema representing a Python Dict[str, Any] or List[Dict[str, Any]] assuming JSON serialization"""
@@ -1156,8 +1182,13 @@ class PlainClassSchema(RecordSchema):
         super().__init__(py_type, namespace=namespace, options=options)
         py_type = _type_from_annotated(py_type)
 
+        # Try to get resolved type hints, but fall back to raw annotations if there are unresolved forward refs
+        try:
+            type_hints = get_type_hints(py_type, include_extras=True)
+        except NameError:
+            type_hints = py_type.__annotations__
         self.py_fields: list[tuple[str, type]] = []
-        for k, v in py_type.__annotations__.items():
+        for k, v in type_hints.items():
             self.py_fields.append((k, v))
         # We store __init__ parameters with default values. They can be used as defaults for the record.
         self.signature_fields = {