""" 后台任务模块 负责创建和管理定期检查Apple Music状态并推送更新的后台任务 """ import time import asyncio import logging from .apple_music import get_music_status from .lyrics import get_lyrics_data from .websocket_manager import ConnectionManager logger = logging.getLogger(__name__) # 存储后台任务引用 background_tasks = set() async def fetch_and_update_lyrics(status, manager): """异步获取歌词并发送更新""" try: # 异步获取歌词 logger.info(f"正在获取歌词...") lyrics = get_lyrics_data(status) logger.info(f"歌词: {lyrics}") # 获取到歌词后，再发送一个歌词更新消息 if lyrics and isinstance(lyrics, dict): # 确保lyrics是一个非空的字典 lyrics_update = { "type": "lyric_update", # 使用不同的消息类型 "timestamp": time.time(), "lyrics": lyrics } await manager.broadcast(lyrics_update) logger.info("歌词获取完成，已发送更新") except Exception as e: logger.error(f"获取歌词并发送更新时出错: {e}") async def update_music_info(manager: ConnectionManager): """定期检查音乐状态并向所有客户端推送更新""" logger.info("后台任务update_music_info开始运行...") # 记录上一次状态 last_position = -1 last_track_name = None last_lyrics = None last_playback_status = None # 记录上一次的播放状态 # 配置更新间隔 check_interval = 0.05 # 内部检查间隔仍保持较短 # 设定重试次数 consecutive_errors = 0 max_consecutive_errors = 10 while True: try: current_time = time.time() # 获取当前状态 status = get_music_status() if status is None: logger.error("获取音乐状态失败，返回None") await asyncio.sleep(1) continue current_status = status.get("status", "unknown") # 错误状态处理 if current_status == "error": logger.error(f"获取音乐状态返回错误: {status.get('error', 'Unknown error')}") await asyncio.sleep(1) continue # 状态变化处理 if current_status != last_playback_status and last_playback_status is not None: logger.info(f"播放状态变化: {last_playback_status} -> {current_status}") # 确定消息类型 message_type = "status_change" if current_status == "playing": message_type = "playback_resumed" elif current_status == "paused": message_type = "playback_paused" elif current_status == "stopped": message_type = "playback_stopped" elif current_status == "notrunning": message_type = "app_not_running" # 向客户端通知状态变化 status_update = { "type": message_type, "timestamp": current_time, "status": status } await manager.broadcast(status_update) last_playback_status = current_status consecutive_errors = 0 await asyncio.sleep(check_interval) continue # 如果没有状态变化，但是还没有记录过状态，进行初始化 if last_playback_status is None: last_playback_status = current_status # 如果应用未运行或已停止，无需进一步处理 if current_status in ["stopped", "notrunning"]: await asyncio.sleep(check_interval) continue # 保存当前播放状态用于下次比较 last_playback_status = current_status # 如果已停止，无需进一步处理 if current_status in ["stopped", "notrunning"]: await asyncio.sleep(check_interval) continue # 以下是正常播放状态下的处理 position = status.get("position", -1) track_name = status.get("track_name") # 安全检查：如果缺少关键数据，跳过本次循环 if track_name is None: logger.debug("歌曲名称为空，跳过本次更新") await asyncio.sleep(check_interval) continue # 检查是否需要发送更新 track_changed = track_name != last_track_name # 如果歌曲变化，获取新歌词 if track_changed: try: # 立即发送歌曲变化通知（不含歌词） logger.info(f"歌曲变化: {last_track_name} -> {track_name}") # 先发送一个不含歌词的track_change消息 initial_update = { "type": "track_change", "timestamp": current_time, "status": status, "lyrics": None # 不包含歌词 } await manager.broadcast(initial_update) # 更新歌曲名称记录 last_track_name = track_name last_position = position # 重置错误计数 consecutive_errors = 0 # 启动一个独立的异步任务来获取歌词并后续发送更新 # 这样主循环可以继续运行，不会阻塞 asyncio.create_task(fetch_and_update_lyrics(status, manager)) except Exception as e: logger.error(f"处理歌曲变化时出错: {e}") await asyncio.sleep(0.5) # 检查歌词是否需要更新（播放位置变化明显） elif abs(last_position - position) > 0.5: # 如果播放位置变化超过0.5秒 try: lyrics = get_lyrics_data(status) # 只有当歌词内容变化时才发送更新 current_lyric = lyrics.get("current_lyric", "") if lyrics and isinstance(lyrics, dict) else "" current_lyric_time = lyrics.get("current_lyric_time", "") if lyrics and isinstance(lyrics, dict) else "" last_current_lyric_time = last_lyrics.get("current_lyric_time", "") if last_lyrics and isinstance(last_lyrics, dict) else "" if current_lyric_time != last_current_lyric_time: if current_lyric is None: current_lyric = "" logger.debug(f"歌词更新: {current_lyric[:20]}...") combined_data = { "type": "lyric_change", "timestamp": current_time, "status": status, "lyrics": lyrics } await manager.broadcast(combined_data) last_lyrics = lyrics last_position = position # 重置错误计数 consecutive_errors = 0 except Exception as e: logger.error(f"处理歌词更新时出错: {e}") await asyncio.sleep(0.5) # 更新上一次位置 if track_changed or (abs(last_position - position) > 0.5): last_position = position # 等待下一次检查 await asyncio.sleep(check_interval) except Exception as e: logger.error(f"更新音乐信息时出错: {e}") consecutive_errors += 1 if consecutive_errors >= max_consecutive_errors: logger.error(f"连续错误次数超过{max_consecutive_errors}，退出后台任务") break await asyncio.sleep(1) # 出错时等待1秒再重试 def start_background_task(manager: ConnectionManager): """启动后台任务并添加到跟踪集合中""" task = asyncio.create_task(update_music_info(manager)) background_tasks.add(task) task.add_done_callback(background_tasks.discard) return task def cancel_all_tasks(): """取消所有后台任务""" for task in background_tasks: task.cancel()