"""网易云音乐二维码登录模块 提供网易云音乐二维码登录功能，包括： - 二维码生成和显示 - 登录状态检查 - Cookie获取和保存 - 用户友好的交互界面 """ import sys import time import logging from typing import Optional, Dict, Any, Tuple from pathlib import Path try: from music_api import QRLoginManager, APIException from cookie_manager import CookieManager, CookieException except ImportError as e: print(f"导入模块失败: {e}") print("请确保 music_api.py 和 cookie_manager.py 文件存在且可用") sys.exit(1) class QRLoginClient: """二维码登录客户端""" def __init__(self, cookie_file: str = "cookie.txt"): """ 初始化二维码登录客户端 Args: cookie_file: Cookie保存文件路径 """ self.cookie_manager = CookieManager(cookie_file) self.qr_manager = QRLoginManager() self.logger = logging.getLogger(__name__) # 配置日志 if not self.logger.handlers: handler = logging.StreamHandler() formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s') handler.setFormatter(formatter) self.logger.addHandler(handler) self.logger.setLevel(logging.INFO) def check_existing_login(self) -> bool: """检查是否已有有效登录 Returns: 是否已登录 """ try: if self.cookie_manager.is_cookie_valid(): self.logger.info("检测到有效的登录Cookie") return True else: self.logger.info("未检测到有效的登录Cookie") return False except Exception as e: self.logger.error(f"检查登录状态失败: {e}") return False def interactive_login(self) -> Tuple[bool, Optional[str]]: """交互式二维码登录 Returns: (登录是否成功, 错误信息) """ try: print("\n=== 网易云音乐二维码登录 ===") # 检查现有登录状态 if self.check_existing_login(): choice = input("检测到已有有效登录，是否重新登录？(y/N): ").strip().lower() if choice not in ['y', 'yes', '是']: print("使用现有登录状态") return True, None print("\n开始二维码登录流程...") # 生成二维码 print("正在生成二维码...") qr_result = self.qr_manager.create_qr_login() if not qr_result['success']: error_msg = f"生成二维码失败: {qr_result.get('message', '未知错误')}" self.logger.error(error_msg) return False, error_msg qr_key = qr_result['qr_key'] print(f"\n二维码已生成！") print(f"请使用网易云音乐手机APP扫描二维码进行登录") print(f"二维码有效期: 3分钟") print("\n等待扫码中...") # 轮询检查登录状态 max_attempts = 60 # 最多等待5分钟 attempt = 0 while attempt < max_attempts: try: # 检查登录状态 status_result = self.qr_manager.check_qr_login(qr_key) if status_result['success']: if status_result['status'] == 'success': # 登录成功 cookie = status_result.get('cookie', '') if cookie: # 保存Cookie success = self.save_cookie(cookie) if success: print("\n✅ 登录成功！Cookie已保存") return True, None else: error_msg = "登录成功但Cookie保存失败" self.logger.error(error_msg) return False, error_msg else: error_msg = "登录成功但未获取到Cookie" self.logger.error(error_msg) return False, error_msg elif status_result['status'] == 'waiting': # 等待扫码 if attempt % 10 == 0: # 每10次显示一次提示 print(f"等待扫码中... ({attempt + 1}/{max_attempts})") elif status_result['status'] == 'scanned': # 已扫码，等待确认 print("二维码已扫描，请在手机上确认登录") elif status_result['status'] == 'expired': # 二维码过期 error_msg = "二维码已过期，请重新尝试" print(f"\n❌ {error_msg}") return False, error_msg elif status_result['status'] == 'error': # 登录错误 error_msg = f"登录失败: {status_result.get('message', '未知错误')}" print(f"\n❌ {error_msg}") return False, error_msg else: self.logger.warning(f"检查登录状态失败: {status_result.get('message', '未知错误')}") # 等待5秒后重试 time.sleep(5) attempt += 1 except KeyboardInterrupt: print("\n用户取消登录") return False, "用户取消登录" except Exception as e: self.logger.error(f"检查登录状态时发生错误: {e}") time.sleep(5) attempt += 1 # 超时 error_msg = "登录超时，请重新尝试" print(f"\n❌ {error_msg}") return False, error_msg except APIException as e: error_msg = f"API调用失败: {e}" self.logger.error(error_msg) return False, error_msg except Exception as e: error_msg = f"登录过程中发生未知错误: {e}" self.logger.error(error_msg) return False, error_msg def save_cookie(self, cookie: str) -> bool: """保存Cookie到文件 Args: cookie: Cookie字符串 Returns: 是否保存成功 """ try: # 备份现有Cookie（如果存在） if self.cookie_manager.cookie_file.exists(): try: backup_path = self.cookie_manager.backup_cookie() self.logger.info(f"已备份现有Cookie到: {backup_path}") except Exception as e: self.logger.warning(f"备份Cookie失败: {e}") # 保存新Cookie success = self.cookie_manager.write_cookie(cookie) if success: # 验证保存的Cookie if self.cookie_manager.is_cookie_valid(): self.logger.info("Cookie保存并验证成功") return True else: self.logger.warning("Cookie保存成功但验证失败") return False else: self.logger.error("Cookie保存失败") return False except CookieException as e: self.logger.error(f"Cookie操作失败: {e}") return False except Exception as e: self.logger.error(f"保存Cookie时发生错误: {e}") return False def show_login_info(self) -> None: """显示登录信息""" try: info = self.cookie_manager.get_cookie_info() print("\n=== 登录状态信息 ===") print(f"Cookie文件: {info['file_path']}") print(f"文件存在: {'是' if info['file_exists'] else '否'}") print(f"Cookie数量: {info['cookie_count']}") print(f"登录状态: {'有效' if info['is_valid'] else '无效'}") if info.get('last_modified'): print(f"最后更新: {info['last_modified']}") if info['is_valid']: present_cookies = info.get('important_cookies_present', []) print(f"重要Cookie: {', '.join(present_cookies)}") else: missing_cookies = info.get('missing_important_cookies', []) if missing_cookies: print(f"缺少Cookie: {', '.join(missing_cookies)}") except Exception as e: print(f"获取登录信息失败: {e}") def logout(self) -> bool: """登出（清除Cookie） Returns: 是否登出成功 """ try: # 备份Cookie if self.cookie_manager.cookie_file.exists(): try: backup_path = self.cookie_manager.backup_cookie("logout") print(f"Cookie已备份到: {backup_path}") except Exception as e: self.logger.warning(f"备份Cookie失败: {e}") # 清除Cookie success = self.cookie_manager.clear_cookie() if success: print("已成功登出") return True else: print("登出失败") return False except Exception as e: print(f"登出时发生错误: {e}") return False def main(): """主函数""" # 配置日志 logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s' ) client = QRLoginClient() # 解析命令行参数 if len(sys.argv) > 1: command = sys.argv[1].lower() if command == 'login': # 执行登录 success, error = client.interactive_login() if success: print("\n登录完成！") client.show_login_info() sys.exit(0) else: print(f"\n登录失败: {error}") sys.exit(1) elif command == 'status' or command == 'info': # 显示登录状态 client.show_login_info() sys.exit(0) elif command == 'logout': # 登出 success = client.logout() sys.exit(0 if success else 1) elif command == 'help' or command == '-h' or command == '--help': # 显示帮助 print("网易云音乐二维码登录工具") print("\n用法:") print(" python qr_login.py [命令]") print("\n命令:") print(" login - 执行二维码登录") print(" status - 显示登录状态") print(" logout - 登出（清除Cookie）") print(" help - 显示此帮助信息") print("\n如果不提供命令，将进入交互模式") sys.exit(0) else: print(f"未知命令: {command}") print("使用 'python qr_login.py help' 查看帮助") sys.exit(1) # 交互模式 try: while True: print("\n=== 网易云音乐登录工具 ===") print("1. 二维码登录") print("2. 查看登录状态") print("3. 登出") print("4. 退出") choice = input("\n请选择操作 (1-4): ").strip() if choice == '1': success, error = client.interactive_login() if success: print("\n登录成功！") client.show_login_info() else: print(f"\n登录失败: {error}") elif choice == '2': client.show_login_info() elif choice == '3': client.logout() elif choice == '4': print("再见！") break else: print("无效选择，请重试") except KeyboardInterrupt: print("\n\n程序已退出") sys.exit(0) except Exception as e: print(f"\n程序运行出错: {e}") sys.exit(1) if __name__ == "__main__": main()