初始化
   ↓
準(zhǔn)備窗口 → 激活并排列窗口
   ↓
掃描文件區(qū)域 → 圖像識(shí)別獲取所有文件圖標(biāo)位置
   ↓
文件分類 → 根據(jù)截圖局部特征判斷類型
   ↓
拖拽移動(dòng) → 模擬鼠標(biāo)拖拽到目標(biāo)窗口
   ↓
日志記錄 → 記錄操作結(jié)果
   ↓
異常處理 → 重試/跳過(guò)/報(bào)警

import pygetwindow as gw
import pyautogui
import time
import subprocess
import sys

def ensure_folder_window(folder_path, window_title_pattern):
    """確保指定文件夾的窗口已打開且激活"""
    # 查找現(xiàn)有窗口
    windows = [w for w in gw.getAllWindows() 
               if w.title and folder_path in w.title]
    if windows:
        win = windows[0]
        win.activate()
        time.sleep(0.5)
        return win
    
    # 未找到，打開新窗口
    if sys.platform == 'win32':
        subprocess.Popen(f'explorer "{folder_path}"')
    elif sys.platform == 'darwin':
        subprocess.Popen(['open', folder_path])
    else:  # Linux 簡(jiǎn)單模擬
        pyautogui.hotkey('ctrl', 'alt', 't')
        time.sleep(1)
        pyautogui.typewrite(f'nautilus "{folder_path}"')
        pyautogui.press('enter')
    
    time.sleep(2)  # 等待窗口打開
    return ensure_folder_window(folder_path, window_title_pattern)  # 遞歸獲取

def arrange_windows(download_win, category_wins):
    """排列窗口：左大右三"""
    screen_w, screen_h = pyautogui.size()
    
    # 下載窗口占左側(cè)2/5
    dw_width = int(screen_w * 0.4)
    dw_height = int(screen_h * 0.8)
    download_win.moveTo(0, 0)
    download_win.resizeTo(dw_width, dw_height)
    time.sleep(0.2)
    
    # 右側(cè)三個(gè)分類窗口均分
    cw_width = int(screen_w * 0.6) - 10
    cw_height = int(screen_h * 0.26)
    for i, win in enumerate(category_wins):
        win.moveTo(dw_width + 5, i * (cw_height + 5))
        win.resizeTo(cw_width, cw_height)
        time.sleep(0.2)

def locate_file_icons(win, item_height=25, max_items=50):
    """基于窗口內(nèi)第一個(gè)文件圖標(biāo)，推斷所有文件位置"""
    region = (win.left, win.top + 60, win.width, win.height - 80)  # 避開工具欄
    
    # 找第一個(gè)文件圖標(biāo)（可使用通用文件圖標(biāo)模板）
    first_icon = pyautogui.locateOnScreen('file_icon_template.png',
                                          region=region,
                                          confidence=0.7,
                                          grayscale=True)
    if not first_icon:
        return []
    
    icons = []
    x, y = first_icon.left, first_icon.top
    for i in range(max_items):
        current_y = y + i * item_height
        if current_y + item_height > win.top + win.height:
            break
        # 在預(yù)測(cè)位置附近搜索確認(rèn)
        search_region = (x - 5, current_y - 5, 50, item_height + 10)
        confirmed = pyautogui.locateOnScreen('file_icon_template.png',
                                             region=search_region,
                                             confidence=0.7,
                                             grayscale=True)
        if confirmed:
            icons.append(confirmed)
        else:
            break  # 連續(xù)找不到則終止
    return icons

def classify_file_icon(icon_box):
    """根據(jù)圖標(biāo)區(qū)域截圖判斷文件類型"""
    # 截取圖標(biāo)區(qū)域（左上角小圖標(biāo)）
    icon_img = pyautogui.screenshot(region=(
        icon_box.left, icon_box.top, 32, 32))
    icon_img.save('temp_icon.png')  # 調(diào)試用
    
    # 分別匹配
    if pyautogui.locateOnScreen('image_icon.png', region=(icon_box.left, icon_box.top, 32, 32), confidence=0.75):
        return 'image'
    elif pyautogui.locateOnScreen('doc_icon.png', confidence=0.75):
        return 'document'
    elif pyautogui.locateOnScreen('zip_icon.png', confidence=0.75):
        return 'archive'
    else:
        return 'other'

