mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
Add zlib compress and decompress functions to file_helpers.py
This commit is contained in:
parent
42e716baa6
commit
c1327adf6b
@ -4,6 +4,7 @@ import logging
|
|||||||
import pathlib
|
import pathlib
|
||||||
import tempfile
|
import tempfile
|
||||||
import zipfile
|
import zipfile
|
||||||
|
import zlib
|
||||||
import hashlib
|
import hashlib
|
||||||
from typing import BinaryIO
|
from typing import BinaryIO
|
||||||
import mimetypes
|
import mimetypes
|
||||||
@ -166,7 +167,7 @@ class FileHelpers:
|
|||||||
with value_to_hash.open("rb") as f:
|
with value_to_hash.open("rb") as f:
|
||||||
while True:
|
while True:
|
||||||
# Read 2^16 byte chunks, 64 KiB.
|
# Read 2^16 byte chunks, 64 KiB.
|
||||||
data = value_to_hash.read(65536)
|
data = f.read(65536)
|
||||||
|
|
||||||
# Break if end of file
|
# Break if end of file
|
||||||
if not data:
|
if not data:
|
||||||
@ -444,3 +445,29 @@ class FileHelpers:
|
|||||||
zip_ref.extractall(temp_dir)
|
zip_ref.extractall(temp_dir)
|
||||||
if user_id:
|
if user_id:
|
||||||
return temp_dir
|
return temp_dir
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def zlib_compress_bytes(bytes_to_compress: bytes) -> bytes:
|
||||||
|
"""Compress provided bytes using zlib default compression settings. Returns
|
||||||
|
compressed bytes.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
bytes_to_compress (bytes): Bytes to compress.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bytes: Compressed bytes.
|
||||||
|
"""
|
||||||
|
return zlib.compress(bytes_to_compress)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def zlib_decompress_bytes(bytes_to_decompress: bytes) -> bytes:
|
||||||
|
"""Decompress provided bytes using zlib default settings. Returns decompressed
|
||||||
|
bytes.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
bytes_to_decompress (bytes): Compresses bytes to decompress.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bytes: Decompressed bytes.
|
||||||
|
"""
|
||||||
|
return zlib.decompress(bytes_to_decompress)
|
||||||
|
Loading…
Reference in New Issue
Block a user