# 核心依賴
import wx              # GUI 框架
import json           # 數(shù)據(jù)持久化
from pathlib import Path  # 跨平臺(tái)路徑處理
import shutil         # 文件操作

{
  "/Users/username/Desktop/report.pdf": {
    "access_count": 15,
    "last_access": 1705132800.0,
    "created": 1704528000.0
  }
}

file_info = {
    'path': str,        # 完整路徑
    'name': str,        # 文件名
    'is_dir': bool,     # 是否為目錄
    'size': int,        # 字節(jié)大小
    'created': float,   # 創(chuàng)建時(shí)間戳
    'modified': float,  # 修改時(shí)間戳
    'accessed': float   # 訪問(wèn)時(shí)間戳
}

def __init__(self):
    super().__init__(None, title="智能桌面整理助手", size=(900, 700))
    
    # 獲取桌面路徑（跨平臺(tái)）
    self.desktop_path = Path.home() / "Desktop"
    
    # 數(shù)據(jù)存儲(chǔ)在用戶主目錄（隱藏文件）
    self.data_file = Path.home() / ".desktop_organizer_data.json"
    self.usage_data = self.load_usage_data()
    
    # 創(chuàng)建三個(gè)整理區(qū)域
    self.organize_root = self.desktop_path / "桌面整理"
    self.focus_zone = self.organize_root / "專注區(qū)"
    self.assist_zone = self.organize_root / "輔助區(qū)"
    self.temp_zone = self.organize_root / "臨時(shí)區(qū)"

def init_ui(self):
    panel = wx.Panel(self)
    main_sizer = wx.BoxSizer(wx.VERTICAL)
    
    # 1. 標(biāo)題區(qū)域
    title = wx.StaticText(panel, label="?? 智能桌面整理助手")
    title_font = wx.Font(16, wx.FONTFAMILY_DEFAULT, 
                         wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_BOLD)
    title.SetFont(title_font)
    
    # 2. 統(tǒng)計(jì)信息區(qū)
    stats_box = wx.StaticBoxSizer(wx.VERTICAL, panel, "桌面統(tǒng)計(jì)")
    self.stats_text = wx.StaticText(panel, label="正在掃描...")
    
    # 3. 標(biāo)簽頁(yè)組件
    notebook = wx.Notebook(panel)
    notebook.AddPage(self.suggestions_panel, "整理建議")
    notebook.AddPage(self.analysis_panel, "使用分析")
    notebook.AddPage(self.settings_panel, "設(shè)置")

self.suggestions_list = wx.ListCtrl(panel, 
                                     style=wx.LC_REPORT | wx.LC_SINGLE_SEL)
self.suggestions_list.AppendColumn("文件名", width=250)
self.suggestions_list.AppendColumn("類(lèi)型", width=80)
self.suggestions_list.AppendColumn("使用頻率", width=100)
self.suggestions_list.AppendColumn("建議操作", width=150)
self.suggestions_list.AppendColumn("原因", width=250)

def scan_desktop(self):
    if not self.desktop_path.exists():
        return
    
    files = []
    total_size = 0
    
    try:
        for item in self.desktop_path.iterdir():
            # 跳過(guò)隱藏文件和整理目錄
            if item.name.startswith('.') or item.name == '桌面整理':
                continue
            
            # 獲取文件元數(shù)據(jù)
            stat = item.stat()
            file_info = {
                'path': str(item),
                'name': item.name,
                'is_dir': item.is_dir(),
                'size': stat.st_size,
                'created': stat.st_ctime,
                'modified': stat.st_mtime,
                'accessed': stat.st_atime
            }
            
            # 更新使用數(shù)據(jù)
            file_key = str(item)
            if file_key not in self.usage_data:
                # 新文件：初始化記錄
                self.usage_data[file_key] = {
                    'access_count': 0,
                    'last_access': stat.st_atime,
                    'created': stat.st_ctime
                }
            else:
                # 已存在：檢測(cè)訪問(wèn)時(shí)間變化
                if stat.st_atime > self.usage_data[file_key]['last_access']:
                    self.usage_data[file_key]['access_count'] += 1
                    self.usage_data[file_key]['last_access'] = stat.st_atime
            
            files.append(file_info)
            total_size += stat.st_size
    
    except Exception as e:
        wx.MessageBox(f"掃描失敗: {str(e)}", "錯(cuò)誤", 
                      wx.OK | wx.ICON_ERROR)
        return
    
    # 保存數(shù)據(jù)并更新 UI
    self.save_usage_data()
    self.update_statistics(files, total_size)
    self.generate_suggestions(files)
    self.update_analysis(files)

if stat.st_atime > self.usage_data[file_key]['last_access']:
    self.usage_data[file_key]['access_count'] += 1

