""" 项目素材管理服务 """ import json import hashlib from pathlib import Path from typing import List, Dict, Any, Optional from datetime import datetime from python_core.utils.logger import logger class ProjectMaterialService: """项目素材管理服务""" def __init__(self): logger.info("ProjectMaterialService 初始化完成") def import_video_materials( self, video_segments: List[Any], project_id: str, project_directory: Path, source_video_path: str, material_tags: Optional[List[str]] = None ) -> Dict[str, Any]: """ 导入视频素材到项目 Args: video_segments: 视频片段列表 project_id: 项目ID project_directory: 项目目录 source_video_path: 源视频路径 material_tags: 素材标签 Returns: Dict: 导入结果 """ try: # 创建项目素材目录 uncategorized_dir = project_directory / "未分类文件夹" uncategorized_dir.mkdir(parents=True, exist_ok=True) # 加载现有的项目素材列表 material_list_file = project_directory / "project_material.json" existing_materials = self._load_material_list(material_list_file) # 处理视频片段 new_materials = [] imported_count = 0 skipped_count = 0 for segment in video_segments: if not segment.success: continue try: result = self._import_single_segment( segment=segment, uncategorized_dir=uncategorized_dir, existing_materials=existing_materials, project_id=project_id, source_video_path=source_video_path, material_tags=material_tags or [] ) if result["imported"]: new_materials.append(result["material_info"]) imported_count += 1 logger.info(f"✅ 素材已导入: {result['original_filename']} -> {result['target_filename']}") else: skipped_count += 1 logger.info(f"📋 素材已存在，跳过: {result['original_filename']}") except Exception as e: logger.error(f"❌ 导入素材失败 {segment.output_path.name}: {e}") skipped_count += 1 # 更新项目素材列表 if new_materials: existing_materials.extend(new_materials) self._save_material_list(material_list_file, existing_materials) logger.info(f"📝 项目素材列表已更新: 新增 {len(new_materials)} 个素材") return { "success": True, "imported_count": imported_count, "skipped_count": skipped_count, "total_count": len(video_segments), "new_materials": new_materials } except Exception as e: logger.error(f"❌ 项目素材导入失败: {e}") return { "success": False, "error": str(e), "imported_count": 0, "skipped_count": 0, "total_count": len(video_segments) } def _import_single_segment( self, segment: Any, uncategorized_dir: Path, existing_materials: List[Dict], project_id: str, source_video_path: str, material_tags: List[str] ) -> Dict[str, Any]: """导入单个视频片段""" # 复制文件到项目目录 import shutil temp_target_path = uncategorized_dir / f"temp_{segment.output_path.name}" shutil.copy2(segment.output_path, temp_target_path) # 计算复制后文件的MD5 file_md5 = self._calculate_file_md5(temp_target_path) # 检查是否已存在相同MD5的素材 if any(material.get('md5') == file_md5 for material in existing_materials): # 删除临时文件 temp_target_path.unlink() return { "imported": False, "original_filename": segment.output_path.name, "reason": "duplicate_md5" } # 重命名为最终文件名 target_filename = f"{file_md5}.mp4" target_path = uncategorized_dir / target_filename temp_target_path.rename(target_path) # 创建素材信息 material_info = { "id": file_md5, "md5": file_md5, "original_filename": segment.output_path.name, "filename": target_filename, "file_path": str(target_path), "relative_path": f"未分类文件夹/{target_filename}", "file_size": segment.file_size, "duration": segment.duration, "start_time": segment.start_time, "end_time": segment.end_time, "scene_index": segment.scene_index, "tags": material_tags.copy(), "created_at": datetime.now().isoformat(), "use_count": 0, "source_video": source_video_path, "project_id": project_id } return { "imported": True, "material_info": material_info, "original_filename": segment.output_path.name, "target_filename": target_filename } def _load_material_list(self, material_list_file: Path) -> List[Dict]: """加载项目素材列表""" if not material_list_file.exists(): return [] try: with open(material_list_file, 'r', encoding='utf-8') as f: return json.load(f) except Exception as e: logger.warning(f"⚠️ 读取现有素材列表失败: {e}") return [] def _save_material_list(self, material_list_file: Path, materials: List[Dict]): """保存项目素材列表""" try: with open(material_list_file, 'w', encoding='utf-8') as f: json.dump(materials, f, ensure_ascii=False, indent=2) except Exception as e: logger.error(f"❌ 保存项目素材列表失败: {e}") raise def _calculate_file_md5(self, file_path: Path) -> str: """计算文件MD5值""" hash_md5 = hashlib.md5() try: with open(file_path, "rb") as f: for chunk in iter(lambda: f.read(4096), b""): hash_md5.update(chunk) return hash_md5.hexdigest() except Exception as e: logger.error(f"❌ 计算MD5失败 {file_path}: {e}") # 如果计算MD5失败，使用文件名和大小作为备用标识 return hashlib.md5(f"{file_path.name}_{file_path.stat().st_size}".encode()).hexdigest() def get_project_materials(self, project_directory: Path) -> List[Dict]: """获取项目素材列表""" material_list_file = project_directory / "project_material.json" return self._load_material_list(material_list_file) def add_material_tags(self, project_directory: Path, material_id: str, tags: List[str]) -> bool: """为素材添加标签""" try: material_list_file = project_directory / "project_material.json" materials = self._load_material_list(material_list_file) for material in materials: if material.get('id') == material_id: existing_tags = set(material.get('tags', [])) new_tags = existing_tags.union(set(tags)) material['tags'] = list(new_tags) self._save_material_list(material_list_file, materials) logger.info(f"✅ 素材标签已更新: {material_id}") return True logger.warning(f"⚠️ 未找到素材: {material_id}") return False except Exception as e: logger.error(f"❌ 添加素材标签失败: {e}") return False def remove_material(self, project_directory: Path, material_id: str) -> bool: """删除项目素材""" try: material_list_file = project_directory / "project_material.json" materials = self._load_material_list(material_list_file) # 查找要删除的素材 material_to_remove = None for i, material in enumerate(materials): if material.get('id') == material_id: material_to_remove = materials.pop(i) break if material_to_remove: # 删除文件 file_path = Path(material_to_remove['file_path']) if file_path.exists(): file_path.unlink() logger.info(f"🗑️ 已删除素材文件: {file_path.name}") # 更新素材列表 self._save_material_list(material_list_file, materials) logger.info(f"✅ 素材已删除: {material_id}") return True else: logger.warning(f"⚠️ 未找到素材: {material_id}") return False except Exception as e: logger.error(f"❌ 删除素材失败: {e}") return False def get_material_stats(self, project_directory: Path) -> Dict[str, Any]: """获取项目素材统计信息""" try: materials = self.get_project_materials(project_directory) total_count = len(materials) total_size = sum(material.get('file_size', 0) for material in materials) total_duration = sum(material.get('duration', 0) for material in materials) # 按标签统计 tag_stats = {} for material in materials: for tag in material.get('tags', []): tag_stats[tag] = tag_stats.get(tag, 0) + 1 # 按使用次数统计 used_count = sum(1 for material in materials if material.get('use_count', 0) > 0) unused_count = total_count - used_count return { "total_count": total_count, "total_size": total_size, "total_duration": total_duration, "used_count": used_count, "unused_count": unused_count, "tag_stats": tag_stats } except Exception as e: logger.error(f"❌ 获取素材统计失败: {e}") return { "total_count": 0, "total_size": 0, "total_duration": 0, "used_count": 0, "unused_count": 0, "tag_stats": {} }