import os
from PIL import Image
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, 
                            QHBoxLayout, QLabel, QPushButton, QListWidget, 
                            QLineEdit, QMessageBox, QFileDialog)

def init_ui(self):
    # 主窗口設(shè)置
    self.setWindowTitle("圖片批量修改尺寸工具")
    self.setGeometry(100, 100, 600, 500)
    
    # 主布局
    main_layout = QVBoxLayout()
    
    # 文件夾選擇部分
    folder_layout = QHBoxLayout()
    self.label_folder_path = QLabel("請選擇圖片目錄：")
    self.button_choose_folder = QPushButton("選擇目錄")
    folder_layout.addWidget(self.label_folder_path)
    folder_layout.addWidget(self.button_choose_folder)
    
    # 圖片列表
    self.image_list = QListWidget()
    
    # 尺寸輸入部分
    size_layout = QHBoxLayout()
    # ...寬度和高度輸入框設(shè)置...
    
    # 操作按鈕
    self.button_resize = QPushButton("批量修改尺寸")
    
    # 將所有部件添加到主布局
    main_layout.addLayout(folder_layout)
    main_layout.addWidget(self.image_list)
    main_layout.addLayout(size_layout)
    main_layout.addWidget(self.button_resize)

def resize_images(self):
    # 獲取用戶輸入的尺寸
    new_width = int(self.entry_width.text())
    new_height = int(self.entry_height.text())
    
    # 遍歷處理每張圖片
    for i in range(self.image_list.count()):
        img_name = self.image_list.item(i).text()
        img_path = os.path.join(self.image_folder_path, img_name)
        
        # 使用Pillow打開并調(diào)整圖片
        with Image.open(img_path) as img:
            resized_img = img.resize((new_width, new_height), Image.Resampling.BICUBIC)
            
            # 根據(jù)用戶選擇保存到備份文件夾或覆蓋原文件
            if self.backup_mode:
                backup_path = os.path.join(self.backup_folder_path, img_name)
                resized_img.save(backup_path)
            else:
                resized_img.save(img_path)

try:
    # 圖片處理代碼
except Exception as e:
    QMessageBox.critical(self, "錯誤", f"處理失敗: {str(e)}")

import os
from PIL import Image
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout,
                             QLabel, QPushButton, QListWidget, QLineEdit, QScrollArea,
                             QMessageBox, QFileDialog)
from PyQt5.QtCore import Qt


class ImageResizerApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.image_folder_path = ""
        self.backup_folder_path = ""

        self.init_ui()

    def init_ui(self):
        self.setWindowTitle("圖片批量修改尺寸工具")
        self.setGeometry(100, 100, 600, 500)

        # 主部件
        main_widget = QWidget()
        self.setCentralWidget(main_widget)

        # 主布局
        main_layout = QVBoxLayout()
        main_widget.setLayout(main_layout)

        # 文件夾選擇部分
        folder_layout = QHBoxLayout()
        self.label_folder_path = QLabel("請選擇圖片目錄：")
        self.button_choose_folder = QPushButton("選擇目錄")
        self.button_choose_folder.clicked.connect(self.open_directory)

        folder_layout.addWidget(self.label_folder_path)
        folder_layout.addWidget(self.button_choose_folder)
        main_layout.addLayout(folder_layout)

        # 圖片列表
        self.image_list = QListWidget()
        main_layout.addWidget(self.image_list)

        # 尺寸輸入部分
        size_layout = QHBoxLayout()

        width_layout = QVBoxLayout()
        self.label_width = QLabel("新寬度:")
        self.entry_width = QLineEdit()
        width_layout.addWidget(self.label_width)
        width_layout.addWidget(self.entry_width)

        height_layout = QVBoxLayout()
        self.label_height = QLabel("新高度:")
        self.entry_height = QLineEdit()
        height_layout.addWidget(self.label_height)
        height_layout.addWidget(self.entry_height)

        size_layout.addLayout(width_layout)
        size_layout.addLayout(height_layout)
        main_layout.addLayout(size_layout)

        # 操作按鈕
        self.button_resize = QPushButton("批量修改尺寸")
        self.button_resize.clicked.connect(self.resize_images)
        main_layout.addWidget(self.button_resize)

    def open_directory(self):
        """打開并選擇一個目錄作為圖片數(shù)據(jù)源"""
        folder = QFileDialog.getExistingDirectory(self, "選擇圖片目錄")
        if folder:
            self.image_folder_path = folder
            self.label_folder_path.setText(f"已選擇目錄: {folder}")
            self.update_image_list()

    def update_image_list(self):
        """更新并顯示圖片列表"""
        self.image_list.clear()
        if self.image_folder_path:
            for im in os.listdir(self.image_folder_path):
                if im.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')):
                    self.image_list.addItem(im)

    def create_backup_folder(self):
        """創(chuàng)建備份文件夾"""
        self.backup_folder_path = os.path.join(self.image_folder_path, "resized")
        if not os.path.exists(self.backup_folder_path):
            os.makedirs(self.backup_folder_path)

    def resize_images(self):
        """批量修改圖片尺寸"""
        if self.image_list.count() == 0:
            QMessageBox.warning(self, "警告", "沒有圖片可供調(diào)整尺寸！")
            return

        reply = QMessageBox.question(self, "提示", "是否創(chuàng)建備份文件夾？",
                                     QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
        should_create_backup = reply == QMessageBox.Yes

        if should_create_backup:
            self.create_backup_folder()

        try:
            new_width = int(self.entry_width.text())
            new_height = int(self.entry_height.text())

            if new_width <= 0 or new_height <= 0:
                raise ValueError("寬度和高度必須大于0")

            for i in range(self.image_list.count()):
                im = self.image_list.item(i).text()
                img_path = os.path.join(self.image_folder_path, im)

                with Image.open(img_path) as img:
                    resized_img = img.resize((new_width, new_height), Image.Resampling.BICUBIC)

                    if should_create_backup:
                        backup_img_path = os.path.join(self.backup_folder_path, im)
                        resized_img.save(backup_img_path)
                        print(f"已修改圖片 {im} 的尺寸為 {new_width}x{new_height} 并保存到備份文件夾。")
                    else:
                        resized_img.save(img_path)  # 注意：這會覆蓋原文件
                        print(f"已修改圖片 {im} 的尺寸為 {new_width}x{new_height}")

            QMessageBox.information(self, "完成", "所有圖片已按指定尺寸修改完畢！")
        except ValueError as e:
            QMessageBox.critical(self, "錯誤", str(e))
        except IOError:
            QMessageBox.critical(self, "錯誤", "無法打開圖片文件，請檢查文件格式。")
        except Exception as e:
            QMessageBox.critical(self, "錯誤", f"未知錯誤: {str(e)}")


if __name__ == "__main__":
    app = QApplication([])
    window = ImageResizerApp()
    window.show()
    app.exec_()