fz0x1/HPI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

core.kompress: support .zst extension, seems more conventional than .zstd

Browse source
This commit is contained in:
Dima Gerasimov 2023-02-28 22:53:06 +00:00 • committed by karlicoss
parent db2cd00bed
commit b94904f5ee
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
my/core/kompress.py
View file

@ -18,13 +18,14 @@ class Ext:
zip = '.zip' zip = '.zip'
lz4 = '.lz4' lz4 = '.lz4'
zstd = '.zstd' zstd = '.zstd'
zst = '.zst'
targz = '.tar.gz' targz = '.tar.gz'
def is_compressed(p: Path) -> bool: def is_compressed(p: Path) -> bool:
# todo kinda lame way for now.. use mime ideally? # todo kinda lame way for now.. use mime ideally?
# should cooperate with kompress.kopen? # should cooperate with kompress.kopen?
return any(p.name.endswith(ext) for ext in {Ext.xz, Ext.zip, Ext.lz4, Ext.zstd, Ext.targz}) return any(p.name.endswith(ext) for ext in {Ext.xz, Ext.zip, Ext.lz4, Ext.zstd, Ext.zst, Ext.targz})
def _zstd_open(path: Path, *args, **kwargs) -> IO[str]: def _zstd_open(path: Path, *args, **kwargs) -> IO[str]:
@ -70,7 +71,7 @@ def kopen(path: PathIsh, *args, mode: str='rt', **kwargs) -> IO[str]:
elif name.endswith(Ext.lz4): elif name.endswith(Ext.lz4):
import lz4.frame # type: ignore import lz4.frame # type: ignore
return lz4.frame.open(str(pp), mode, *args, **kwargs) return lz4.frame.open(str(pp), mode, *args, **kwargs)
elif name.endswith(Ext.zstd): elif name.endswith(Ext.zstd) or name.endswith(Ext.zst):
return _zstd_open(pp, mode, *args, **kwargs) return _zstd_open(pp, mode, *args, **kwargs)
elif name.endswith(Ext.targz): elif name.endswith(Ext.targz):
import tarfile import tarfile