Python/ResizeByFile.py at main · Leet-Lyn/Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# 请帮我写个中文的 Python 脚本，批注也是中文：
# 在脚本开始前询问我源文件位置。
# 将该图片文件（bmp、jpg、jpeg、png、webp、avif、heic）调整大小。
# 让我选择：1.640*480：宽度640，高度根据宽度调整保持纵横比；2.800*600：宽度800，高度根据宽度调整保持纵横比；3.1024*768：宽度1024，高度根据宽度调整保持纵横比；4.1280*720：宽度1280，高度根据宽度调整保持纵横比；5.1920*1080：宽度1920，高度根据宽度调整保持纵横比；5.3840*2160：宽度3840，高度根据宽度调整保持纵横比；5.7680*4320：宽度7680，高度根据宽度调整保持纵横比；
# 生成的文件替换源文件。
# 如此循环，再次前询问我源文件位置。

# 导入模块
from PIL import Image
import os
import sys

def resize_image(image_path, width):
    """调整图片大小并覆盖原始文件"""
    try:
        # 打开原始图片
        with Image.open(image_path) as img:
            # 计算新高度（保持原始宽高比）
            w_percent = width / float(img.size[0])
            new_height = int(float(img.size[1]) * float(w_percent))

            # 调整图片尺寸
            resized_img = img.resize((width, new_height), Image.LANCZOS)

            # 保存图片（覆盖原始文件，保持原始格式）
            resized_img.save(image_path, quality=95)
            print(f"✓ 已调整尺寸: {width}x{new_height} | 覆盖保存: {image_path}")

    except Exception as e:
        print(f"✗ 处理失败: {e}")

def main():
    """主函数：用户交互和循环处理"""
    # 支持的图片格式
    valid_exts = ('.bmp', '.jpg', '.jpeg', '.png', '.webp', '.avif', '.heic')

    # 尺寸选项配置 (1280×720 设为默认选项)
    size_options = {
        '1': 640,
        '2': 800,
        '3': 1024,
        '4': 1280,  # 默认选项
        '5': 1920,
        '6': 3840,
        '7': 7680
    }

    while True:
        # 获取文件路径
        file_path = input("\n请输入图片路径（输入 q 退出）: ").strip()

        # 退出条件
        if file_path.lower() in ['q', 'quit', 'exit']:
            print("\n程序已退出")
            break

        # 检查文件是否存在
        if not os.path.isfile(file_path):
            print("错误：文件不存在，请重新输入")
            continue

        # 检查文件格式
        ext = os.path.splitext(file_path)[1].lower()
        if ext not in valid_exts:
            print(f"错误：不支持的格式 {ext}，支持格式: {', '.join(valid_exts)}")
            continue

        # 显示尺寸选项（默认选项4）
        print("\n请选择目标尺寸（直接回车将使用默认1280×720）:")
        print("1. 640×480   (标清)")
        print("2. 800×600   (SVGA)")
        print("3. 1024×768  (XGA)")
        print("4. 1280×720  (高清) [默认]")
        print("5. 1920×1080 (全高清)")
        print("6. 3840×2160 (4K)")
        print("7. 7680×4320 (8K)")

        # 获取用户选择（默认选项4）
        choice = input("请输入选项数字 (1-7, 默认4): ").strip()
        if choice == "":
            choice = "4"  # 设置默认选项

        if choice in size_options:
            target_width = size_options[choice]
            # 执行调整操作
            resize_image(file_path, target_width)
        else:
            print("无效选择，将使用默认尺寸1280×720")
            resize_image(file_path, size_options["4"])

if __name__ == "__main__":
    # 检查Pillow库是否安装
    try:
        from PIL import Image
    except ImportError:
        print("错误：需要安装Pillow库，请执行: pip install Pillow")
        sys.exit(1)

    main()

if __name__ == "__main__":
    compress_file()
    input("按回车键退出...")