|
|
|
import pickle |
|
|
|
from .base import BaseFileHandler |
|
|
|
|
|
class PickleHandler(BaseFileHandler): |
|
|
|
str_like = False |
|
|
|
def load_from_fileobj(self, file, **kwargs): |
|
return pickle.load(file, **kwargs) |
|
|
|
def load_from_path(self, filepath, **kwargs): |
|
return super(PickleHandler, self).load_from_path( |
|
filepath, mode='rb', **kwargs) |
|
|
|
def dump_to_str(self, obj, **kwargs): |
|
kwargs.setdefault('protocol', 2) |
|
return pickle.dumps(obj, **kwargs) |
|
|
|
def dump_to_fileobj(self, obj, file, **kwargs): |
|
kwargs.setdefault('protocol', 2) |
|
pickle.dump(obj, file, **kwargs) |
|
|
|
def dump_to_path(self, obj, filepath, **kwargs): |
|
super(PickleHandler, self).dump_to_path( |
|
obj, filepath, mode='wb', **kwargs) |
|
|
