Skip to content

Commit

Permalink
2.0
Browse files Browse the repository at this point in the history 
新增单文件处理
  • Loading branch information
@riceshowerX
riceshowerX committed Jul 26, 2024
1 parent 5c63a82 commit a8b5ae7
Showing 1 changed file with 159 additions and 75 deletions.
234 changes: 159 additions & 75 deletions gui.py
View file
Original file line number Diff line number Diff line change
@@ -1,9 +1,11 @@
# gui.py
from PyQt6.QtWidgets import QApplication, QMainWindow, QFileDialog, QVBoxLayout, QPushButton, QLineEdit, QLabel, QWidget, QMessageBox, QProgressBar
import os
from PyQt6.QtWidgets import QApplication, QMainWindow, QFileDialog, QVBoxLayout, QPushButton, QLineEdit, QLabel, QWidget, QMessageBox, QProgressBar, QCheckBox
from PyQt6.QtCore import Qt
from utils import image_processor
import sys
from PyQt6.QtGui import QIcon # 导入 QIcon 类
from PIL import Image # 导入 Image 类

class ImageProcessingApp(QMainWindow):
def __init__(self):
Expand All @@ -16,20 +18,27 @@ def __init__(self):
app_icon = QIcon("snapforge.ico") # 创建 QIcon 对象
self.setWindowIcon(app_icon) # 设置窗口图标

self.resize(400, 350)
self.resize(400, 380) # 稍微增加窗口高度

self.source_directory = None
self.target_directory = None
self.single_file_mode = False
self.source_file = None # 用于存储单个文件路径

self.init_ui()

def init_ui(self):
layout = QVBoxLayout()

# 选择文件夹
self.source_label = QLabel("选择原始图片文件夹:")
self.source_button = QPushButton("选择文件夹")
self.source_button.clicked.connect(self.select_source_directory)
# 单个文件处理模式选择
self.single_file_checkbox = QCheckBox("单个文件处理")
self.single_file_checkbox.stateChanged.connect(self.toggle_file_mode)
layout.addWidget(self.single_file_checkbox)

# 选择文件夹/文件
self.source_label = QLabel("选择原始图片文件夹/文件:")
self.source_button = QPushButton("选择")
self.source_button.clicked.connect(self.select_source)

self.target_label = QLabel("选择保存图片文件夹:")
self.target_button = QPushButton("选择文件夹")
Expand Down Expand Up @@ -84,84 +93,152 @@ def init_ui(self):
container.setLayout(layout)
self.setCentralWidget(container)

def select_source_directory(self):
self.source_directory = QFileDialog.getExistingDirectory(self, "选择原始图片文件夹")
def select_source(self):
if self.single_file_mode:
self.source_file, _ = QFileDialog.getOpenFileName(self, "选择图片文件", filter="Image Files (*.png *.jpg *.jpeg *.bmp *.tiff)")
if self.source_file:
self.source_label.setText(f"已选择文件: {self.source_file}")
else:
self.source_directory = QFileDialog.getExistingDirectory(self, "选择原始图片文件夹")

def select_target_directory(self):
self.target_directory = QFileDialog.getExistingDirectory(self, "选择保存图片文件夹")

def rename_files(self):
if not self.source_directory or not self.target_directory:
QMessageBox.warning(self, "警告", "请先选择原始和保存图片文件夹！")
return

prefix = self.prefix_entry.text()
try:
start_number = int(self.start_number_entry.text())
except ValueError:
QMessageBox.warning(self, "警告", "起始编号必须是一个整数！")
return

try:
renamed_count = image_processor.batch_rename_files(
self.source_directory,
self.target_directory,
prefix,
start_number,
self.update_progress_bar
)
QMessageBox.information(self, "结果", f"重命名完成！共重命名了 {renamed_count} 个文件。")
except Exception as e:
# 处理其他潜在的异常
QMessageBox.warning(self, "错误", f"发生错误: {e}")

if self.single_file_mode:
if not self.source_file or not self.target_directory:
QMessageBox.warning(self, "警告", "请先选择文件和目标文件夹！")
return

prefix = self.prefix_entry.text()
try:
start_number = int(self.start_number_entry.text())
except ValueError:
QMessageBox.warning(self, "警告", "起始编号必须是一个整数！")
return

try:
file_name, file_ext = os.path.splitext(os.path.basename(self.source_file))
new_file_name = f"{prefix}{start_number}{file_ext}"
new_file_path = os.path.join(self.target_directory, new_file_name)
os.rename(self.source_file, new_file_path)
QMessageBox.information(self, "结果", f"重命名完成！")
except Exception as e:
QMessageBox.warning(self, "错误", f"发生错误: {e}")

else: # 批量处理模式
if not self.source_directory or not self.target_directory:
QMessageBox.warning(self, "警告", "请先选择原始和保存图片文件夹！")
return

prefix = self.prefix_entry.text()
try:
start_number = int(self.start_number_entry.text())
except ValueError:
QMessageBox.warning(self, "警告", "起始编号必须是一个整数！")
return

