基于Python编写一个桌面文件自动整理工具

mkdir ~/desktop-organizer
cd ~/desktop-organizer
touch organizer.py organize_log.json

# 扩展名 → 目标文件夹名
CATEGORY_MAP = {
    "pdf": "文档",
    "doc": "文档", "docx": "文档", "txt": "文档", "md": "文档",
    "jpg": "图片", "jpeg": "图片", "png": "图片", "gif": "图片",
    "svg": "图片", "webp": "图片", "bmp": "图片",
    "mp4": "视频", "mov": "视频", "avi": "视频", "mkv": "视频",
    "mp3": "音频", "wav": "音频", "flac": "音频",
    "zip": "压缩包", "rar": "压缩包", "7z": "压缩包", "tar.gz": "压缩包",
    "py": "代码", "js": "代码", "html": "代码", "css": "代码",
    "exe": "可执行", "dmg": "可执行", "deb": "可执行",
}

# 默认兜底分类
DEFAULT_CATEGORY = "其他"

#!/usr/bin/env python3
"""桌面文件自动整理工具"""

import os
import shutil
import logging
from pathlib import Path

# 配置日志
logging.basicConfig(
    filename="organize_log.json",
    level=logging.INFO,
    format="%(asctime)s | %(message)s"
)

CATEGORY_MAP = {
    "pdf": "文档", "doc": "文档", "docx": "文档", "txt": "文档", "md": "文档",
    "jpg": "图片", "jpeg": "图片", "png": "图片", "gif": "图片",
    "mp4": "视频", "mov": "视频", "mp3": "音频", "wav": "音频",
    "zip": "压缩包", "rar": "压缩包", "7z": "压缩包",
    "py": "代码", "js": "代码", "html": "代码",
}
DEFAULT_CATEGORY = "其他"


def get_category(filename: str) -> str:
    """根据文件名返回分类文件夹名"""
    ext = Path(filename).suffix.lower().lstrip(".")
    return CATEGORY_MAP.get(ext, DEFAULT_CATEGORY)


def move_file(src: Path, dest_dir: Path) -> Path:
    """移动文件，同名时自动加 _1, _2 后缀"""
    dest_dir.mkdir(parents=True, exist_ok=True)
    dest = dest_dir / src.name
    if dest.exists():
        stem, suffix = src.stem, src.suffix
        counter = 1
        while dest.exists():
            dest = dest_dir / f"{stem}_{counter}{suffix}"
            counter += 1
    shutil.move(str(src), str(dest))
    logging.info(f"移动: {src.name} -> {dest}")
    return dest


def organize(target_dir: str = None):
    """扫描并整理目标目录"""
    if target_dir is None:
        target_dir = os.path.expanduser("~/Desktop")

    root = Path(target_dir).resolve()
    if not root.exists():
        print(f"❌ 目录不存在: {root}")
        return

    count = 0
    for item in root.iterdir():
        if not item.is_file():
            continue
        if item.name.startswith("."):  # 跳过隐藏文件
            continue

        category = get_category(item.name)
        dest_dir = root / category
        move_file(item, dest_dir)
        count += 1

    print(f"✅ 完成，共整理 {count} 个文件")
    logging.info(f"=== 本次整理 {count} 个文件 ===")


if __name__ == "__main__":
    organize()

# 整理桌面（默认）
python3 organizer.py

# 整理指定目录
python3 organizer.py -- /path/to/messy/folder

# 每周一、四早上 9 点自动整理
0 9 * * 1,4 cd ~/desktop-organizer && python3 organizer.py

MARKER_FILE = Path.home() / ".desktop_organize_done"

