Add XCom serilizer for pendulum.date.Date

Author: mrlimacz

task-sdk/src/airflow/sdk/serde/serializers/datetime.py

@@ -33,7 +33,13 @@
 
 __version__ = 2
 
-serializers = ["datetime.date", "datetime.datetime", "datetime.timedelta", "pendulum.datetime.DateTime"]
+serializers = [
+    "datetime.date",
+    "datetime.datetime",
+    "datetime.timedelta",
+    "pendulum.datetime.DateTime",
+    "pendulum.date.Date",
+]
 deserializers = serializers
 
 TIMESTAMP = "timestamp"
@@ -62,7 +68,7 @@ def serialize(o: object) -> tuple[U, str, int, bool]:
 def deserialize(cls: type, version: int, data: dict | str) -> datetime.date | datetime.timedelta:
     import datetime
 
-    from pendulum import DateTime
+    from pendulum import Date, DateTime
 
     tz: datetime.tzinfo | None = None
     if isinstance(data, dict) and TIMEZONE in data:
@@ -95,6 +101,9 @@ def deserialize(cls: type, version: int, data: dict | str) -> datetime.date | da
     if cls is datetime.timedelta and isinstance(data, str | float):
         return datetime.timedelta(seconds=float(data))
 
+    if cls is Date and isinstance(data, str):
+        return Date.fromisoformat(data)
+
     if cls is datetime.date and isinstance(data, str):
         return datetime.date.fromisoformat(data)
 

task-sdk/tests/task_sdk/serde/test_serializers.py

@@ -33,7 +33,7 @@
 from dateutil.tz import tzutc
 from kubernetes.client import models as k8s
 from packaging import version
-from pendulum import DateTime
+from pendulum import Date, DateTime
 from pendulum.tz.timezone import FixedTimezone, Timezone
 from pydantic import BaseModel, Field
 from pydantic.dataclasses import dataclass as pydantic_dataclass
@@ -91,6 +91,7 @@ class TestSerializers:
                 id="datetime_utc",
             ),
             pytest.param(DateTime(2022, 7, 10, tzinfo=pendulum.tz.UTC), id="pendulum_datetime_utc"),
+            pytest.param(Date(2026, 1, 21), id="pendulum_date"),
             pytest.param(datetime.date(2022, 7, 10), id="date"),
             pytest.param(datetime.timedelta(days=320), id="timedelta"),
             pytest.param(
@@ -112,7 +113,7 @@ def test_datetime(self, input_obj):
         """Test serialization and deserialization of various datetime-related objects."""
         serialized_obj = serialize(input_obj)
         deserialized_obj = deserialize(serialized_obj)
-        if isinstance(input_obj, (datetime.date, datetime.timedelta)):
+        if isinstance(input_obj, (datetime.date, datetime.timedelta, Date)):
             assert input_obj == deserialized_obj
         else:
             assert input_obj.timestamp() == deserialized_obj.timestamp()