def get_file_names_from_folder(win):
    """激活窗口，Ctrl+A全選，Ctrl+C復(fù)制文件名，從剪貼板解析"""
    win.activate()
    time.sleep(0.3)
    pyautogui.hotkey('ctrl', 'a')
    time.sleep(0.2)
    pyautogui.hotkey('ctrl', 'c')
    time.sleep(0.2)
    text = pyperclip.paste()
    # 解析文件名（Windows資源管理器復(fù)制的是完整路徑？需測(cè)試）
    lines = text.splitlines()
    return [line for line in lines if line and not line.startswith(' ')]

def drag_file(icon_box, target_win):
    """將文件從圖標(biāo)位置拖拽到目標(biāo)窗口中心"""
    # 起點(diǎn)：圖標(biāo)中心
    start_x = icon_box.left + icon_box.width // 2
    start_y = icon_box.top + icon_box.height // 2
    
    # 終點(diǎn)：目標(biāo)窗口中心偏左（避免覆蓋滾動(dòng)條）
    end_x = target_win.left + target_win.width // 3
    end_y = target_win.top + target_win.height // 2
    
    pyautogui.moveTo(start_x, start_y)
    time.sleep(0.1)
    pyautogui.dragTo(end_x, end_y, duration=0.5, button='left')
    time.sleep(0.3)  # 等待系統(tǒng)完成移動(dòng)

import logging

def setup_logger():
    logger = logging.getLogger('DesktopCleaner')
    logger.setLevel(logging.INFO)
    
    fh = logging.FileHandler('cleaner.log', encoding='utf-8')
    ch = logging.StreamHandler()
    
    formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
    fh.setFormatter(formatter)
    ch.setFormatter(formatter)
    
    logger.addHandler(fh)
    logger.addHandler(ch)
    return logger

logger = setup_logger()

import pyautogui
import pygetwindow as gw
import pyperclip
import time
import logging
import sys
import re

