Implement web-poet serializers

scrapy_zyte_api/__init__.py

@@ -5,6 +5,9 @@
 
     install_reactor("twisted.internet.asyncioreactor.AsyncioSelectorReactor")
 
+# Register web-poet serializers
+from . import _serialization  # noqa: F401
+
 from ._annotations import ExtractFrom, actions, custom_attrs
 from ._middlewares import (
     ScrapyZyteAPIDownloaderMiddleware,

scrapy_zyte_api/_serialization.py

@@ -0,0 +1,44 @@
+import json
+
+from ._page_inputs import Actions, Geolocation, Screenshot
+
+try:
+    from web_poet.serialization import SerializedLeafData, register_serialization
+except ImportError:
+    pass
+else:
+    def _serialize_Actions(o: Actions) -> SerializedLeafData:
+        return {"results.json": json.dumps(o.results).encode()}
+
+
+    def _deserialize_Actions(cls: type[Actions], data: SerializedLeafData) -> Actions:
+        return cls(results=json.loads(data["results.json"]))
+
+
+    register_serialization(_serialize_Actions, _deserialize_Actions)
+
+
+    def _serialize_Geolocation(o: Geolocation) -> SerializedLeafData:
+        return {}
+
+
+    def _deserialize_Geolocation(
+        cls: type[Geolocation], data: SerializedLeafData
+    ) -> Geolocation:
+        return cls()
+
+
+    register_serialization(_serialize_Geolocation, _deserialize_Geolocation)
+
+
+    def _serialize_Screenshot(o: Screenshot) -> SerializedLeafData:
+        return {"body": o.body}
+
+
+    def _deserialize_Screenshot(
+        cls: type[Screenshot], data: SerializedLeafData
+    ) -> Screenshot:
+        return cls(body=data["body"])
+
+
+    register_serialization(_serialize_Screenshot, _deserialize_Screenshot)