29 lines
739 B
Python
29 lines
739 B
Python
# -*- coding: utf-8 -*-
|
|
|
|
import base64
|
|
import pickle
|
|
|
|
from benedict.serializers.abstract import AbstractSerializer
|
|
|
|
|
|
class PickleSerializer(AbstractSerializer):
|
|
"""
|
|
This class describes a pickle serializer.
|
|
"""
|
|
|
|
def __init__(self):
|
|
super(PickleSerializer, self).__init__(
|
|
extensions=[
|
|
"pickle",
|
|
],
|
|
)
|
|
|
|
def decode(self, s, **kwargs):
|
|
encoding = kwargs.pop("encoding", "utf-8")
|
|
return pickle.loads(base64.b64decode(s.encode(encoding)), **kwargs)
|
|
|
|
def encode(self, d, **kwargs):
|
|
encoding = kwargs.pop("encoding", "utf-8")
|
|
kwargs.setdefault("protocol", 2)
|
|
return base64.b64encode(pickle.dumps(d, **kwargs)).decode(encoding)
|