Skip to content

Instantly share code, notes, and snippets.

@iisaka51
Created October 25, 2022 05:27
Show Gist options
  • Save iisaka51/ebe7845fd1911c283e726d0181e6d77a to your computer and use it in GitHub Desktop.
Save iisaka51/ebe7845fd1911c283e726d0181e6d77a to your computer and use it in GitHub Desktop.
example for adding serializer
from datajuggler.serializer.abstract import (
AbstractSerializer, register_serializer
)
try:
import cloudpickle
cloudpickle_enable = True
except ImportError: # pragma: no cover
cloudpickle_enable = False
cloudpickle = AbstractSerializer()
class CloudpickleSerializer(AbstractSerializer):
def __init__(self):
super().__init__(format='cloudpickle',
extension=['pickle', 'cpickle'],
package='cloudpickle',
enable=cloudpickle_enable,
overwrite=True)
def loads(self, s, **kwargs):
data = cloudpickle.loads(s, **kwargs)
return data
def dumps(self, d, **kwargs):
data = cloudpickle.dumps(d, **kwargs)
return data
register_serializer(CloudpickleSerializer)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment