Add supported for Annotated to struct type. (#375)

Signed-off-by: Pascal Tomecek <[email protected]>

conda/dev-environment-unix.yml

@@ -52,6 +52,7 @@ dependencies:
  - threadpoolctl
  - tornado
  - twine
+ - typing-extensions
  - unzip
  - wheel
  - zip

conda/dev-environment-win.yml

@@ -50,4 +50,5 @@ dependencies:
  - threadpoolctl
  - tornado
  - twine
+ - typing-extensions
  - wheel

csp/impl/types/container_type_normalizer.py

@@ -1,5 +1,6 @@
 import numpy
 import typing
+import typing_extensions
 
 import csp.typing
 from csp.impl.types.typing_utils import CspTypingUtils, FastList
@@ -70,6 +71,8 @@ def _convert_containers_to_typing_generic_meta(cls, typ, is_within_container):
 
     @classmethod
     def normalized_type_to_actual_python_type(cls, typ, level=0):
+        if isinstance(typ, typing_extensions._AnnotatedAlias):
+            typ = CspTypingUtils.get_origin(typ)
         if CspTypingUtils.is_generic_container(typ):
             if CspTypingUtils.get_origin(typ) is FastList and level == 0:
                 return [cls.normalized_type_to_actual_python_type(typ.__args__[0], level + 1), True]

csp/tests/impl/test_struct.py

@@ -7,6 +7,7 @@
 import unittest
 from datetime import date, datetime, time, timedelta
 from typing import Dict, List, Set, Tuple
+from typing_extensions import Annotated
 
 import csp
 from csp.impl.struct import define_nested_struct, define_struct, defineNestedStruct, defineStruct
@@ -2943,6 +2944,22 @@ def test_dir(self):
         self.assertIn("__metadata__", dir_output)
         self.assertEqual(dir_output, sorted(dir_output))
 
+    def test_annotations(self):
+        class StructWithAnnotations(csp.Struct):
+            b: Annotated[float, "test"]
+            d: Annotated[Dict[str, Annotated[int, "test_int"]], "test_dict"]
+            s: str
+
+        self.assertEqual(
+            StructWithAnnotations.metadata(typed=True),
+            {
+                "b": Annotated[float, "test"],
+                "d": Annotated[Dict[str, Annotated[int, "test_int"]], "test_dict"],
+                "s": str,
+            },
+        )
+        self.assertEqual(StructWithAnnotations.metadata(typed=False), {"b": float, "d": dict, "s": str})
+
 
 if __name__ == "__main__":
     unittest.main()

pyproject.toml

@@ -7,6 +7,7 @@ requires = [
     "ruamel.yaml",
     "scikit-build",
     "setuptools>=69,<74",
+    "typing-extensions",
 ]
 build-backend="setuptools.build_meta"
 
@@ -29,6 +30,7 @@ dependencies = [
     "pytz",
     "ruamel.yaml",
     "sqlalchemy",
+    "typing-extensions",
 ]
 
 classifiers = [