dstandish commented on code in PR #58289:
URL: https://github.com/apache/airflow/pull/58289#discussion_r2565194192
##########
airflow-core/src/airflow/serialization/serialized_objects.py:
##########
@@ -458,6 +471,45 @@ def decode_timetable(var: dict[str, Any]) -> Timetable:
return timetable_class.deserialize(var[Encoding.VAR])
+def _load_partition_mapper(importable_string) -> PartitionMapper | None:
+ if importable_string.startswith("airflow.timetables."):
+ return import_string(importable_string)
+ else:
+ return None
+
+
+def encode_partition_mapper(var: PartitionMapper) -> dict[str, Any]:
+ """
+ Encode a PartitionMapper instance.
+
+ This delegates most of the serialization work to the type, so the behavior
+ can be completely controlled by a custom subclass.
+
+ :meta private:
+ """
+ partition_mapper_class = type(var)
+ importable_string = qualname(partition_mapper_class)
+ if _load_partition_mapper(importable_string) is None:
+ raise _PartitionMapperNotFound(importable_string)
Review Comment:
i copied the behavior for timetables; i think we might ultimately make this
pluggable in the same way.
but meantime there's an odd error message that says something about sessions
and such in the timetable case; i'm not sure what that's about and if it might
also apply here.... so i just copied it; if you think it's not necessary, happy
to remove it. lmk.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
