[F] Fix write file when parent dir doesn't exist
This commit is contained in:
@@ -1,6 +1,6 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
__version__ = "1.0.26"
|
__version__ = "1.0.27"
|
||||||
|
|
||||||
import time
|
import time
|
||||||
import logging
|
import logging
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ from pathlib import Path
|
|||||||
import zstandard as zstd
|
import zstandard as zstd
|
||||||
import orjson
|
import orjson
|
||||||
|
|
||||||
|
from . import write
|
||||||
|
|
||||||
zstd_d = zstd.ZstdDecompressor()
|
zstd_d = zstd.ZstdDecompressor()
|
||||||
zstd_c = zstd.ZstdCompressor(level=5, write_checksum=True, threads=-1)
|
zstd_c = zstd.ZstdCompressor(level=5, write_checksum=True, threads=-1)
|
||||||
@@ -31,16 +32,16 @@ def write_json_zst(file_path: str | Path, data: dict | list, **kwargs):
|
|||||||
file_path (str): The path to the .json.zst file.
|
file_path (str): The path to the .json.zst file.
|
||||||
data (dict or list): The data to dump.
|
data (dict or list): The data to dump.
|
||||||
"""
|
"""
|
||||||
Path(file_path).write_bytes(zstd_c.compress(orjson.dumps(data, **kwargs)))
|
write(file_path, zstd_c.compress(orjson.dumps(data, **kwargs)))
|
||||||
|
|
||||||
|
|
||||||
def load_pickle_zst(file_path: str | Path) -> object:
|
def load_pickle_zst(file_path: str | Path):
|
||||||
with Path(file_path).open('rb') as f:
|
with Path(file_path).open('rb') as f:
|
||||||
return pickle.loads(zstd_d.stream_reader(f).read())
|
return pickle.loads(zstd_d.stream_reader(f).read())
|
||||||
|
|
||||||
|
|
||||||
def write_pickle_zst(file_path: str | Path, data: object):
|
def write_pickle_zst(file_path: str | Path, data):
|
||||||
Path(file_path).write_bytes(zstd_c.compress(pickle.dumps(data)))
|
write(file_path, zstd_c.compress(pickle.dumps(data)))
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|||||||
Reference in New Issue
Block a user