截图识别飞书个人任务清单

utils/notifier.py

@@ -0,0 +1,306 @@
+"""
+通知工具 - 支持桌面通知、声音提示等
+"""
+
+import platform
+import subprocess
+import logging
+from typing import Optional
+
+
+def send_notification(title: str, message: str, urgency: str = "normal") -> bool:
+    """
+    发送桌面通知
+    
+    Args:
+        title: 通知标题
+        message: 通知内容
+        urgency: 通知紧急程度 (low, normal, high)
+    
+    Returns:
+        是否成功发送
+    """
+    system = platform.system()
+    
+    try:
+        if system == "Windows":
+            # Windows 使用 PowerShell 发送通知
+            ps_script = f"""
+            [Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType=WindowsRuntime] > $null
+            $template = [Windows.UI.Notifications.ToastNotificationManager]::GetTemplateContent([Windows.UI.Notifications.ToastTemplateType]::ToastText02)
+            $template.SelectSingleNode("//text[@id='1']").AppendChild($template.CreateTextNode('{title}')) > $null
+            $template.SelectSingleNode("//text[@id='2']").AppendChild($template.CreateTextNode('{message}')) > $null
+            $toast = [Windows.UI.Notifications.ToastNotification]::new($template)
+            [Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier("Screen2Feishu").Show($toast)
+            """
+            
+            # 使用 PowerShell 执行
+            subprocess.run(
+                ["powershell", "-Command", ps_script],
+                capture_output=True,
+                timeout=5
+            )
+            return True
+            
+        elif system == "Darwin":  # macOS
+            # macOS 使用 osascript 发送通知
+            script = f'display notification "{message}" with title "{title}"'
+            subprocess.run(
+                ["osascript", "-e", script],
+                capture_output=True,
+                timeout=5
+            )
+            return True
+            
+        elif system == "Linux":
+            # Linux 使用 notify-send
+            try:
+                subprocess.run(
+                    ["notify-send", "-u", urgency, title, message],
+                    capture_output=True,
+                    timeout=5
+                )
+                return True
+            except FileNotFoundError:
+                # 如果 notify-send 不可用，尝试使用 zenity
+                try:
+                    subprocess.run(
+                        ["zenity", "--info", "--title", title, "--text", message],
+                        capture_output=True,
+                        timeout=5
+                    )
+                    return True
+                except FileNotFoundError:
+                    return False
+        else:
+            return False
+            
+    except Exception as e:
+        logging.error(f"发送通知失败: {str(e)}")
+        return False
+
+
+def send_notification_with_icon(title: str, message: str, icon_path: Optional[str] = None) -> bool:
+    """
+    发送带图标的桌面通知
+    
+    Args:
+        title: 通知标题
+        message: 通知内容
+        icon_path: 图标文件路径
+    
+    Returns:
+        是否成功发送
+    """
+    system = platform.system()
+    
+    try:
+        if system == "Linux" and icon_path:
+            # Linux 支持图标
+            subprocess.run(
+                ["notify-send", "-u", "normal", "-i", icon_path, title, message],
+                capture_output=True,
+                timeout=5
+            )
+            return True
+        else:
+            # 其他系统使用普通通知
+            return send_notification(title, message)
+            
+    except Exception as e:
+        logging.error(f"发送带图标通知失败: {str(e)}")
+        return False
+
+
+def play_sound(sound_type: str = "success") -> bool:
+    """
+    播放系统声音
+    
+    Args:
+        sound_type: 声音类型 (success, error, warning, info)
+    
+    Returns:
+        是否成功播放
+    """
+    system = platform.system()
+    
+    try:
+        if system == "Windows":
+            # Windows 使用 PowerShell 播放系统声音
+            sound_map = {
+                "success": "SystemAsterisk",
+                "error": "SystemHand",
+                "warning": "SystemExclamation",
+                "info": "SystemDefault"
+            }
+            
+            sound = sound_map.get(sound_type, "SystemDefault")
+            ps_script = f'[System.Media.SystemSounds]::{sound}.Play()'
+            
+            subprocess.run(
+                ["powershell", "-Command", ps_script],
+                capture_output=True,
+                timeout=5
+            )
+            return True
+            
+        elif system == "Darwin":  # macOS
+            # macOS 使用 afplay 播放系统声音
+            sound_map = {
+                "success": "/System/Library/Sounds/Purr.aiff",
+                "error": "/System/Library/Sounds/Basso.aiff",
+                "warning": "/System/Library/Sounds/Funk.aiff",
+                "info": "/System/Library/Sounds/Glass.aiff"
+            }
+            
+            sound_file = sound_map.get(sound_type, "/System/Library/Sounds/Glass.aiff")
+            
+            # 异步播放声音
+            subprocess.Popen(
+                ["afplay", sound_file],
+                stdout=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL
+            )
+            return True
+            
+        elif system == "Linux":
+            # Linux 使用 paplay 或 aplay 播放声音
+            # 这里使用简单的 beep 声音
+            try:
+                # 尝试使用 beep
+                subprocess.run(
+                    ["beep"],
+                    capture_output=True,
+                    timeout=5
+                )
+                return True
+            except FileNotFoundError:
+                # 如果 beep 不可用，尝试使用 paplay
+                try:
+                    # 创建临时的声音文件
+                    import tempfile
+                    import wave
+                    import struct
+                    
+                    # 创建简单的正弦波声音
+                    with tempfile.NamedTemporaryFile(suffix='.wav', delete=False) as f:
+                        # 写入 WAV 文件头
+                        sample_rate = 44100
+                        duration = 0.2
+                        num_samples = int(sample_rate * duration)
+                        
+                        wav_file = wave.open(f.name, 'w')
+                        wav_file.setnchannels(1)
+                        wav_file.setsampwidth(2)
+                        wav_file.setframerate(sample_rate)
+                        
+                        # 生成正弦波
+                        for i in range(num_samples):
+                            value = int(32767 * 0.3 * (1 if i < num_samples // 2 else 0))
+                            wav_file.writeframes(struct.pack('<h', value))
+                        
+                        wav_file.close()
+                        
+                        # 播放声音
+                        subprocess.run(
+                            ["paplay", f.name],
+                            capture_output=True,
+                            timeout=5
+                        )
+                        
+                        # 删除临时文件
+                        import os
+                        os.unlink(f.name)
+                        
+                    return True
+                except Exception:
+                    return False
+        else:
+            return False
+            
+    except Exception as e:
+        logging.error(f"播放声音失败: {str(e)}")
+        return False
+
+
+def notify_task_processed(success: bool, file_name: str, count: int = 1) -> bool:
+    """
+    通知任务处理结果
+    
+    Args:
+        success: 是否成功
+        file_name: 文件名
+        count: 处理数量
+    
+    Returns:
+        是否成功发送通知
+    """
+    if success:
+        title = "  任务处理成功"
+        message = f"已成功处理 {count} 个文件: {file_name}"
+        sound_type = "success"
+    else:
+        title = "  任务处理失败"
+        message = f"处理文件失败: {file_name}"
+        sound_type = "error"
+    
+    # 发送桌面通知
+    notification_sent = send_notification(title, message)
+    
+    # 播放声音提示
+    sound_played = play_sound(sound_type)
+    
+    return notification_sent or sound_played
+
+
+def notify_batch_result(results: dict) -> bool:
+    """
+    通知批量处理结果
+    
+    Args:
+        results: 处理结果字典 {文件名: 是否成功}
+    
+    Returns:
+        是否成功发送通知
+    """
+    success_count = sum(1 for v in results.values() if v)
+    total_count = len(results)
+    
+    if success_count == total_count:
+        title = "  批量处理完成"
+        message = f"所有 {total_count} 个文件处理成功"
+        sound_type = "success"
+    elif success_count == 0:
+        title = "  批量处理失败"
+        message = f"所有 {total_count} 个文件处理失败"
+        sound_type = "error"
+    else:
+        title = "⚠️ 批量处理部分成功"
+        message = f"{success_count}/{total_count} 个文件处理成功"
+        sound_type = "warning"
+    
+    # 发送桌面通知
+    notification_sent = send_notification(title, message)
+    
+    # 播放声音提示
+    sound_played = play_sound(sound_type)
+    
+    return notification_sent or sound_played
+
+
+if __name__ == "__main__":
+    # 测试通知功能
+    import logging
+    logging.basicConfig(level=logging.INFO)
+    
+    print("测试桌面通知...")
+    success = send_notification("测试通知", "这是一个测试通知消息")
+    print(f"通知发送: {'成功' if success else '失败'}")
+    
+    print("测试声音播放...")
+    success = play_sound("success")
+    print(f"声音播放: {'成功' if success else '失败'}")
+    
+    print("测试任务处理通知...")
+    success = notify_task_processed(True, "test_image.png")
+    print(f"任务通知: {'成功' if success else '失败'}")