zhou
abc1152538
1. 将folderzip/folderback/gittool中的旧TaskSpec定义替换为@px.task装饰器 2. 重构pymake模块，将maturin_build_cmd转为常量定义，合并别名配置 3. 精简测试文件中的冗余测试用例
77 lines
1.9 KiB
Python
77 lines
1.9 KiB
Python
"""文件夹压缩工具.
|
|
|
|
压缩目录下的所有文件/文件夹为 zip 文件,
|
|
默认压缩当前目录下的所有子文件夹.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
import pyflowx as px
|
|
|
|
# ============================================================================
|
|
# 配置
|
|
# ============================================================================
|
|
|
|
IGNORE_DIRS: list[str] = [".git", ".idea", ".vscode", "__pycache__"]
|
|
IGNORE_FILES: list[str] = [".gitignore"]
|
|
IGNORE: list[str] = [*IGNORE_DIRS, *IGNORE_FILES]
|
|
IGNORE_EXT: list[str] = [".zip", ".rar", ".7z", ".tar", ".gz"]
|
|
|
|
|
|
# ============================================================================
|
|
# 辅助函数
|
|
# ============================================================================
|
|
|
|
|
|
def archive_folder(folder: Path) -> None:
|
|
"""压缩单个文件夹."""
|
|
shutil.make_archive(
|
|
str(folder.with_name(folder.name)),
|
|
format="zip",
|
|
base_dir=folder,
|
|
)
|
|
print(f"压缩完成: {folder.name}.zip")
|
|
|
|
|
|
def zip_folders(cwd: str = ".") -> None:
|
|
"""压缩目录下的所有文件夹.
|
|
|
|
Parameters
|
|
----------
|
|
cwd : str
|
|
工作目录
|
|
"""
|
|
cwd_path = Path(cwd)
|
|
if not cwd_path.exists():
|
|
print(f"目录不存在: {cwd_path}")
|
|
return
|
|
|
|
dirs: list[Path] = [
|
|
e for e in cwd_path.iterdir() if e.is_dir() and e.name not in IGNORE_DIRS and e.suffix not in IGNORE_EXT
|
|
]
|
|
|
|
for dir_path in dirs:
|
|
archive_folder(dir_path)
|
|
|
|
|
|
@px.task
|
|
def folderzip_default() -> None:
|
|
"""压缩当前目录下的所有文件夹."""
|
|
zip_folders(".")
|
|
|
|
|
|
def main() -> None:
|
|
"""文件夹压缩工具主函数."""
|
|
runner = px.CliRunner(
|
|
strategy="thread",
|
|
description="FolderZip - 文件夹压缩工具",
|
|
aliases={
|
|
# 压缩当前目录下的所有文件夹
|
|
"z": folderzip_default,
|
|
},
|
|
)
|
|
runner.run_cli()
|