フォルダ内のファイルを拡張子別に自動仕分けするスクリプト

import sys
import json
import os
import shutil
from datetime import datetime

with open(sys.argv[1], encoding="utf-8") as f:
    inputs = json.load(f)

target_dir = inputs["対象フォルダ"]
include_sub = inputs["サブフォルダも含める"] == "true"
file_mode = inputs["元ファイルの扱い"]

CATEGORY_MAP = {
    "画像":       {".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".tif", ".webp", ".svg", ".ico", ".heic", ".raw"},
    "PDF":        {".pdf"},
    "Excel":      {".xlsx", ".xls", ".xlsm", ".xlsb", ".csv"},
    "Word":       {".docx", ".doc", ".dotx"},
    "PowerPoint": {".pptx", ".ppt", ".potx"},
    "動画":       {".mp4", ".mov", ".avi", ".mkv", ".wmv", ".flv", ".webm", ".m4v"},
    "音声":       {".mp3", ".wav", ".aac", ".flac", ".m4a", ".ogg", ".wma"},
    "テキスト":   {".txt", ".md", ".log"},
    "圧縮":       {".zip", ".rar", ".7z", ".tar", ".gz", ".lzh"},
}

SKIP_DIRS = set(CATEGORY_MAP.keys()) | {"その他", "サマリー"}

def get_category(ext):
    ext = ext.lower()
    for category, exts in CATEGORY_MAP.items():
        if ext in exts:
            return category
    return "その他"

def resolve_dest(dest_dir, filename):
    dest_path = os.path.join(dest_dir, filename)
    if not os.path.exists(dest_path):
        return dest_path
    name, ext = os.path.splitext(filename)
    ts = datetime.now().strftime("%Y%m%d_%H%M%S")
    return os.path.join(dest_dir, f"{name}_{ts}{ext}")

def collect_files(base_dir, include_sub):
    result = []
    base_abs = os.path.abspath(base_dir)
    if include_sub:
        for root, dirs, files in os.walk(base_dir):
            root_abs = os.path.abspath(root)
            if root_abs == base_abs:
                dirs[:] = [d for d in dirs if d not in SKIP_DIRS]
            for fname in files:
                result.append(os.path.join(root, fname))
    else:
        for fname in os.listdir(base_dir):
            fpath = os.path.join(base_dir, fname)
            if os.path.isfile(fpath):
                result.append(fpath)
    return result

try:
    print(f"対象フォルダ: {target_dir}")
    print(f"サブフォルダ含める: {'はい' if include_sub else 'いいえ'}")
    print(f"元ファイル: {file_mode}")
    print("---")

    files = collect_files(target_dir, include_sub)
    total = len(files)
    print(f"ファイル数: {total} 件")

    summary = {}
    action_log = []

    for i, src_path in enumerate(files):
        fname = os.path.basename(src_path)
        _, ext = os.path.splitext(fname)
        category = get_category(ext)

        dest_dir = os.path.join(target_dir, category)
        os.makedirs(dest_dir, exist_ok=True)

        dest_path = resolve_dest(dest_dir, fname)

        if file_mode == "移動する":
            shutil.move(src_path, dest_path)
            action = "移動"
        else:
            shutil.copy2(src_path, dest_path)
            action = "コピー"

        summary.setdefault(category, []).append(fname)
        action_log.append((src_path, category, dest_path))

        print(f"  [{i+1}/{total}] {action}: {fname} → {category}/")

    # サマリーTXT生成（カテゴリ別集計 + 詳細ログ表形式）
    ts_now = datetime.now().strftime("%Y%m%d_%H%M%S")
    summary_dir = os.path.join(target_dir, "サマリー")
    os.makedirs(summary_dir, exist_ok=True)
    summary_path = os.path.join(summary_dir, f"仕分けサマリー_{ts_now}.txt")

    # 表の列幅を計算
    col_src      = max((len(src)      for src, _, _      in action_log), default=10)
    col_category = max((len(category) for _, category, _ in action_log), default=8)
    col_dest     = max((len(dest)     for _, _, dest     in action_log), default=10)
    col_src      = max(col_src,      len("元のパス"))
    col_category = max(col_category, len("カテゴリ"))
    col_dest     = max(col_dest,     len("移動先パス"))

    sep    = f"+{'-' * (col_src + 2)}+{'-' * (col_category + 2)}+{'-' * (col_dest + 2)}+"
    header = f"| {'元のパス':<{col_src}} | {'カテゴリ':<{col_category}} | {'移動先パス':<{col_dest}} |"

    with open(summary_path, encoding="utf-8", mode="w") as f:
        f.write("仕分けサマリー\n")
        f.write(f"実行日時: {datetime.now().strftime('%Y/%m/%d %H:%M:%S')}\n")
        f.write(f"対象フォルダ: {target_dir}\n")
        f.write(f"サブフォルダ含める: {'はい' if include_sub else 'いいえ'}\n")
        f.write(f"元ファイルの扱い: {file_mode}\n")
        f.write(f"合計: {total} 件\n")
        f.write("=" * 40 + "\n\n")

        # カテゴリ別集計
        f.write("【カテゴリ別集計】\n")
        for category in sorted(summary.keys()):
            fnames = summary[category]
            f.write(f"  {category}: {len(fnames)} 件\n")
        f.write("\n")

        # 詳細ログ（表形式）
        f.write("【詳細ログ】\n")
        f.write(sep + "\n")
        f.write(header + "\n")
        f.write(sep + "\n")
        for src, category, dest in action_log:
            f.write(f"| {src:<{col_src}} | {category:<{col_category}} | {dest:<{col_dest}} |\n")
        f.write(sep + "\n")

    print("---")
    print(f"完了！　合計: {total} 件")
    for category, fnames in sorted(summary.items()):
        print(f"  {category}: {len(fnames)} 件")
    print(f"サマリー: {summary_path}")

except Exception as e:
    print(f"エラーが発生しました: {e}", file=sys.stderr)