commit 949e8ff515db49411fdf7d509ffba2f5005f9578 Author: feie9456 Date: Mon Jul 21 10:51:00 2025 +0800 first commit diff --git a/.DS_Store b/.DS_Store new file mode 100644 index 0000000..30906bc Binary files /dev/null and b/.DS_Store differ diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..505a3b1 --- /dev/null +++ b/.gitignore @@ -0,0 +1,10 @@ +# Python-generated files +__pycache__/ +*.py[oc] +build/ +dist/ +wheels/ +*.egg-info + +# Virtual environments +.venv diff --git a/.python-version b/.python-version new file mode 100644 index 0000000..e4fba21 --- /dev/null +++ b/.python-version @@ -0,0 +1 @@ +3.12 diff --git a/README.md b/README.md new file mode 100644 index 0000000..e69de29 diff --git a/compress.py b/compress.py new file mode 100644 index 0000000..23855db --- /dev/null +++ b/compress.py @@ -0,0 +1,12 @@ + +import zipfile +import os + +temp_dir= "/Users/feie9454/Documents/PyTools/P1太阳总" +with zipfile.ZipFile("P1太阳总.zip", 'w', zipfile.ZIP_DEFLATED) as zipf: + webp_files_to_zip = os.listdir(temp_dir) + + for webp_file in webp_files_to_zip: + file_path_in_temp = os.path.join(temp_dir, webp_file) + zipf.write(file_path_in_temp, arcname=webp_file) + print(f" - 已添加 {webp_file} 到 ZIP 包") \ No newline at end of file diff --git a/process_images.py b/process_images.py new file mode 100644 index 0000000..12885f1 --- /dev/null +++ b/process_images.py @@ -0,0 +1,141 @@ +# -*- coding: utf-8 -*- + +import os +import re +import sys +import subprocess +import shutil +import tempfile +import zipfile + +# --- 配置区 --- +IMAGE_EXTENSIONS = ('.png', '.jpg', '.jpeg', '.bmp', '.gif', '.tiff') +# --- 配置区结束 --- + +def check_ffmpeg(): + """检查系统中是否安装了 ffmpeg""" + if shutil.which("ffmpeg") is None: + print("错误：找不到 ffmpeg。") + print("请确保你已经安装了 ffmpeg，并将其添加到了系统的 PATH 环境变量中。") + print("安装指南: https://ffmpeg.org/download.html") + return False + return True + +def extract_number_from_filename(filename): + """从文件名中提取最后一个遇到的数字序列用于排序。""" + numbers = re.findall(r'\d+', filename) + if numbers: + return int(numbers[-1]) + print(f"警告：文件名 '{filename}' 中未找到数字，将按0进行排序。") + return 0 + +def main(): + """主执行函数""" + print("--- 图片序列转 WebP 并压缩为 ZIP 脚本 (支持自定义起始编号) ---") + + # 1. 检查命令行参数 (现在支持3个或4个参数) + if not (3 <= len(sys.argv) <= 4): + print("\n使用方法:") + print(f" python {sys.argv[0]} <源图片文件夹> <输出的ZIP文件名> [<起始编号>]") + print("\n示例 (从0开始):") + print(f" python {sys.argv[0]} ./my_images my_archive.zip") + print("\n示例 (从40开始):") + print(f" python {sys.argv[0]} ./my_images my_archive.zip 40") + sys.exit(1) + + source_dir = sys.argv[1] + output_zip_name = sys.argv[2] + + # 初始化起始编号为默认值 0 + start_index = 0 + + # 如果提供了第三个参数，则用它作为起始编号 + if len(sys.argv) == 4: + try: + start_index = int(sys.argv[3]) + if start_index < 0: + print("错误：起始编号不能为负数。") + sys.exit(1) + except ValueError: + print(f"错误：提供的起始编号 '{sys.argv[3]}' 不是一个有效的整数。") + sys.exit(1) + + if not output_zip_name.lower().endswith('.zip'): + output_zip_name += '.zip' + + # 2. 验证输入 + if not check_ffmpeg(): + sys.exit(1) + + if not os.path.isdir(source_dir): + print(f"错误：源文件夹 '{source_dir}' 不存在或不是一个目录。") + sys.exit(1) + + # 3. 查找并排序图片文件 + print(f"\n[1/5] 正在扫描文件夹: {source_dir}") + try: + all_files = os.listdir(source_dir) + image_files = [f for f in all_files if f.lower().endswith(IMAGE_EXTENSIONS)] + except OSError as e: + print(f"错误：无法读取文件夹 '{source_dir}': {e}") + sys.exit(1) + + if not image_files: + print("指定的文件夹中未找到任何支持的图片文件。") + sys.exit(0) + + sorted_images = sorted(image_files, key=extract_number_from_filename) + + print("找到并排序后的图片文件：") + for img in sorted_images: + print(f" - {img}") + + # 使用临时目录来存放生成的 webp 文件 + with tempfile.TemporaryDirectory() as temp_dir: + print(f"\n[2/5] 开始转换图片 (起始编号: {start_index})...") + total_files = len(sorted_images) + + # 4. 依次使用 ffmpeg 进行编码 + for index, image_name in enumerate(sorted_images): + # 计算最终的输出文件名编号 + output_number = start_index + index + output_filename = f"{output_number}.webp" + + input_path = os.path.join(source_dir, image_name) + output_path = os.path.join(temp_dir, output_filename) + + command = ["ffmpeg", "-i", input_path, "-y", output_path] + + print(f" [{index + 1}/{total_files}] 正在转换: {image_name} -> {output_filename}") + + try: + subprocess.run( + command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, encoding='utf-8' + ) + except subprocess.CalledProcessError as e: + print(f" 错误: 转换 '{image_name}' 时失败。\n FFmpeg 错误信息:\n{e.stderr}") + except Exception as e: + print(f" 发生未知错误: {e}") + + # 5. 将临时目录中的 WebP 文件压缩成 ZIP + print(f"\n[3/5] 正在将 WebP 文件压缩到: {output_zip_name}") + + try: + with zipfile.ZipFile(output_zip_name, 'w', zipfile.ZIP_DEFLATED) as zipf: + webp_files_to_zip = sorted(os.listdir(temp_dir), key=lambda f: int(f.split('.')[0])) + + for webp_file in webp_files_to_zip: + file_path_in_temp = os.path.join(temp_dir, webp_file) + zipf.write(file_path_in_temp, arcname=webp_file) + print(f" - 已添加 {webp_file} 到 ZIP 包") + + except Exception as e: + print(f"错误：创建 ZIP 文件时失败: {e}") + sys.exit(1) + + print("\n[4/5] 临时文件已自动清理。") + print(f"\n[5/5] --- 所有任务完成！输出文件已保存为 '{output_zip_name}' ---") + + +if __name__ == "__main__": + main() \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..c271f89 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,7 @@ +[project] +name = "pytools" +version = "0.1.0" +description = "Add your description here" +readme = "README.md" +requires-python = ">=3.12" +dependencies = [] diff --git a/uv.lock b/uv.lock new file mode 100644 index 0000000..0f7b21b --- /dev/null +++ b/uv.lock @@ -0,0 +1,8 @@ +version = 1 +revision = 2 +requires-python = ">=3.12" + +[[package]] +name = "pytools" +version = "0.1.0" +source = { virtual = "." }