| |
| 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) |
|
|