try:
renamed_count = image_processor.batch_rename_files(
self.source_directory,
self.target_directory,
prefix,
start_number,
self.update_progress_bar
)
QMessageBox.information(self, "结果", f"重命名完成！共重命名了 {renamed_count} 个文件。")
except Exception as e:
# 处理其他潜在的异常
QMessageBox.warning(self, "错误", f"发生错误: {e}")

def convert_files(self):
if not self.source_directory or not self.target_directory:
QMessageBox.warning(self, "警告", "请先选择原始和保存图片文件夹！")
return

target_format = self.format_entry.text().lower()
if not target_format:
QMessageBox.warning(self, "警告", "目标格式不能为空！")
return

try:
converted_count = image_processor.batch_convert_images(
self.source_directory,
self.target_directory,
target_format,
self.update_progress_bar
)
QMessageBox.information(self, "结果", f"格式转换完成！共转换了 {converted_count} 个文件。")
except Exception as e:
# 处理其他潜在的异常
QMessageBox.warning(self, "错误", f"发生错误: {e}")
if self.single_file_mode:
if not self.source_file or not self.target_directory:
QMessageBox.warning(self, "警告", "请先选择文件和目标文件夹！")
return

target_format = self.format_entry.text().lower()
if not target_format:
QMessageBox.warning(self, "警告", "目标格式不能为空！")
return

try:
file_name, _ = os.path.splitext(os.path.basename(self.source_file))
new_file_name = f"{file_name}.{target_format}"
new_file_path = os.path.join(self.target_directory, new_file_name)
img = Image.open(self.source_file)
img.save(new_file_path, target_format.upper())
QMessageBox.information(self, "结果", f"转换完成！")
except Exception as e:
QMessageBox.warning(self, "错误", f"发生错误: {e}")

else: # 批量处理模式
if not self.source_directory or not self.target_directory:
QMessageBox.warning(self, "警告", "请先选择原始和保存图片文件夹！")
return

target_format = self.format_entry.text().lower()
if not target_format:
QMessageBox.warning(self, "警告", "目标格式不能为空！")
return

try:
converted_count = image_processor.batch_convert_images(
self.source_directory,
self.target_directory,
target_format,
self.update_progress_bar
)
QMessageBox.information(self, "结果", f"格式转换完成！共转换了 {converted_count} 个文件。")
except Exception as e:
# 处理其他潜在的异常
QMessageBox.warning(self, "错误", f"发生错误: {e}")

def compress_files(self):
if not self.source_directory or not self.target_directory:
QMessageBox.warning(self, "警告", "请先选择原始和保存图片文件夹！")
return

try:
quality = int(self.quality_entry.text())
if quality < 1 or quality > 100:
raise ValueError
except ValueError:
QMessageBox.warning(self, "警告", "压缩质量必须是1到100之间的整数！")
return

try:
compressed_count = image_processor.batch_compress_images(
self.source_directory,
self.target_directory,
quality,
self.update_progress_bar
)
QMessageBox.information(self, "结果", f"压缩完成！共压缩了 {compressed_count} 个文件。")
except Exception as e:
# 处理其他潜在的异常
QMessageBox.warning(self, "错误", f"发生错误: {e}")
if self.single_file_mode:
if not self.source_file or not self.target_directory:
QMessageBox.warning(self, "警告", "请先选择文件和目标文件夹！")
return

try:
quality = int(self.quality_entry.text())
if quality < 1 or quality > 100:
raise ValueError
except ValueError:
QMessageBox.warning(self, "警告", "压缩质量必须是1到100之间的整数！")
return

try:
new_file_path = os.path.join(self.target_directory, os.path.basename(self.source_file))
img = Image.open(self.source_file)
img.save(new_file_path, optimize=True, quality=quality)
QMessageBox.information(self, "结果", f"压缩完成！")
except Exception as e:
QMessageBox.warning(self, "错误", f"发生错误: {e}")
else: # 批量处理模式
if not self.source_directory or not self.target_directory:
QMessageBox.warning(self, "警告", "请先选择原始和保存图片文件夹！")
return

try:
quality = int(self.quality_entry.text())
if quality < 1 or quality > 100:
raise ValueError
except ValueError:
QMessageBox.warning(self, "警告", "压缩质量必须是1到100之间的整数！")
return

try:
compressed_count = image_processor.batch_compress_images(
self.source_directory,
self.target_directory,
quality,
self.update_progress_bar
)
QMessageBox.information(self, "结果", f"压缩完成！共压缩了 {compressed_count} 个文件。")
except Exception as e:
# 处理其他潜在的异常
QMessageBox.warning(self, "错误", f"发生错误: {e}")

def update_progress_bar(self, current, total):
"""更新进度条。
Expand All @@ -172,6 +249,13 @@ def update_progress_bar(self, current, total):
"""
self.progress_bar.setValue(int(current / total * 100))

def toggle_file_mode(self, state):
self.single_file_mode = state == Qt.Checked
if self.single_file_mode:
self.source_label.setText("选择原始图片文件:")
else:
self.source_label.setText("选择原始图片文件夹:")

if __name__ == "__main__":
app = QApplication(sys.argv)
window = ImageProcessingApp()
Expand Down

0 comments on commit a8b5ae7

Please sign in to comment.