better implementation

Author: giograno

src/py_avro_schema/_alias.py

@@ -7,7 +7,7 @@
 
 import dataclasses
 from collections import defaultdict
-from typing import Annotated, Final, Type, get_args, get_origin
+from typing import Annotated, Type, get_args, get_origin
 
 FQN = str
 """Fully qualified name for a Python type"""
@@ -108,10 +108,6 @@ def get_field_aliases_and_actual_type(py_type: Type) -> tuple[list[str] | None,
     Check if a type contains an alias metadata via `Alias` or `Aliases` as metadata.
     It returns the eventual aliases and the type.
     """
-
-    if get_origin(py_type) is Final:
-        return [], get_args(py_type)[0]
-
     # py_type is not annotated. It can't have aliases
     if get_origin(py_type) is not Annotated:
         return [], py_type

src/py_avro_schema/_schemas.py

@@ -33,6 +33,7 @@
     Annotated,
     Any,
     Dict,
+    Final,
     ForwardRef,
     List,
     Literal,
@@ -378,6 +379,28 @@ def data(self, names: NamesType) -> JSONType:
         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)
+        real_type = get_args(py_type)[0]
+        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
+
+
 @register_schema
 class DictAsJSONSchema(Schema):
     """An Avro string schema representing a Python Dict[str, Any] or List[Dict[str, Any]] assuming JSON serialization"""