def should_run() -> bool:
    """避免同一天重复整理"""
    if MARKER_FILE.exists():
        last_run = MARKER_FILE.stat().st_mtime
        if (time.time() - last_run) < 86400:  # 24 小时内
            return False
    return True

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
AI文件整理助手 - 自动分类桌面文件
作者：技术实战派
"""
import os
import shutil
from pathlib import Path
from datetime import datetime
class AIFileOrganizer:
    """AI驱动的文件整理器"""
    def __init__(self, source_dir, target_dir=None):
        self.source_dir = Path(source_dir)
        self.target_dir = Path(target_dir) if target_dir else self.source_dir / "已整理"
        self.stats = {"moved": 0, "skipped": 0, "errors": 0}
        # 定义分类规则
        self.rules = {
            "截图": {
                "keywords": ["截图", "screen", "screencapture", "qq截图", "微信截图"],
                "exts": [".png", ".jpg", ".jpeg"],
                "folder": "图片/截图"
            },
            "合同文档": {
                "keywords": ["合同", "协议", "agreement", "contract"],
                "exts": [".pdf", ".doc", ".docx"],
                "folder": "文档/合同"
            },
            "财务票据": {
                "keywords": ["发票", "收据", "invoice", "receipt", "报销"],
                "exts": [".pdf", ".png", ".jpg"],
                "folder": "文档/财务"
            },
            "图片素材": {
                "keywords": ["img", "image", "photo", "pic"],
                "exts": [".jpg", ".jpeg", ".png", ".gif", ".bmp", ".webp"],
                "folder": "图片/素材"
            },
            "PDF文档": {
                "exts": [".pdf"],
                "folder": "文档/PDF"
            },
            "办公软件": {
                "keywords": ["word", "excel", "ppt", "doc", "xls"],
                "exts": [".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx"],
                "folder": "文档/办公"
            },
            "压缩包": {
                "exts": [".zip", ".rar", ".7z", ".tar", ".gz"],
                "folder": "压缩包"
            },
            "安装包": {
                "keywords": ["setup", "install", "installer"],
                "exts": [".exe", ".msi", ".dmg", ".pkg"],
                "folder": "软件/安装包"
            },
            "视频": {
                "exts": [".mp4", ".avi", ".mkv", ".mov", ".wmv", ".flv"],
                "folder": "视频"
            },
            "音频": {
                "exts": [".mp3", ".wav", ".flac", ".aac", ".ogg"],
                "folder": "音频"
            },
            "代码文件": {
                "exts": [".py", ".js", ".html", ".css", ".java", ".cpp", ".c", ".go", ".rs"],
                "folder": "代码"
            }
        }
    def ai_classify(self, filename):
        """
        AI智能分类：根据文件名和扩展名判断类型
        """
        name_lower = filename.lower()
        ext = Path(filename).suffix.lower()
        # 按优先级匹配规则
        for category, rule in self.rules.items():
            # 检查关键词
            keywords = rule.get("keywords", [])
            if any(kw.lower() in name_lower for kw in keywords):
                return rule["folder"]
            # 检查扩展名
            exts = rule.get("exts", [])
            if ext in exts:
                return rule["folder"]
        # 默认分类：按日期
        return f"其他/{datetime.now().strftime('%Y-%m')}"
    def organize(self, dry_run=False):
        """
        执行整理
        :param dry_run: 试运行模式（只打印，不移动）
        """
        print(f"📁 源目录: {self.source_dir}")
        print(f"📂 目标目录: {self.target_dir}")
        print(f"🔍 扫描文件中...\n")
        # 遍历源目录
        for item in self.source_dir.iterdir():
            if item.is_dir():
                continue  # 跳过目录
            filename = item.name
            # 跳过脚本自身
            if filename == os.path.basename(__file__):
                continue
            # AI分类
            category = self.ai_classify(filename)
            target_folder = self.target_dir / category
            # 显示结果
            action = "【模拟】" if dry_run else "【移动】"
            print(f"{action} {filename:<30} → {category}")
            if not dry_run:
                try:
                    # 创建目标目录
                    target_folder.mkdir(parents=True, exist_ok=True)
                    # 处理重名文件
                    target_path = target_folder / filename
                    counter = 1
                    while target_path.exists():
                        stem = Path(filename).stem
                        suffix = Path(filename).suffix
                        target_path = target_folder / f"{stem}_{counter}{suffix}"
                        counter += 1
                    # 移动文件
                    shutil.move(str(item), str(target_path))
                    self.stats["moved"] += 1
                except Exception as e:
                    print(f"  ❌ 错误: {e}")
                    self.stats["errors"] += 1
            else:
                self.stats["moved"] += 1
        # 打印统计
        print(f"\n{'='*50}")
        print(f"📊 整理完成！")
        print(f"   已处理: {self.stats['moved']} 个文件")
        print(f"   错误: {self.stats['errors']} 个")
        print(f"{'='*50}")
def main():
    """主函数"""
    import argparse
    parser = argparse.ArgumentParser(description='AI文件整理助手')
    parser.add_argument('path', nargs='?', default='.', help='要整理的目录（默认当前目录）')
    parser.add_argument('--output', '-o', help='输出目录（默认在源目录创建"已整理"）')
    parser.add_argument('--dry-run', '-d', action='store_true', help='试运行模式（不实际移动文件）')
    args = parser.parse_args()
    # 创建整理器
    organizer = AIFileOrganizer(args.path, args.output)
    # 执行整理
    organizer.organize(dry_run=args.dry_run)
if __name__ == "__main__":
    main()