""" 平台配置管理 包含各平台的特定配置信息。 """ from typing import Dict, Any from ..core.models import PlatformType, PlatformConfig, XIAOHONGSHU_CONFIG, DOUYIN_CONFIG class PlatformConfigs: """平台配置管理类""" def __init__(self): self._configs: Dict[PlatformType, PlatformConfig] = { PlatformType.XIAOHONGSHU: XIAOHONGSHU_CONFIG, PlatformType.DOUYIN: DOUYIN_CONFIG } self._custom_configs: Dict[PlatformType, PlatformConfig] = {} def get_config(self, platform: PlatformType) -> PlatformConfig: """获取平台配置""" return self._custom_configs.get(platform, self._configs.get(platform)) def set_config(self, platform: PlatformType, config: PlatformConfig): """设置平台配置""" self._custom_configs[platform] = config def update_config(self, platform: PlatformType, updates: Dict[str, Any]): """更新平台配置""" current_config = self.get_config(platform) if current_config: # 创建新的配置对象 updated_config = PlatformConfig( name=updates.get("name", current_config.name), login_url=updates.get("login_url", current_config.login_url), supported_content_types=updates.get("supported_content_types", current_config.supported_content_types), max_file_size=updates.get("max_file_size", current_config.max_file_size), max_duration=updates.get("max_duration", current_config.max_duration), supported_formats=updates.get("supported_formats", current_config.supported_formats), extra_config={**current_config.extra_config, **updates.get("extra_config", {})} ) self.set_config(platform, updated_config) def get_all_configs(self) -> Dict[PlatformType, PlatformConfig]: """获取所有平台配置""" return {**self._configs, **self._custom_configs} def reset_config(self, platform: PlatformType): """重置平台配置为默认值""" if platform in self._custom_configs: del self._custom_configs[platform] def load_from_dict(self, config_dict: Dict[str, Any]): """从字典加载配置""" for platform_name, config_data in config_dict.items(): try: platform = PlatformType(platform_name) config = PlatformConfig.from_dict(config_data) self.set_config(platform, config) except ValueError: continue # 忽略未知的平台名称 # 全局平台配置实例 platform_configs = PlatformConfigs() def get_platform_config(platform: PlatformType) -> PlatformConfig: """获取平台配置的便捷函数""" return platform_configs.get_config(platform) # 平台特定的选择器配置 PLATFORM_SELECTORS = { PlatformType.XIAOHONGSHU: { "login": { "qr_code": "[data-testid='qr-code']", "qr_code_container": ".qr-code-container", "login_success_indicators": [ "[data-testid='avatar']", ".creator-center-avatar", ".user-info", ".user-avatar" ] }, "publish": { "image_note": { "title_input": "input[placeholder*='标题']", "content_input": "textarea[placeholder*='内容']", "image_upload": "[data-testid='image-upload']", "tag_input": "input[placeholder*='标签']", "publish_button": "[data-testid='publish-btn']", "success_indicator": ".publish-success", "upload_progress": ".upload-progress" }, "video_note": { "title_input": "input[placeholder*='标题']", "content_input": "textarea[placeholder*='内容']", "video_upload": "[data-testid='video-upload']", "tag_input": "input[placeholder*='标签']", "publish_button": "[data-testid='publish-btn']", "success_indicator": ".publish-success", "upload_progress": ".video-upload-progress" } } }, PlatformType.DOUYIN: { "login": { "qr_code": ".login-qr-code", "qr_code_container": ".qr-container", "login_success_indicators": [ ".user-avatar", ".creator-workspace", ".publish-btn", ".user-info" ] }, "publish": { "video": { "title_input": "input[placeholder*='标题']", "description_input": "textarea[placeholder*='描述']", "video_upload": "[data-testid='video-upload']", "tag_input": "input[placeholder*='#']", "publish_button": "[data-testid='publish-btn']", "success_indicator": ".upload-success", "upload_progress": ".upload-progress-bar" } } } } def get_selectors(platform: PlatformType, page_type: str, content_type: str = None) -> Dict[str, str]: """获取平台选择器配置""" platform_selectors = PLATFORM_SELECTORS.get(platform, {}) page_selectors = platform_selectors.get(page_type, {}) if content_type and page_type == "publish": content_selectors = page_selectors.get(content_type, {}) return content_selectors return page_selectors # 平台特定的等待时间配置（秒） PLATFORM_WAIT_TIMES = { PlatformType.XIAOHONGSHU: { "login": { "qr_code_load": 10, "login_success": 5, "login_timeout": 300 # 5分钟 }, "publish": { "page_load": 10, "image_upload": 30, "video_upload": 120, "publish_success": 10, "operation_delay": (1, 3) } }, PlatformType.DOUYIN: { "login": { "qr_code_load": 10, "login_success": 5, "login_timeout": 300 # 5分钟 }, "publish": { "page_load": 10, "video_upload": 180, "publish_success": 15, "operation_delay": (2, 4) } } } def get_wait_times(platform: PlatformType, operation: str) -> Dict[str, Any]: """获取平台等待时间配置""" return PLATFORM_WAIT_TIMES.get(platform, {}).get(operation, {}) # 平台特定的URL配置 PLATFORM_URLS = { PlatformType.XIAOHONGSHU: { "login": "https://creator.xiaohongshu.com/", "image_note": "https://creator.xiaohongshu.com/publish/publish", "video_note": "https://creator.xiaohongshu.com/publish/video", "home": "https://creator.xiaohongshu.com/creator-center" }, PlatformType.DOUYIN: { "login": "https://creator.douyin.com/creator-micro/home", "video_upload": "https://creator.douyin.com/creator-micro/content/upload", "home": "https://creator.douyin.com/creator-micro/home", "content_manage": "https://creator.douyin.com/creator-micro/content/manage" } } def get_platform_url(platform: PlatformType, page_type: str) -> str: """获取平台URL配置""" return PLATFORM_URLS.get(platform, {}).get(page_type, "")