def generate_suggestions(self, files):
    self.suggestions_list.DeleteAllItems()
    self.suggestions = []
    
    now = time.time()
    focus_threshold = self.focus_days.GetValue() * 86400  # 轉(zhuǎn)換為秒
    assist_threshold = self.assist_days.GetValue() * 86400
    
    for file_info in files:
        file_key = file_info['path']
        usage = self.usage_data.get(file_key, {})
        
        # 計(jì)算關(guān)鍵指標(biāo)
        last_access = usage.get('last_access', file_info['accessed'])
        access_count = usage.get('access_count', 0)
        days_since_access = (now - last_access) / 86400
        
        # 決策邏輯
        if days_since_access <= focus_threshold and access_count > 2:
            suggestion = "保留在專注區(qū)"
            reason = f"最近{int(days_since_access)}天使用過(guò){access_count}次"
        
        elif days_since_access <= assist_threshold and access_count > 0:
            suggestion = "移至輔助區(qū)"
            reason = f"{int(days_since_access)}天前使用，訪問(wèn){access_count}次"
        
        else:
            suggestion = "移至臨時(shí)區(qū)"
            reason = f"已{int(days_since_access)}天未使用"
        
        # 添加到界面
        file_type = "文件夾" if file_info['is_dir'] else "文件"
        freq = "高頻" if access_count > 5 else \
               "中頻" if access_count > 2 else "低頻"
        
        idx = self.suggestions_list.InsertItem(
            self.suggestions_list.GetItemCount(), 
            file_info['name']
        )
        self.suggestions_list.SetItem(idx, 1, file_type)
        self.suggestions_list.SetItem(idx, 2, freq)
        self.suggestions_list.SetItem(idx, 3, suggestion)
        self.suggestions_list.SetItem(idx, 4, reason)

def apply_organization(self):
    try:
        # 1. 創(chuàng)建目標(biāo)目錄
        self.organize_root.mkdir(exist_ok=True)
        self.focus_zone.mkdir(exist_ok=True)
        self.assist_zone.mkdir(exist_ok=True)
        self.temp_zone.mkdir(exist_ok=True)
        
        operations = []  # 記錄所有操作，用于撤銷(xiāo)
        
        for suggestion in self.suggestions:
            file_info = suggestion['file_info']
            action = suggestion['suggestion']
            source = Path(file_info['path'])
            
            if not source.exists():
                continue
            
            # 2. 確定目標(biāo)目錄
            if "專注區(qū)" in action:
                target_dir = self.focus_zone
            elif "輔助區(qū)" in action:
                target_dir = self.assist_zone
            elif "臨時(shí)區(qū)" in action:
                target_dir = self.temp_zone
            else:
                continue
            
            target = target_dir / source.name
            
            # 3. 處理文件名沖突
            counter = 1
            while target.exists():
                stem = source.stem
                suffix = source.suffix
                target = target_dir / f"{stem}_{counter}{suffix}"
                counter += 1
            
            # 4. 執(zhí)行移動(dòng)
            try:
                shutil.move(str(source), str(target))
                operations.append({
                    'source': str(source), 
                    'target': str(target)
                })
            except Exception as e:
                print(f"移動(dòng)失敗: {source} -> {target}, 錯(cuò)誤: {e}")
        
        # 5. 保存撤銷(xiāo)日志
        if operations:
            undo_file = self.organize_root / ".undo_log.json"
            with open(undo_file, 'w', encoding='utf-8') as f:
                json.dump(operations, f, ensure_ascii=False, indent=2)
            
            self.undo_btn.Enable(True)
        
        wx.MessageBox(f"整理完成！共處理 {len(operations)} 個(gè)項(xiàng)目", 
                      "成功", wx.OK | wx.ICON_INFORMATION)
        self.scan_desktop()
    
    except Exception as e:
        wx.MessageBox(f"整理失敗: {str(e)}", "錯(cuò)誤", 
                      wx.OK | wx.ICON_ERROR)

while target.exists():
    target = target_dir / f"{stem}_{counter}{suffix}"
    counter += 1

[
  {
    "source": "/Users/xxx/Desktop/file.txt",
    "target": "/Users/xxx/Desktop/桌面整理/臨時(shí)區(qū)/file.txt"
  }
]

def on_undo(self, event):
    undo_file = self.organize_root / ".undo_log.json"
    
    if not undo_file.exists():
        wx.MessageBox("沒(méi)有可撤銷(xiāo)的操作", "提示", 
                      wx.OK | wx.ICON_INFORMATION)
        return
    
    try:
        # 1. 讀取操作日志
        with open(undo_file, 'r', encoding='utf-8') as f:
            operations = json.load(f)
        
        # 2. 反向執(zhí)行所有操作
        for op in operations:
            target = Path(op['target'])
            source = Path(op['source'])
            
            if target.exists():
                shutil.move(str(target), str(source))
        
        # 3. 刪除日志并禁用撤銷(xiāo)按鈕
        undo_file.unlink()
        self.undo_btn.Enable(False)
        
        wx.MessageBox("撤銷(xiāo)成功！", "成功", 
                      wx.OK | wx.ICON_INFORMATION)
        self.scan_desktop()
    
    except Exception as e:
        wx.MessageBox(f"撤銷(xiāo)失敗: {str(e)}", "錯(cuò)誤", 
                      wx.OK | wx.ICON_ERROR)

# 初始化定時(shí)器
self.timer = wx.Timer(self)
self.Bind(wx.EVT_TIMER, self.on_timer, self.timer)
self.timer.Start(300000)  # 每 5 分鐘（300,000 毫秒）

def on_timer(self, event):
    """定時(shí)掃描"""
    self.scan_desktop()

{
  "總掃描次數(shù)": 2016,  // 每 5 分鐘掃描一次
  "追蹤文件數(shù)": 47,
  "平均訪問(wèn)頻率": {
    "專注區(qū)文件": 12.3,
    "輔助區(qū)文件": 3.5,
    "臨時(shí)區(qū)文件": 0.2
  },
  "整理操作次數(shù)": 3,
  "撤銷(xiāo)次數(shù)": 1
}