Исходный код vk_maria.dispatcher.fsm.storage.file.pickle
import pathlib
import pickle
from .base import FileStorage
[документация]class PickleStorage(FileStorage):
"""
Pickle File storage based on MemoryStorage
"""
[документация] def read(self, path: pathlib.Path):
if not pathlib.Path(path).exists():
with path.open('wb+') as f:
pickle.dump({}, f, protocol=pickle.HIGHEST_PROTOCOL)
return {}
with path.open('rb') as f:
return pickle.load(f)
[документация] def write(self, path: pathlib.Path):
with path.open('wb') as f:
return pickle.dump(self.data, f, protocol=pickle.HIGHEST_PROTOCOL)