Added pickle serializer. (#232)

Author: s3rius

taskiq/serializers/__init__.py

@@ -3,10 +3,12 @@
 from .json_serializer import JSONSerializer
 from .msgpack_serializer import MSGPackSerializer
 from .orjson_serializer import ORJSONSerializer
+from .pickle import PickleSerializer
 
 __all__ = [
     "JSONSerializer",
     "ORJSONSerializer",
     "MSGPackSerializer",
     "CBORSerializer",
+    "PickleSerializer",
 ]

taskiq/serializers/pickle.py

@@ -0,0 +1,16 @@
+import pickle
+from typing import Any
+
+from taskiq.abc.serializer import TaskiqSerializer
+
+
+class PickleSerializer(TaskiqSerializer):
+    """Serializer that uses pickle."""
+
+    def dumpb(self, value: Any) -> bytes:
+        """Dumps value to bytes."""
+        return pickle.dumps(value)
+
+    def loadb(self, value: bytes) -> Any:
+        """Loads value from bytes."""
+        return pickle.loads(value)  # noqa: S301

tests/serializers/test_serializers.py

@@ -11,6 +11,7 @@
     JSONSerializer,
     MSGPackSerializer,
     ORJSONSerializer,
+    PickleSerializer,
 )
 
 
@@ -21,6 +22,7 @@
         ORJSONSerializer(),
         CBORSerializer(),
         MSGPackSerializer(),
+        PickleSerializer(),
     ],
 )
 @pytest.mark.parametrize(
@@ -43,6 +45,7 @@ def test_generic_serializer(serializer: TaskiqSerializer, data: Any) -> None:
     [
         ORJSONSerializer(),
         CBORSerializer(),
+        PickleSerializer(),
     ],
 )
 def test_uuid_serialization(serializer: TaskiqSerializer) -> None:
@@ -55,6 +58,7 @@ def test_uuid_serialization(serializer: TaskiqSerializer) -> None:
     [
         CBORSerializer(),
         MSGPackSerializer(),
+        PickleSerializer(),
     ],
 )
 def test_datetime_serialization(serializer: TaskiqSerializer) -> None: