更新 main.py
This commit is contained in:
@@ -0,0 +1,432 @@
|
||||
import os
|
||||
import shutil
|
||||
import sys
|
||||
import pathlib
|
||||
import subprocess
|
||||
from io import BytesIO
|
||||
|
||||
current_mode = None
|
||||
|
||||
def force_close_handles(folder_path):
|
||||
try:
|
||||
import psutil
|
||||
closed_count = 0
|
||||
|
||||
for proc in psutil.process_iter(['pid', 'name']):
|
||||
try:
|
||||
for item in proc.open_files():
|
||||
if folder_path.lower() in item.path.lower():
|
||||
proc.kill()
|
||||
closed_count += 1
|
||||
print(f"[系统] 已关闭占用进程: {proc.info['name']} (PID: {proc.info['pid']})")
|
||||
break
|
||||
except (psutil.NoSuchProcess, psutil.AccessDenied):
|
||||
pass
|
||||
|
||||
if closed_count > 0:
|
||||
import time
|
||||
time.sleep(0.5)
|
||||
|
||||
return closed_count > 0
|
||||
except ImportError:
|
||||
return False
|
||||
|
||||
|
||||
|
||||
|
||||
def delete_system_files(folder_path):
|
||||
system_files = ['thumbs.db', 'desktop.ini', '.ds_store', 'thumbs.db:encryptable']
|
||||
deleted_count = 0
|
||||
|
||||
try:
|
||||
for root, dirs, files in os.walk(folder_path):
|
||||
for filename in files:
|
||||
if filename.lower() in system_files or filename.startswith('._'):
|
||||
file_path = os.path.join(root, filename)
|
||||
try:
|
||||
os.chmod(file_path, 0o777)
|
||||
os.remove(file_path)
|
||||
deleted_count += 1
|
||||
except:
|
||||
pass
|
||||
|
||||
if deleted_count > 0:
|
||||
print(f"[系统] 已清理 {deleted_count} 个系统隐藏文件")
|
||||
except:
|
||||
pass
|
||||
|
||||
|
||||
def get_next_log_index():
|
||||
index = 1
|
||||
while pathlib.Path(f"path_list_{index}.log").exists():
|
||||
index += 1
|
||||
return index
|
||||
|
||||
|
||||
def open_folder(path: pathlib.Path):
|
||||
if sys.platform.startswith("win"):
|
||||
os.startfile(path)
|
||||
elif sys.platform.startswith("darwin"):
|
||||
subprocess.run(["open", path])
|
||||
else:
|
||||
subprocess.run(["xdg-open", path])
|
||||
|
||||
def mode_wb():
|
||||
print("\n" + "="*60)
|
||||
print("已进入 WB 模式 - 文件夹批处理工具")
|
||||
print("功能：复制文件夹 -> 重命名文件夹和内部文件 -> 标记原文件夹")
|
||||
print("="*60)
|
||||
|
||||
while True:
|
||||
print("\n================== 新的工作任务 ==================")
|
||||
work_dir = input("请输入主文件夹路径 (工作路径, 输入 :back 返回): ").strip()
|
||||
|
||||
if work_dir == ":back":
|
||||
return
|
||||
|
||||
if not os.path.exists(work_dir):
|
||||
print(f"[错误] 工作路径 '{work_dir}' 不存在，请先创建或检查路径。")
|
||||
continue
|
||||
|
||||
print(f"[系统] 已锁定工作路径: {work_dir}")
|
||||
|
||||
process_wb_folder(work_dir)
|
||||
|
||||
print("\n------------------------------------------------------")
|
||||
while True:
|
||||
choice = input("任务结束。输入 :back 返回主菜单，或输入 :y 开启下一个工作路径: ").strip().lower()
|
||||
if choice == ":back":
|
||||
return
|
||||
elif choice == ":y":
|
||||
break
|
||||
else:
|
||||
print("输入无效，请输入 :back 或 :y")
|
||||
|
||||
|
||||
def process_wb_folder(work_dir):
|
||||
while True:
|
||||
print("\n------------------------------------------------------")
|
||||
src_path_input = input("请输入待复制的文件夹路径 (输入 :end 结束当前阶段): ").strip()
|
||||
|
||||
if src_path_input == ":end":
|
||||
return
|
||||
|
||||
if not os.path.exists(src_path_input):
|
||||
print(f"[错误] 路径 '{src_path_input}' 不存在，请重新输入。")
|
||||
continue
|
||||
|
||||
target_name = input("请输入想要重命名的名称: ").strip()
|
||||
if not target_name:
|
||||
print("[错误] 名称不能为空。")
|
||||
continue
|
||||
|
||||
dest_path = os.path.join(work_dir, target_name)
|
||||
|
||||
try:
|
||||
if os.path.exists(dest_path):
|
||||
print(f"[错误] 目标文件夹 '{dest_path}' 已经在工作路径下存在，跳过此任务。")
|
||||
continue
|
||||
|
||||
shutil.copytree(src_path_input, dest_path)
|
||||
print(f"[成功] 文件夹已复制并重命名为: {target_name}")
|
||||
|
||||
all_items = os.listdir(dest_path)
|
||||
files = []
|
||||
for f in all_items:
|
||||
full_path = os.path.join(dest_path, f)
|
||||
if os.path.isfile(full_path) and not f.startswith('.') and f.lower() not in ['thumbs.db', 'desktop.ini']:
|
||||
files.append(f)
|
||||
|
||||
files.sort()
|
||||
|
||||
count = 1
|
||||
for filename in files:
|
||||
file_ext = os.path.splitext(filename)[1]
|
||||
new_filename = f"{target_name} {count}{file_ext}"
|
||||
|
||||
old_file_path = os.path.join(dest_path, filename)
|
||||
new_file_path = os.path.join(dest_path, new_filename)
|
||||
|
||||
os.rename(old_file_path, new_file_path)
|
||||
count += 1
|
||||
|
||||
print(f"[成功] 已重命名文件夹内的 {count-1} 个文件 (从 1 开始)。")
|
||||
|
||||
clean_src_path = os.path.normpath(src_path_input)
|
||||
parent_dir = os.path.dirname(clean_src_path)
|
||||
original_folder_name = os.path.basename(clean_src_path)
|
||||
new_src_folder_name = "used_" + original_folder_name
|
||||
used_src_path = os.path.join(parent_dir, new_src_folder_name)
|
||||
|
||||
try:
|
||||
delete_system_files(clean_src_path)
|
||||
|
||||
os.rename(clean_src_path, used_src_path)
|
||||
print(f"[成功] 原文件夹已重命名为: {new_src_folder_name}")
|
||||
except PermissionError:
|
||||
print(f"[警告] 文件夹被占用，尝试强制解锁...")
|
||||
if force_close_handles(clean_src_path):
|
||||
try:
|
||||
os.rename(clean_src_path, used_src_path)
|
||||
print(f"[成功] 原文件夹已重命名为: {new_src_folder_name}")
|
||||
except Exception as e:
|
||||
print(f"[警告] 解锁后仍无法重命名: {e}")
|
||||
print(f" 原文件夹路径: {clean_src_path}")
|
||||
else:
|
||||
print(f"[警告] 无法重命名原文件夹为 used_ 开头（需要安装psutil库）")
|
||||
print(f" 安装命令: python.exe -m pip install psutil --break-system-packages")
|
||||
print(f" 原文件夹路径: {clean_src_path}")
|
||||
except FileExistsError:
|
||||
print(f"[警告] 目标名称 '{new_src_folder_name}' 已存在，跳过重命名")
|
||||
except Exception as e:
|
||||
print(f"[警告] 重命名原文件夹时出错: {e}")
|
||||
|
||||
except PermissionError:
|
||||
print(f"[错误] 权限不足，无法访问或复制文件夹")
|
||||
except Exception as e:
|
||||
print(f"[异常] 发生未知错误: {e}")
|
||||
|
||||
|
||||
def mode_path():
|
||||
print("\n" + "="*60)
|
||||
print("已进入 PATH 模式 - 文件夹批量创建工具")
|
||||
print("操作指南：")
|
||||
print("1. 输入路径后按回车 -> 加入待创建队列")
|
||||
print("2. 输入 ':y' 并回车 -> 立即批量创建队列中所有文件夹")
|
||||
print("3. 输入 ':open x' -> 打开 path_list_x.log 中的所有文件夹")
|
||||
print("4. 输入 ':back' -> 返回主菜单")
|
||||
print("="*60 + "\n")
|
||||
|
||||
path_queue = []
|
||||
|
||||
while True:
|
||||
queue_count = len(path_queue)
|
||||
prompt = f"[{queue_count} 个待创建] >>> 请输入路径: "
|
||||
raw_path = input(prompt).strip()
|
||||
|
||||
if raw_path.lower().startswith(":open"):
|
||||
parts = raw_path.split()
|
||||
if len(parts) != 2 or not parts[1].isdigit():
|
||||
print("用法错误，应为 :open x")
|
||||
continue
|
||||
|
||||
log_index = parts[1]
|
||||
log_file = pathlib.Path(f"path_list_{log_index}.log")
|
||||
|
||||
if not log_file.exists():
|
||||
print(f"日志文件不存在: {log_file}")
|
||||
continue
|
||||
|
||||
print(f"正在打开 {log_file} 中的文件夹...")
|
||||
with log_file.open("r", encoding="utf-8") as f:
|
||||
for line in f:
|
||||
folder = pathlib.Path(line.strip())
|
||||
if folder.exists():
|
||||
open_folder(folder)
|
||||
else:
|
||||
print(f"路径不存在，已跳过: {folder}")
|
||||
continue
|
||||
|
||||
if raw_path.lower() == ':y':
|
||||
if not path_queue:
|
||||
print("警告：队列为空，没有什么可创建的。")
|
||||
print("-" * 30)
|
||||
continue
|
||||
|
||||
print(f"\n开始执行批量创建，共 {len(path_queue)} 个任务...")
|
||||
success_count = 0
|
||||
created_paths = []
|
||||
|
||||
for p_str in path_queue:
|
||||
target_path = pathlib.Path(p_str)
|
||||
try:
|
||||
target_path.mkdir(parents=True, exist_ok=True)
|
||||
print(f" [OK] Created: {target_path}")
|
||||
success_count += 1
|
||||
created_paths.append(str(target_path))
|
||||
except Exception as e:
|
||||
print(f" [ERR] Failed: {target_path} | 原因: {e}")
|
||||
|
||||
print(f"\n执行完毕。成功: {success_count} / 总计: {len(path_queue)}")
|
||||
|
||||
log_index = get_next_log_index()
|
||||
log_file = pathlib.Path(f"path_list_{log_index}.log")
|
||||
with log_file.open("w", encoding="utf-8") as f:
|
||||
for p in created_paths:
|
||||
f.write(p + "\n")
|
||||
print(f"已生成日志文件: {log_file}")
|
||||
|
||||
print("-" * 30)
|
||||
path_queue.clear()
|
||||
continue
|
||||
|
||||
if raw_path.lower() == ':back':
|
||||
if path_queue:
|
||||
confirm = input(f"队列中还有 {len(path_queue)} 个路径未创建，确定要返回吗？(y/n): ")
|
||||
if confirm.lower() != 'y':
|
||||
continue
|
||||
return
|
||||
|
||||
if not raw_path:
|
||||
continue
|
||||
|
||||
path_queue.append(raw_path)
|
||||
print(f"已加入队列: '{raw_path}'")
|
||||
print("-" * 20)
|
||||
|
||||
|
||||
def mode_photo():
|
||||
print("\n" + "="*60)
|
||||
print("已进入 PHOTO 模式 - 图片左右镜像工具")
|
||||
print("功能：批量将文件夹中的所有图片进行左右镜像翻转")
|
||||
print("支持格式：.jpg, .jpeg, .png, .bmp, .gif, .webp, .tif, .tiff")
|
||||
print("操作指南：")
|
||||
print("1. 输入文件夹路径后按回车 -> 加入待处理队列")
|
||||
print("2. 输入 ':y' 并回车 -> 批量处理队列中所有文件夹的图片")
|
||||
print("3. 输入 ':back' -> 返回主菜单")
|
||||
print("="*60 + "\n")
|
||||
|
||||
try:
|
||||
from PIL import Image
|
||||
except ImportError:
|
||||
print("[错误] 未检测到 Pillow 库！")
|
||||
print("请先安装 Pillow：")
|
||||
print(" python.exe -m pip install pillow")
|
||||
print("或者手动下载 pillow 的 .whl 文件后安装")
|
||||
input("\n按回车返回主菜单...")
|
||||
return
|
||||
|
||||
folder_queue = []
|
||||
supported_extensions = {'.jpg', '.jpeg', '.png', '.bmp', '.gif', '.webp', '.tif', '.tiff'}
|
||||
|
||||
while True:
|
||||
queue_count = len(folder_queue)
|
||||
prompt = f"[{queue_count} 个文件夹待处理] >>> 请输入文件夹路径: "
|
||||
folder_path = input(prompt).strip()
|
||||
|
||||
if folder_path.lower() == ':y':
|
||||
if not folder_queue:
|
||||
print("警告：队列为空，没有什么可处理的。")
|
||||
print("-" * 30)
|
||||
continue
|
||||
|
||||
print(f"\n开始执行图片镜像处理，共 {len(folder_queue)} 个文件夹...")
|
||||
|
||||
for folder in folder_queue:
|
||||
print(f"\n处理文件夹: {folder}")
|
||||
if not os.path.exists(folder):
|
||||
print(f" [警告] 文件夹不存在，跳过")
|
||||
continue
|
||||
|
||||
if not os.path.isdir(folder):
|
||||
print(f" [警告] 路径不是文件夹，跳过")
|
||||
continue
|
||||
|
||||
image_count = 0
|
||||
error_count = 0
|
||||
|
||||
for filename in os.listdir(folder):
|
||||
file_path = os.path.join(folder, filename)
|
||||
|
||||
if not os.path.isfile(file_path):
|
||||
continue
|
||||
|
||||
_, ext = os.path.splitext(filename)
|
||||
if ext.lower() not in supported_extensions:
|
||||
continue
|
||||
|
||||
try:
|
||||
if ext.lower() in ['.tif', '.tiff']:
|
||||
with open(file_path, 'rb') as f:
|
||||
img_data = f.read()
|
||||
|
||||
img = Image.open(BytesIO(img_data))
|
||||
|
||||
flipped_img = img.transpose(Image.FLIP_LEFT_RIGHT)
|
||||
|
||||
output = BytesIO()
|
||||
flipped_img.save(output, format='TIFF')
|
||||
|
||||
with open(file_path, 'wb') as f:
|
||||
f.write(output.getvalue())
|
||||
else:
|
||||
img = Image.open(file_path)
|
||||
flipped_img = img.transpose(Image.FLIP_LEFT_RIGHT)
|
||||
flipped_img.save(file_path)
|
||||
|
||||
print(f" [OK] {filename}")
|
||||
image_count += 1
|
||||
|
||||
except Exception as e:
|
||||
print(f" [ERR] {filename} | 原因: {e}")
|
||||
error_count += 1
|
||||
|
||||
print(f" 文件夹处理完毕。成功: {image_count}, 失败: {error_count}")
|
||||
|
||||
print(f"\n全部执行完毕！共处理 {len(folder_queue)} 个文件夹。")
|
||||
print("-" * 30)
|
||||
folder_queue.clear()
|
||||
continue
|
||||
|
||||
if folder_path.lower() == ':back':
|
||||
if folder_queue:
|
||||
confirm = input(f"队列中还有 {len(folder_queue)} 个文件夹未处理，确定要返回吗？(y/n): ")
|
||||
if confirm.lower() != 'y':
|
||||
continue
|
||||
return
|
||||
|
||||
if not folder_path:
|
||||
continue
|
||||
|
||||
folder_queue.append(folder_path)
|
||||
print(f"已加入队列: '{folder_path}'")
|
||||
print("-" * 20)
|
||||
|
||||
|
||||
def print_main_menu():
|
||||
print("\n" + "="*60)
|
||||
print(" 集成工具箱 by FlandreScarle7")
|
||||
print("="*60)
|
||||
print("可用模式：")
|
||||
print(" :mode wb - 文件夹批处理工具 (复制、重命名、标记)")
|
||||
print(" :mode path - 文件夹批量创建工具")
|
||||
print(" :mode photo - 图片左右镜像工具")
|
||||
print(" :exit - 退出程序")
|
||||
print("="*60)
|
||||
|
||||
|
||||
def main():
|
||||
print_main_menu()
|
||||
|
||||
while True:
|
||||
command = input("\n>>> 请输入命令: ").strip().lower()
|
||||
|
||||
if command == ":exit":
|
||||
print("[系统] 程序已退出。")
|
||||
sys.exit(0)
|
||||
|
||||
elif command == ":mode wb":
|
||||
mode_wb()
|
||||
print_main_menu()
|
||||
|
||||
elif command == ":mode path":
|
||||
mode_path()
|
||||
print_main_menu()
|
||||
|
||||
elif command == ":mode photo":
|
||||
mode_photo()
|
||||
print_main_menu()
|
||||
|
||||
elif command == ":help":
|
||||
print_main_menu()
|
||||
|
||||
else:
|
||||
print("[错误] 未知命令。请输入 :mode wb / :mode path / :mode photo / :exit")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
main()
|
||||
except KeyboardInterrupt:
|
||||
print("\n\n[系统] 程序被强制中断。")
|
||||
sys.exit(0)
|
||||
Reference in New Issue
Block a user