84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
fix_commas_and_pad_multi.py
|
||
批量修复多个文件夹中含逗号的数字文件名并补齐到6位，例如:
|
||
"01,000.png" -> "001000.png"
|
||
支持扫描指定文件夹中所有以"25"开头的子文件夹
|
||
"""
|
||
|
||
import os
|
||
import argparse
|
||
|
||
VALID_EXTS = {'.png', '.exr'}
|
||
|
||
def safe_rename(src, dst, dry_run=False):
|
||
if os.path.exists(dst):
|
||
print(f"[跳过] 目标已存在，未重命名：{os.path.basename(src)} -> {os.path.basename(dst)}")
|
||
return False
|
||
print(f"[重命名] {os.path.basename(src)} -> {os.path.basename(dst)}")
|
||
if not dry_run:
|
||
os.rename(src, dst)
|
||
return True
|
||
|
||
def fix_folder(root_folder, dry_run=False):
|
||
changed = 0
|
||
for root, dirs, files in os.walk(root_folder):
|
||
for fname in files:
|
||
name, ext = os.path.splitext(fname)
|
||
if ext.lower() not in VALID_EXTS:
|
||
continue
|
||
|
||
# 若文件名中含逗号，则移除逗号
|
||
if ',' in name:
|
||
candidate = name.replace(',', '')
|
||
else:
|
||
candidate = None
|
||
|
||
if candidate is None or not candidate.isdigit():
|
||
continue
|
||
|
||
padded = candidate.zfill(6)
|
||
if padded == name:
|
||
continue
|
||
|
||
src_path = os.path.join(root, fname)
|
||
dst_path = os.path.join(root, padded + ext)
|
||
if safe_rename(src_path, dst_path, dry_run=dry_run):
|
||
changed += 1
|
||
|
||
print(f"[完成] {root_folder} 共重命名 {changed} 个文件。\n")
|
||
|
||
def find_25_folders(root_folder):
|
||
"""查找所有以'25'开头的子文件夹"""
|
||
folders = []
|
||
for item in os.listdir(root_folder):
|
||
item_path = os.path.join(root_folder, item)
|
||
if os.path.isdir(item_path) and item.startswith('25'):
|
||
folders.append(item_path)
|
||
return folders
|
||
|
||
def main():
|
||
p = argparse.ArgumentParser(
|
||
description="修复含逗号的数字文件名并补齐到6位，例如 01,000.png -> 001000.png（扫描指定文件夹中所有以25开头的子文件夹）"
|
||
)
|
||
p.add_argument("root_folder", help="包含多个以25开头的子文件夹的根目录")
|
||
p.add_argument("--dry-run", action="store_true", help="仅模拟显示将要做的重命名，不执行")
|
||
args = p.parse_args()
|
||
|
||
if not os.path.isdir(args.root_folder):
|
||
print(f"❌ 路径无效：{args.root_folder}")
|
||
return
|
||
|
||
target_folders = find_25_folders(args.root_folder)
|
||
if not target_folders:
|
||
print(f"❌ 在 {args.root_folder} 中未找到以'25'开头的子文件夹")
|
||
return
|
||
|
||
print(f"\n=== 在 {args.root_folder} 中找到 {len(target_folders)} 个以'25'开头的子文件夹 ===")
|
||
|
||
for folder in target_folders:
|
||
print(f"\n=== 处理文件夹：{folder} ===")
|
||
fix_folder(folder, dry_run=args.dry_run)
|
||
|
||
if __name__ == "__main__":
|
||
main() |