class DesktopCleaner:
    def __init__(self, download_path, categories):
        """
        categories: {'image': 'C:\\分類\\圖片', 
                     'document': 'C:\\分類\\文檔', 
                     'archive': 'C:\\分類\\壓縮包'}
        """
        self.download_path = download_path
        self.categories = categories
        self.logger = self._setup_logger()
        self.windows = {}  # 存儲(chǔ)窗口對(duì)象
        
    def _setup_logger(self):
        # ... 如上logging配置 ...
    
    def ensure_all_windows(self):
        """打開/激活所有需要的文件夾窗口"""
        self.logger.info("準(zhǔn)備窗口...")
        self.windows['download'] = self._ensure_window(self.download_path)
        for name, path in self.categories.items():
            self.windows[name] = self._ensure_window(path)
        self._arrange_windows()
        self.logger.info("窗口準(zhǔn)備完成")
    
    def _ensure_window(self, folder_path):
        # ... 實(shí)現(xiàn)參考2.1 ...
    
    def _arrange_windows(self):
        # ... 窗口布局 ...
    
    def scan_files(self):
        """獲取文件列表及圖標(biāo)位置"""
        win = self.windows['download']
        win.activate()
        time.sleep(0.5)
        
        # 全選并復(fù)制文件名列表
        pyautogui.hotkey('ctrl', 'a')
        time.sleep(0.2)
        pyautogui.hotkey('ctrl', 'c')
        time.sleep(0.2)
        file_names = [f.strip() for f in pyperclip.paste().splitlines() if f.strip()]
        
        # 獲取圖標(biāo)位置列表（假設(shè)順序一致）
        icons = self._locate_icons(win)
        if len(icons) != len(file_names):
            self.logger.warning(f"圖標(biāo)數(shù)量({len(icons)})與文件名數(shù)量({len(file_names)})不一致，按較小值處理")
            min_len = min(len(icons), len(file_names))
            icons = icons[:min_len]
            file_names = file_names[:min_len]
        
        # 為每個(gè)文件加上類型標(biāo)簽
        files_info = []
        for name, icon in zip(file_names, icons):
            ftype = self._get_file_type(name)
            files_info.append((name, icon, ftype))
        
        self.logger.info(f"掃描到{len(files_info)}個(gè)文件")
        return files_info
    
    def _locate_icons(self, win):
        # 基于行高推斷，參考2.2
        # 簡(jiǎn)化：直接使用全選后的視覺反饋，找到第一個(gè)被選中的圖標(biāo)
        # 實(shí)際需精細(xì)調(diào)整
        pass
    
    def _get_file_type(self, filename):
        """根據(jù)擴(kuò)展名判斷類型"""
        ext = filename.lower().split('.')[-1] if '.' in filename else ''
        if ext in ('jpg', 'jpeg', 'png', 'gif', 'bmp', 'tiff'):
            return 'image'
        elif ext in ('txt', 'doc', 'docx', 'pdf', 'xls', 'xlsx', 'ppt', 'pptx'):
            return 'document'
        elif ext in ('zip', 'rar', '7z', 'tar', 'gz'):
            return 'archive'
        else:
            return 'other'
    
    def move_file(self, file_info):
        """移動(dòng)單個(gè)文件"""
        name, icon, ftype = file_info
        if ftype not in self.windows:
            self.logger.warning(f"文件{name}類型{ftype}無(wú)對(duì)應(yīng)目標(biāo)窗口，跳過(guò)")
            return False
        
        target_win = self.windows[ftype]
        try:
            start_time = time.time()
            self._drag_file(icon, target_win)
            elapsed = time.time() - start_time
            self.logger.info(f"成功移動(dòng): {name} -> {ftype}, 耗時(shí){elapsed:.2f}s")
            return True
        except Exception as e:
            self.logger.error(f"移動(dòng)失敗: {name}, 錯(cuò)誤: {str(e)}")
            return False
    
    def _drag_file(self, icon_box, target_win):
        # ... 拖拽實(shí)現(xiàn) ...
    
    def run(self):
        self.ensure_all_windows()
        files = self.scan_files()
        self.logger.info(f"開始移動(dòng)文件，共{len(files)}個(gè)")
        
        success_count = 0
        for file_info in files:
            if self.move_file(file_info):
                success_count += 1
            time.sleep(0.2)  # 操作間隔
        
        self.logger.info(f"整理完成，成功: {success_count}, 失敗: {len(files)-success_count}")

if __name__ == '__main__':
    cleaner = DesktopCleaner(
        download_path='C:\\Users\\YourName\\Downloads',
        categories={
            'image': 'D:\\Sorted\\Images',
            'document': 'D:\\Sorted\\Documents',
            'archive': 'D:\\Sorted\\Archives'
        }
    )
    cleaner.run()

def locate_selected_icons(win):
    """通過(guò)全選高亮定位所有文件"""
    # 全選
    win.activate()
    pyautogui.hotkey('ctrl', 'a')
    time.sleep(0.3)
    
    # 截圖并找出所有藍(lán)色高亮區(qū)域
    region = (win.left, win.top, win.width, win.height)
    screenshot = pyautogui.screenshot(region=region)
    # 使用OpenCV顏色過(guò)濾，找出選中狀態(tài)的矩形框
    # ... 此處省略O(shè)penCV處理代碼 ...
    # 返回每個(gè)選中框的中心坐標(biāo)

def batch_drag(icon_boxes, target_win):
    """批量拖拽多個(gè)文件"""
    # 先依次Ctrl+點(diǎn)擊每個(gè)圖標(biāo)（選中）
    for box in icon_boxes[:-1]:
        pyautogui.click(box.left+5, box.top+5, button='left')
        time.sleep(0.1)
    # 拖拽最后一個(gè)文件
    pyautogui.mouseDown(icon_boxes[-1].left+5, icon_boxes[-1].top+5)
    time.sleep(0.1)
    pyautogui.moveTo(target_win.left+100, target_win.top+200, duration=0.5)
    pyautogui.mouseUp()