232 lines
9.3 KiB
Python
232 lines
9.3 KiB
Python
import os
|
||
import zipfile
|
||
import shutil
|
||
import tempfile
|
||
import glob
|
||
import subprocess
|
||
from pathlib import Path
|
||
import threading
|
||
from concurrent.futures import ThreadPoolExecutor, as_completed
|
||
import time
|
||
|
||
# 导入可能需要安装的库
|
||
import tqdm
|
||
import patoolib
|
||
|
||
# 全局锁用于线程安全的输出和文件操作
|
||
print_lock = threading.Lock()
|
||
file_operation_lock = threading.Lock()
|
||
|
||
def safe_print(*args, **kwargs):
|
||
"""线程安全的打印函数"""
|
||
with print_lock:
|
||
print(*args, **kwargs)
|
||
|
||
def safe_file_operation(operation, *args, **kwargs):
|
||
"""线程安全的文件操作函数"""
|
||
with file_operation_lock:
|
||
return operation(*args, **kwargs)
|
||
|
||
def extract_archive(archive_path, extract_dir):
|
||
"""解压缩文件到指定目录"""
|
||
safe_print("正在解压文件...")
|
||
|
||
if archive_path.lower().endswith('.zip'):
|
||
# 使用zipfile处理zip文件，可以显示进度条
|
||
with zipfile.ZipFile(archive_path, 'r') as zip_ref:
|
||
total = len(zip_ref.namelist())
|
||
for file in tqdm.tqdm(zip_ref.namelist(), total=total, desc="解压进度"):
|
||
zip_ref.extract(file, extract_dir)
|
||
else:
|
||
# 使用patoolib处理其他类型的压缩文件，包括RAR
|
||
patoolib.extract_archive(archive_path, outdir=extract_dir)
|
||
safe_print("解压完成")
|
||
|
||
# 如果解压后只有一个文件夹，将其内容移到extract_dir
|
||
items = os.listdir(extract_dir)
|
||
if len(items) == 1 and os.path.isdir(os.path.join(extract_dir, items[0])):
|
||
subfolder = os.path.join(extract_dir, items[0])
|
||
# 将子文件夹中的所有内容移到extract_dir
|
||
for item in os.listdir(subfolder):
|
||
shutil.move(os.path.join(subfolder, item), extract_dir)
|
||
# 删除空文件夹
|
||
os.rmdir(subfolder)
|
||
|
||
def process_files(directory):
|
||
"""处理目录中的文件：删除.bat文件，重命名.jar文件"""
|
||
safe_print("正在处理文件...")
|
||
|
||
# 查找所有.bat文件并删除
|
||
bat_files = []
|
||
for root, _, files in os.walk(directory):
|
||
for file in files:
|
||
if file.lower().endswith('.bat'):
|
||
bat_path = os.path.join(root, file)
|
||
bat_files.append(bat_path)
|
||
|
||
# 使用线程池并行删除.bat文件
|
||
if bat_files:
|
||
def safe_remove_file(file_path):
|
||
"""安全删除文件"""
|
||
try:
|
||
safe_file_operation(os.remove, file_path)
|
||
return True
|
||
except Exception as e:
|
||
safe_print(f"删除文件 {file_path} 时出错: {e}")
|
||
return False
|
||
|
||
with ThreadPoolExecutor(max_workers=4) as executor:
|
||
futures = [executor.submit(safe_remove_file, bat_file) for bat_file in bat_files]
|
||
successful_deletions = 0
|
||
for future in tqdm.tqdm(as_completed(futures), total=len(futures), desc="删除.bat文件"):
|
||
if future.result():
|
||
successful_deletions += 1
|
||
safe_print(f"成功删除 {successful_deletions}/{len(bat_files)} 个.bat文件")
|
||
else:
|
||
safe_print("未找到需要删除的.bat文件")
|
||
|
||
# 查找所有.jar文件
|
||
jar_files = []
|
||
for root, _, files in os.walk(directory):
|
||
for file in files:
|
||
if file.lower().endswith('.jar'):
|
||
jar_path = os.path.join(root, file)
|
||
jar_files.append(jar_path)
|
||
|
||
# 处理.jar文件
|
||
if jar_files:
|
||
for jar_file in tqdm.tqdm(jar_files, desc="处理.jar文件"):
|
||
dir_path = os.path.dirname(jar_file)
|
||
|
||
# 检查是否存在fabric-server-launch.jar和server.jar
|
||
if os.path.basename(jar_file).lower() == 'fabric-server-launch.jar':
|
||
server_jar = os.path.join(dir_path, 'server.jar')
|
||
if os.path.exists(server_jar):
|
||
# 将server.jar重命名为ser.jar
|
||
ser_jar = os.path.join(dir_path, 'ser.jar')
|
||
safe_file_operation(shutil.move, server_jar, ser_jar)
|
||
|
||
# 将fabric-server-launch.jar重命名为server.jar
|
||
safe_file_operation(shutil.move, jar_file, os.path.join(dir_path, 'server.jar'))
|
||
|
||
# 如果是普通的jar文件且不是server.jar或ser.jar，则重命名为server.jar
|
||
elif os.path.basename(jar_file).lower() != 'server.jar' and os.path.basename(jar_file).lower() != 'ser.jar':
|
||
# 检查目录中是否已存在server.jar
|
||
server_jar = os.path.join(dir_path, 'server.jar')
|
||
if not os.path.exists(server_jar):
|
||
safe_file_operation(shutil.move, jar_file, server_jar)
|
||
else:
|
||
safe_print("未找到需要处理的.jar文件")
|
||
|
||
def compress_directory(source_dir, output_path):
|
||
"""将目录压缩为zip文件，确保解压后直接是文件而不是文件夹"""
|
||
safe_print("正在压缩文件...")
|
||
|
||
with zipfile.ZipFile(output_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
|
||
# 获取要压缩的文件列表
|
||
file_paths = []
|
||
for root, dirs, files in os.walk(source_dir):
|
||
for file in files:
|
||
file_path = os.path.join(root, file)
|
||
file_paths.append(file_path)
|
||
|
||
# 显示压缩进度
|
||
for file in tqdm.tqdm(file_paths, desc="压缩进度"):
|
||
# 计算相对路径，确保解压后不会有额外的目录层级
|
||
# 获取文件相对于source_dir的路径
|
||
rel_path = os.path.relpath(file, source_dir)
|
||
|
||
# 如果文件在子文件夹中，我们需要保留子文件夹结构，但去掉最外层文件夹
|
||
# 例如，如果文件在 temp_dir/folder1/file.txt，我们希望在zip中存储为 folder1/file.txt
|
||
# 如果文件在 temp_dir/file.txt，我们希望在zip中存储为 file.txt
|
||
zipf.write(file, rel_path)
|
||
|
||
def process_single_archive(archive_path, current_dir, index, total):
|
||
"""处理单个压缩文件的函数"""
|
||
safe_print(f"\n[线程 {threading.current_thread().name}] 处理第 {index+1}/{total} 个压缩文件: {os.path.basename(archive_path)}")
|
||
|
||
try:
|
||
# 创建临时目录用于解压
|
||
with tempfile.TemporaryDirectory() as temp_dir:
|
||
# 解压文件
|
||
try:
|
||
extract_archive(archive_path, temp_dir)
|
||
except Exception as e:
|
||
safe_print(f"解压失败: {e}")
|
||
return False
|
||
|
||
# 处理文件
|
||
process_files(temp_dir)
|
||
|
||
# 确保successzip目录存在
|
||
success_dir = os.path.join(current_dir, 'successzip')
|
||
if not os.path.exists(success_dir):
|
||
safe_file_operation(os.makedirs, success_dir, exist_ok=True)
|
||
|
||
# 创建新的压缩文件名，使用"cjsyun-"前缀，保存到successzip文件夹
|
||
processed_archive = os.path.join(success_dir, f"cjsyun-{os.path.basename(archive_path)}")
|
||
if processed_archive.lower().endswith('.rar'):
|
||
processed_archive = processed_archive[:-4] + '.zip' # 将RAR转换为ZIP
|
||
|
||
# 压缩处理后的文件
|
||
compress_directory(temp_dir, processed_archive)
|
||
|
||
safe_print(f"[线程 {threading.current_thread().name}] 处理完成: {os.path.basename(processed_archive)}")
|
||
return True
|
||
except Exception as e:
|
||
safe_print(f"处理文件时出错: {e}")
|
||
return False
|
||
|
||
def main():
|
||
# 获取当前目录下的所有压缩文件
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
archives = []
|
||
|
||
for ext in ['*.zip', '*.rar']:
|
||
archives.extend(glob.glob(os.path.join(current_dir, ext)))
|
||
|
||
if not archives:
|
||
safe_print("当前目录下没有找到压缩文件")
|
||
return
|
||
|
||
safe_print(f"找到 {len(archives)} 个压缩文件")
|
||
|
||
# 确定线程数量，最多使用4个线程，但不超过文件数量
|
||
max_workers = min(4, len(archives))
|
||
safe_print(f"使用 {max_workers} 个线程并行处理")
|
||
|
||
start_time = time.time()
|
||
|
||
# 使用线程池并行处理压缩文件
|
||
with ThreadPoolExecutor(max_workers=max_workers, thread_name_prefix="Archive") as executor:
|
||
# 提交所有任务
|
||
futures = []
|
||
for i, archive_path in enumerate(archives):
|
||
future = executor.submit(process_single_archive, archive_path, current_dir, i, len(archives))
|
||
futures.append(future)
|
||
|
||
# 等待所有任务完成并收集结果
|
||
successful = 0
|
||
failed = 0
|
||
for future in as_completed(futures):
|
||
try:
|
||
result = future.result()
|
||
if result:
|
||
successful += 1
|
||
else:
|
||
failed += 1
|
||
except Exception as e:
|
||
safe_print(f"任务执行出错: {e}")
|
||
failed += 1
|
||
|
||
end_time = time.time()
|
||
total_time = end_time - start_time
|
||
|
||
safe_print(f"\n处理完成！")
|
||
safe_print(f"成功处理: {successful} 个文件")
|
||
safe_print(f"处理失败: {failed} 个文件")
|
||
safe_print(f"总耗时: {total_time:.2f} 秒")
|
||
safe_print(f"平均每个文件: {total_time/len(archives):.2f} 秒")
|
||
|
||
if __name__ == "__main__":
|
||
main() |