#!/usr/bin/env python3 # -*- coding: utf-8 -*- ""...
Erstellt am: 22. Juni 2025
Frage
#!/usr/bin/env python3
-- coding: utf-8 --
"""
商务车素材管理系统 - 完整版本
视频文件标签管理和预览工具
"""
import tkinter as tk
from tkinter import ttk, filedialog, messagebox, simpledialog
import json
import os
from pathlib import Path
import threading
from typing import Dict, List, Set
import subprocess
import cv2
from PIL import Image, ImageTk
import tempfile
import shutil
class VideoAssetManager:
def init(self):
self.root = tk.Tk()
self.root.title("商务车素材管理系统")
self.root.geometry("1400x900")
text# 配置文件路径 self.config_dir = Path.home() / ".video_asset_manager" self.metadata_file = self.config_dir / "metadata.json" self.current_dir_file = self.config_dir / "current_dir" self.thumbnails_dir = self.config_dir / "thumbnails" # 初始化配置 self.init_config() # 当前工作目录 self.current_dir = self.load_current_dir() # 预览设置 self.thumbnail_size = (200, 150) # 默认缩略图大小 self.preview_cache = {} # 缩略图缓存 # 多选相关变量 self.selected_files = set() # 剪贴板用于复制标签 self.clipboard_tags = [] # 预设标签 - 根据你的需求优化 self.preset_tags = { "全车视角": ["正面", "侧面", "后面"], "前排部位": ["360大连屏", "方向盘", "脚垫", "门板", "工作台", "包覆"], "中排部位": ["中排座椅", "旋转座椅", "脚踏", "地板", "电视", "会客", "多功能桌板"], "后排部位": ["三排座椅", "沙发床", "窗帘"], "其他部位": ["星空顶", "灯", "踏板", "滑轨", "空间"], "拍摄角度": ["正面", "侧面", "仰视", "俯视", "左移右", "右移左", "后拉", "前推"] } # 创建界面 self.create_widgets() self.bind_keyboard_shortcuts() self.refresh_file_list() def init_config(self): """初始化配置目录""" self.config_dir.mkdir(exist_ok=True) self.thumbnails_dir.mkdir(exist_ok=True) if not self.metadata_file.exists(): with open(self.metadata_file, 'w', encoding='utf-8') as f: json.dump({}, f) def load_current_dir(self): """加载当前工作目录""" if self.current_dir_file.exists(): return self.current_dir_file.read_text().strip() return "" def save_current_dir(self, directory): """保存当前工作目录""" self.current_dir_file.write_text(directory) self.current_dir = directory def load_metadata(self): """加载元数据""" try: with open(self.metadata_file, 'r', encoding='utf-8') as f: return json.load(f) except (FileNotFoundError, json.JSONDecodeError): return {} def save_metadata(self, metadata): """保存元数据""" with open(self.metadata_file, 'w', encoding='utf-8') as f: json.dump(metadata, f, ensure_ascii=False, indent=2) def bind_keyboard_shortcuts(self): """绑定键盘快捷键""" self.root.bind('<Command-equal>', self.zoom_in) # Cmd + = 放大 self.root.bind('<Command-minus>', self.zoom_out) # Cmd + - 缩小 self.root.bind('<Command-a>', self.select_all) # Cmd + A 全选 self.root.bind('<space>', self.preview_selected) # 空格键预览 self.root.bind('<Delete>', self.remove_tags_from_selected) # 删除键删除标签 self.root.bind('<Command-c>', self.copy_tags) # Cmd + C 复制标签 self.root.bind('<Command-v>', self.paste_tags) # Cmd + V 粘贴标签 def create_widgets(self): """创建界面组件""" # 主框架 main_frame = ttk.Frame(self.root) main_frame.pack(fill=tk.BOTH, expand=True, padx=15, pady=15) # 顶部工具栏 self.create_toolbar(main_frame) # 中间内容区域 content_frame = ttk.Frame(main_frame) content_frame.pack(fill=tk.BOTH, expand=True, pady=(15, 0)) # 左侧文件列表 self.create_file_list(content_frame) # 右侧标签管理 self.create_tag_panel(content_frame) # 底部状态栏 self.create_status_bar(main_frame) def create_toolbar(self, parent): """创建工具栏""" toolbar = ttk.Frame(parent) toolbar.pack(fill=tk.X, pady=(0, 15)) # 左侧按钮组 left_buttons = ttk.Frame(toolbar) left_buttons.pack(side=tk.LEFT) ttk.Button(left_buttons, text="📁 选择目录", command=self.select_directory).pack(side=tk.LEFT, padx=(0, 10)) ttk.Button(left_buttons, text="🔍 生成预览", command=self.generate_thumbnails_async).pack(side=tk.LEFT, padx=(0, 10)) ttk.Button(left_buttons, text="📊 统计报告", command=self.show_statistics).pack(side=tk.LEFT, padx=(0, 10)) ttk.Button(left_buttons, text="💾 导出数据", command=self.export_metadata).pack(side=tk.LEFT, padx=(0, 10)) # 中间目录显示 self.dir_label = ttk.Label(toolbar, text=f"📂 {self.current_dir or '未选择目录'}") self.dir_label.pack(side=tk.LEFT, padx=(20, 0)) # 右侧搜索和缩放控制 right_frame = ttk.Frame(toolbar) right_frame.pack(side=tk.RIGHT) # 缩放控制 zoom_frame = ttk.Frame(right_frame) zoom_frame.pack(side=tk.RIGHT, padx=(0, 15)) ttk.Label(zoom_frame, text="缩略图:").pack(side=tk.LEFT) ttk.Button(zoom_frame, text="➖", width=3, command=self.zoom_out).pack(side=tk.LEFT, padx=2) ttk.Button(zoom_frame, text="➕", width=3, command=self.zoom_in).pack(side=tk.LEFT, padx=2) # 搜索框 search_frame = ttk.Frame(right_frame) search_frame.pack(side=tk.RIGHT) ttk.Label(search_frame, text="🔍").pack(side=tk.LEFT) self.search_var = tk.StringVar() search_entry = ttk.Entry(search_frame, textvariable=self.search_var, width=25) search_entry.pack(side=tk.LEFT, padx=(5, 0)) search_entry.bind('<KeyRelease>', self.on_search) ttk.Button(search_frame, text="✖️", width=3, command=self.clear_search).pack(side=tk.LEFT, padx=(5, 0)) def create_file_list(self, parent): """创建文件列表""" left_frame = ttk.LabelFrame(parent, text="📹 视频文件列表") left_frame.pack(side=tk.LEFT, fill=tk.BOTH, expand=True, padx=(0, 10)) # 文件列表容器 list_container = ttk.Frame(left_frame) list_container.pack(fill=tk.BOTH, expand=True, padx=15, pady=15) # 创建Canvas和Scrollbar用于自定义滚动 self.canvas = tk.Canvas(list_container, bg='white', highlightthickness=0) self.scrollbar = ttk.Scrollbar(list_container, orient="vertical", command=self.canvas.yview) self.scrollable_frame = ttk.Frame(self.canvas) # 绑定滚动事件 self.scrollable_frame.bind( "<Configure>", lambda e: self.canvas.configure(scrollregion=self.canvas.bbox("all")) ) self.canvas.create_window((0, 0), window=self.scrollable_frame, anchor="nw") self.canvas.configure(yscrollcommand=self.scrollbar.set) # 布局 self.canvas.pack(side="left", fill="both", expand=True) self.scrollbar.pack(side="right", fill="y") # 绑定鼠标滚轮 self.canvas.bind("<MouseWheel>", self.on_mousewheel) self.root.bind("<Button-1>", self.on_canvas_click) # 存储文件项目的字典 self.file_items = {} def create_tag_panel(self, parent): """创建标签管理面板""" right_frame = ttk.LabelFrame(parent, text="🏷️ 标签管理") right_frame.pack(side=tk.RIGHT, fill=tk.Y, padx=(10, 0)) right_frame.configure(width=450) # 选中文件信息 info_frame = ttk.Frame(right_frame) info_frame.pack(fill=tk.X, padx=15, pady=15) self.selected_count_label = ttk.Label(info_frame, text="未选择文件", foreground='#007AFF') self.selected_count_label.pack() # 预设标签区域 self.create_preset_tags(right_frame) # 自定义标签 self.create_custom_tags(right_frame) # 当前选中文件的标签 self.create_current_tags(right_frame) # 批量操作按钮 self.create_batch_operations(right_frame) def create_preset_tags(self, parent): """创建预设标签区域""" preset_frame = ttk.LabelFrame(parent, text="⚡ 快速标签") preset_frame.pack(fill=tk.X, padx=15, pady=(0, 10)) # 创建可滚动的标签区域 canvas = tk.Canvas(preset_frame, height=300, bg='white', highlightthickness=0) scrollbar_tags = ttk.Scrollbar(preset_frame, orient="vertical", command=canvas.yview) scrollable_tags = ttk.Frame(canvas) scrollable_tags.bind( "<Configure>", lambda e: canvas.configure(scrollregion=canvas.bbox("all")) ) canvas.create_window((0, 0), window=scrollable_tags, anchor="nw") canvas.configure(yscrollcommand=scrollbar_tags.set) # 创建标签按钮 for category, tags in self.preset_tags.items(): cat_frame = ttk.LabelFrame(scrollable_tags, text=category) cat_frame.pack(fill=tk.X, padx=5, pady=5) # 按钮网格 for i, tag in enumerate(tags): row = i // 3 col = i % 3 btn = ttk.Button(cat_frame, text=tag, width=12, command=lambda t=tag: self.add_tag_to_selected(t)) btn.grid(row=row, column=col, padx=3, pady=3, sticky='ew') # 设置列权重以实现均匀分布 cat_frame.grid_columnconfigure(col, weight=1) canvas.pack(side="left", fill="both", expand=True, padx=5, pady=5) scrollbar_tags.pack(side="right", fill="y") def create_custom_tags(self, parent): """创建自定义标签区域""" custom_frame = ttk.LabelFrame(parent, text="✏️ 自定义标签") custom_frame.pack(fill=tk.X, padx=15, pady=(0, 10)) # 标签输入 input_frame = ttk.Frame(custom_frame) input_frame.pack(fill=tk.X, padx=10, pady=10) self.custom_tag_var = tk.StringVar() tag_entry = ttk.Entry(input_frame, textvariable=self.custom_tag_var) tag_entry.pack(side=tk.LEFT, fill=tk.X, expand=True) tag_entry.bind('<Return>', lambda e: self.add_custom_tag()) ttk.Button(input_frame, text="➕ 添加", command=self.add_custom_tag).pack(side=tk.RIGHT, padx=(10, 0)) def create_current_tags(self, parent): """创建当前选中文件标签显示区域""" current_frame = ttk.LabelFrame(parent, text="📋 当前标签") current_frame.pack(fill=tk.BOTH, expand=True, padx=15, pady=(0, 10)) # 标签列表 self.current_tags_listbox = tk.Listbox(current_frame, height=6) self.current_tags_listbox.pack(fill=tk.BOTH, expand=True, padx=10, pady=10) # 删除按钮 ttk.Button(current_frame, text="🗑️ 删除选中标签", command=self.remove_selected_tag).pack(pady=(0, 10)) def create_batch_operations(self, parent): """创建批量操作区域""" batch_frame = ttk.LabelFrame(parent, text="⚡ 批量操作") batch_frame.pack(fill=tk.X, padx=15, pady=(0, 10)) button_frame = ttk.Frame(batch_frame) button_frame.pack(fill=tk.X, padx=10, pady=10) ttk.Button(button_frame, text="🏷️ 批量添加标签", command=self.batch_add_tags).pack(fill=tk.X, pady=2) ttk.Button(button_frame, text="🗑️ 批量删除标签", command=self.batch_remove_tags).pack(fill=tk.X, pady=2) ttk.Button(button_frame, text="📋 复制标签", command=self.copy_tags).pack(fill=tk.X, pady=2) ttk.Button(button_frame, text="📋 粘贴标签", command=self.paste_tags).pack(fill=tk.X, pady=2) ttk.Button(button_frame, text="🔄 清除所有标签", command=self.clear_all_tags).pack(fill=tk.X, pady=2) def create_status_bar(self, parent): """创建状态栏""" status_frame = ttk.Frame(parent) status_frame.pack(side=tk.BOTTOM, fill=tk.X, pady=(15, 0)) self.status_bar = ttk.Label(status_frame, text="✅ 就绪") self.status_bar.pack(side=tk.LEFT) # 添加帮助提示 help_text = "💡 提示: Cmd+=/- 缩放 | 空格预览 | Cmd+A全选 | 拖拽多选" ttk.Label(status_frame, text=help_text, foreground='#666666').pack(side=tk.RIGHT) def select_directory(self): """选择目录""" directory = filedialog.askdirectory(title="选择视频文件目录") if directory: self.save_current_dir(directory) self.dir_label.config(text=f"📂 {directory}") self.refresh_file_list() self.update_status(f"📁 已选择目录: {directory}") def generate_thumbnail(self, video_path): """生成视频缩略图""" thumbnail_path = self.thumbnails_dir / f"{Path(video_path).stem}.jpg" if thumbnail_path.exists(): return str(thumbnail_path) try: # 使用OpenCV生成缩略图 cap = cv2.VideoCapture(str(video_path)) if not cap.isOpened(): return None # 跳到视频的10%位置 frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) cap.set(cv2.CAP_PROP_POS_FRAMES, frame_count // 10) ret, frame = cap.read() cap.release() if ret: # 调整大小并保存 height, width = frame.shape[:2] aspect_ratio = width / height if aspect_ratio > 1: new_width = self.thumbnail_size[0] new_height = int(new_width / aspect_ratio) else: new_height = self.thumbnail_size[1] new_width = int(new_height * aspect_ratio) resized = cv2.resize(frame, (new_width, new_height)) cv2.imwrite(str(thumbnail_path), resized) return str(thumbnail_path) except Exception as e: print(f"生成缩略图失败: {e}") return None def generate_thumbnails_async(self): """异步生成所有缩略图""" if not self.current_dir: messagebox.showwarning("警告", "请先选择目录") return self.update_status("🔄 正在生成缩略图...") def generate_all(): video_extensions = {'.mp4', '.mov', '.avi', '.mkv', '.m4v', '.wmv', '.flv'} video_files = [] for file_path in Path(self.current_dir).rglob('*'): if file_path.is_file() and file_path.suffix.lower() in video_extensions: video_files.append(file_path) for i, video_path in enumerate(video_files): self.generate_thumbnail(video_path) progress = f"📊 生成进度: {i+1}/{len(video_files)}" self.root.after(0, lambda p=progress: self.update_status(p)) self.root.after(0, lambda: self.update_status("✅ 缩略图生成完成")) self.root.after(0, self.refresh_file_list) threading.Thread(target=generate_all, daemon=True).start() def create_file_item(self, file_path, metadata_info): """创建单个文件项目""" relative_path = str(file_path.relative_to(self.current_dir)) # 主容器 item_frame = ttk.Frame(self.scrollable_frame, relief='solid', borderwidth=1) item_frame.pack(fill=tk.X, padx=5, pady=5) # 选择状态变量 selected_var = tk.BooleanVar() # 左侧缩略图和选择框 left_frame = ttk.Frame(item_frame) left_frame.pack(side=tk.LEFT, padx=10, pady=10) # 选择框 check_btn = ttk.Checkbutton(left_frame, variable=selected_var, command=lambda: self.toggle_file_selection(relative_path, selected_var.get())) check_btn.pack(anchor='nw') # 缩略图 thumbnail_label = ttk.Label(left_frame, text="📹") thumbnail_label.pack(pady=(5, 0)) # 尝试加载缩略图 thumbnail_path = self.thumbnails_dir / f"{file_path.stem}.jpg" if thumbnail_path.exists(): try: img = Image.open(thumbnail_path) img = img.resize(self.thumbnail_size, Image.Resampling.LANCZOS) photo = ImageTk.PhotoImage(img) thumbnail_label.configure(image=photo, text="") thumbnail_label.image = photo # 保持引用 except Exception as e: print(f"加载缩略图失败: {e}") # 右侧文件信息 right_frame = ttk.Frame(item_frame) right_frame.pack(side=tk.LEFT, fill=tk.BOTH, expand=True, padx=10, pady=10) # 文件名 name_label = ttk.Label(right_frame, text=file_path.name, font=('TkDefaultFont', 12, 'bold')) name_label.pack(anchor='w') # 文件大小 try: size = file_path.stat().st_size size_str = self.format_file_size(size) except: size_str = "未知" size_label = ttk.Label(right_frame, text=f"📏 大小: {size_str}") size_label.pack(anchor='w', pady=(2, 0)) # 标签 tags = metadata_info.get('tags', []) if tags: tags_text = f"🏷️ 标签: {', '.join(tags)}" else: tags_text = "🏷️ 暂无标签" tags_label = ttk.Label(right_frame, text=tags_text, foreground='#007AFF' if tags else '#999999') tags_label.pack(anchor='w', pady=(2, 0)) # 绑定双击预览事件 for widget in [item_frame, left_frame, right_frame, name_label, size_label, tags_label]: widget.bind('<Double-Button-1>', lambda e, fp=file_path: self.preview_file(fp)) # 存储项目信息 self.file_items[relative_path] = { 'frame': item_frame, 'selected_var': selected_var, 'file_path': file_path, 'tags_label': tags_label } def format_file_size(self, size): """格式化文件大小""" for unit in ['B', 'KB', 'MB', 'GB']: if size < 1024.0: return f"{size:.1f} {unit}" size /= 1024.0 return f"{size:.1f} TB" def toggle_file_selection(self, relative_path, selected): """切换文件选择状态""" if selected: self.selected_files.add(relative_path) else: self.selected_files.discard(relative_path) self.update_selection_display() def update_selection_display(self): """更新选中文件显示""" count = len(self.selected_files) if count == 0: self.selected_count_label.config(text="未选择文件") elif count == 1: filename = Path(list(self.selected_files)[0]).name self.selected_count_label.config(text=f"选中: {filename}") else: self.selected_count_label.config(text=f"选中 {count} 个文件") # 更新当前标签显示 if count == 1: self.update_current_tags_display(list(self.selected_files)[0]) else: self.current_tags_listbox.delete(0, tk.END) def update_current_tags_display(self, relative_path): """更新当前标签显示""" metadata = self.load_metadata() file_info = metadata.get(relative_path, {}) tags = file_info.get('tags', []) self.current_tags_listbox.delete(0, tk.END) for tag in tags: self.current_tags_listbox.insert(tk.END, tag) def refresh_file_list(self): """刷新文件列表""" if not self.current_dir or not os.path.exists(self.current_dir): return # 清空现有项目 for item_info in self.file_items.values(): item_info['frame'].destroy() self.file_items.clear() self.selected_files.clear() # 加载元数据 metadata = self.load_metadata() # 扫描视频文件 video_extensions = {'.mp4', '.mov', '.avi', '.mkv', '.m4v', '.wmv', '.flv'} video_files = [] for file_path in Path(self.current_dir).rglob('*'): if file_path.is_file() and file_path.suffix.lower() in video_extensions: video_files.append(file_path) # 按文件名排序 video_files.sort(key=lambda x: x.name.lower()) # 创建文件项目 for file_path in video_files: relative_path = str(file_path.relative_to(self.current_dir)) file_info = metadata.get(relative_path, {}) self.create_file_item(file_path, file_info) self.update_status(f"📊 已加载 {len(video_files)} 个视频文件") self.update_selection_display() def on_search(self, event): """搜索功能""" search_text = self.search_var.get().strip().lower() for relative_path, item_info in self.file_items.items(): file_path = item_info['file_path'] if not search_text: # 显示所有文件 item_info['frame'].pack(fill=tk.X, padx=5, pady=5) else: # 检查文件名或标签是否匹配 metadata = self.load_metadata() file_metadata = metadata.get(relative_path, {}) tags = file_metadata.get('tags', []) filename_match = search_text in file_path.name.lower() tag_match = any(search_text in tag.lower() for tag in tags) if filename_match or tag_match: item_info['frame'].pack(fill=tk.X, padx=5, pady=5) else: item_info['frame'].pack_forget() def clear_search(self): """清除搜索""" self.search_var.set("") self.on_search(None) def zoom_in(self, event=None): """放大缩略图""" current_width, current_height = self.thumbnail_size new_width = min(current_width + 50, 400) new_height = int(new_width * 3/4) self.thumbnail_size = (new_width, new_height) self.refresh_file_list() self.update_status(f"🔍 缩略图大小: {new_width}x{new_height}") def zoom_out(self, event=None): """缩小缩略图""" current_width, current_height = self.thumbnail_size new_width = max(current_width - 50, 100) new_height = int(new_width * 3/4) self.thumbnail_size = (new_width, new_height) self.refresh_file_list() self.update_status(f"🔍 缩略图大小: {new_width}x{new_height}") def select_all(self, event=None): """全选文件
继续为我完成这个代码！
这是我第一版的代码：#!/usr/bin/env python3
-- coding: utf-8 --
"""
商务车素材管理系统 - 优化版本
支持大图预览、多选操作、优化分类、Mac风格UI
"""
import tkinter as tk
from tkinter import ttk, filedialog, messagebox
import json
import os
from pathlib import Path
import threading
from typing import Dict, List, Set
import subprocess
import cv2
from PIL import Image, ImageTk
import tempfile
class VideoAssetManager:
def init(self):
self.root = tk.Tk()
self.root.title("商务车素材管理系统")
self.root.geometry("1400x900")
text# 设置Mac风格样式 self.setup_mac_style() # 配置文件路径 self.config_dir = Path.home() / ".video_asset_manager" self.metadata_file = self.config_dir / "metadata.json" self.current_dir_file = self.config_dir / "current_dir" self.thumbnails_dir = self.config_dir / "thumbnails" # 初始化配置 self.init_config() # 当前工作目录 self.current_dir = self.load_current_dir() # 预览设置 self.thumbnail_size = (200, 150) # 默认缩略图大小 self.preview_cache = {} # 缩略图缓存 # 多选相关变量 self.selected_files = set() # 预设标签 - 根据你的需求优化 self.preset_tags = { "全车视角": ["正面", "侧面", "后面"], "前排部位": ["360大连屏", "方向盘", "脚垫", "门板", "工作台", "包覆"], "中排部位": ["中排座椅", "旋转座椅", "脚踏", "地板", "电视", "会客", "多功能桌板"], "后排部位": ["三排座椅", "沙发床", "窗帘"], "其他部位": ["星空顶", "灯", "踏板", "滑轨", "空间"], "拍摄角度": ["正面", "侧面", "仰视", "俯视", "左移右", "右移左", "后拉", "前推"] } # 创建界面 self.create_widgets() self.bind_keyboard_shortcuts() self.refresh_file_list() def setup_mac_style(self): """设置Mac风格样式""" style = ttk.Style() # 尝试使用macOS原生主题 try: style.theme_use('aqua') except: style.theme_use('clam') # 自定义样式 style.configure('Title.TLabel', font=('SF Pro Display', 16, 'bold')) style.configure('Heading.TLabel', font=('SF Pro Display', 12, 'bold')) style.configure('Mac.TButton', padding=(10, 5)) style.configure('Mac.TLabelFrame', relief='flat', borderwidth=1) # 设置整体背景色 self.root.configure(bg='#f0f0f0') def init_config(self): """初始化配置目录""" self.config_dir.mkdir(exist_ok=True) self.thumbnails_dir.mkdir(exist_ok=True) if not self.metadata_file.exists(): with open(self.metadata_file, 'w', encoding='utf-8') as f: json.dump({}, f) def load_current_dir(self): """加载当前工作目录""" if self.current_dir_file.exists(): return self.current_dir_file.read_text().strip() return "" def save_current_dir(self, directory): """保存当前工作目录""" self.current_dir_file.write_text(directory) self.current_dir = directory def bind_keyboard_shortcuts(self): """绑定键盘快捷键""" self.root.bind('<Command-equal>', self.zoom_in) # Cmd + = 放大 self.root.bind('<Command-minus>', self.zoom_out) # Cmd + - 缩小 self.root.bind('<Command-a>', self.select_all) # Cmd + A 全选 self.root.bind('<space>', self.preview_selected) # 空格键预览 self.root.bind('<Delete>', self.remove_tags_from_selected) # 删除键删除标签 def create_widgets(self): """创建界面组件""" # 主框架 main_frame = ttk.Frame(self.root) main_frame.pack(fill=tk.BOTH, expand=True, padx=15, pady=15) # 顶部工具栏 self.create_toolbar(main_frame) # 中间内容区域 content_frame = ttk.Frame(main_frame) content_frame.pack(fill=tk.BOTH, expand=True, pady=(15, 0)) # 左侧文件列表 self.create_file_list(content_frame) # 右侧标签管理 self.create_tag_panel(content_frame) # 底部状态栏 self.create_status_bar(main_frame) def create_toolbar(self, parent): """创建工具栏""" toolbar = ttk.Frame(parent) toolbar.pack(fill=tk.X, pady=(0, 15)) # 左侧按钮组 left_buttons = ttk.Frame(toolbar) left_buttons.pack(side=tk.LEFT) ttk.Button(left_buttons, text="📁 选择目录", style='Mac.TButton', command=self.select_directory).pack(side=tk.LEFT, padx=(0, 10)) ttk.Button(left_buttons, text="🔍 生成预览", style='Mac.TButton', command=self.generate_thumbnails_async).pack(side=tk.LEFT, padx=(0, 10)) # 中间目录显示 self.dir_label = ttk.Label(toolbar, text=f"📂 {self.current_dir or '未选择目录'}", style='Heading.TLabel') self.dir_label.pack(side=tk.LEFT, padx=(20, 0)) # 右侧搜索和缩放控制 right_frame = ttk.Frame(toolbar) right_frame.pack(side=tk.RIGHT) # 缩放控制 zoom_frame = ttk.Frame(right_frame) zoom_frame.pack(side=tk.RIGHT, padx=(0, 15)) ttk.Label(zoom_frame, text="缩略图:").pack(side=tk.LEFT) ttk.Button(zoom_frame, text="➖", width=3, command=self.zoom_out).pack(side=tk.LEFT, padx=2) ttk.Button(zoom_frame, text="➕", width=3, command=self.zoom_in).pack(side=tk.LEFT, padx=2) # 搜索框 search_frame = ttk.Frame(right_frame) search_frame.pack(side=tk.RIGHT) ttk.Label(search_frame, text="🔍").pack(side=tk.LEFT) self.search_var = tk.StringVar() search_entry = ttk.Entry(search_frame, textvariable=self.search_var, width=25) search_entry.pack(side=tk.LEFT, padx=(5, 0)) search_entry.bind('<KeyRelease>', self.on_search) ttk.Button(search_frame, text="✖️", width=3, command=self.clear_search).pack(side=tk.LEFT, padx=(5, 0)) def create_file_list(self, parent): """创建文件列表""" left_frame = ttk.LabelFrame(parent, text="📹 视频文件列表", style='Mac.TLabelFrame') left_frame.pack(side=tk.LEFT, fill=tk.BOTH, expand=True, padx=(0, 10)) # 文件列表容器 list_container = ttk.Frame(left_frame) list_container.pack(fill=tk.BOTH, expand=True, padx=15, pady=15) # 创建Canvas和Scrollbar用于自定义滚动 self.canvas = tk.Canvas(list_container, bg='white', highlightthickness=0) self.scrollbar = ttk.Scrollbar(list_container, orient="vertical", command=self.canvas.yview) self.scrollable_frame = ttk.Frame(self.canvas) # 绑定滚动事件 self.scrollable_frame.bind( "<Configure>", lambda e: self.canvas.configure(scrollregion=self.canvas.bbox("all")) ) self.canvas.create_window((0, 0), window=self.scrollable_frame, anchor="nw") self.canvas.configure(yscrollcommand=self.scrollbar.set) # 布局 self.canvas.pack(side="left", fill="both", expand=True) self.scrollbar.pack(side="right", fill="y") # 绑定鼠标滚轮 self.canvas.bind("<MouseWheel>", self.on_mousewheel) self.root.bind("<Button-1>", self.on_canvas_click) # 存储文件项目的字典 self.file_items = {} def create_tag_panel(self, parent): """创建标签管理面板""" right_frame = ttk.LabelFrame(parent, text="🏷️ 标签管理", style='Mac.TLabelFrame') right_frame.pack(side=tk.RIGHT, fill=tk.Y, padx=(10, 0)) right_frame.configure(width=450) # 选中文件信息 info_frame = ttk.Frame(right_frame) info_frame.pack(fill=tk.X, padx=15, pady=15) self.selected_count_label = ttk.Label(info_frame, text="未选择文件", style='Title.TLabel', foreground='#007AFF') self.selected_count_label.pack() # 预设标签区域 self.create_preset_tags(right_frame) # 自定义标签 self.create_custom_tags(right_frame) # 当前选中文件的标签 self.create_current_tags(right_frame) # 批量操作按钮 self.create_batch_operations(right_frame) def create_preset_tags(self, parent): """创建预设标签区域""" preset_frame = ttk.LabelFrame(parent, text="⚡ 快速标签", style='Mac.TLabelFrame') preset_frame.pack(fill=tk.X, padx=15, pady=(0, 10)) # 创建可滚动的标签区域 canvas = tk.Canvas(preset_frame, height=300, bg='white', highlightthickness=0) scrollbar_tags = ttk.Scrollbar(preset_frame, orient="vertical", command=canvas.yview) scrollable_tags = ttk.Frame(canvas) scrollable_tags.bind( "<Configure>", lambda e: canvas.configure(scrollregion=canvas.bbox("all")) ) canvas.create_window((0, 0), window=scrollable_tags, anchor="nw") canvas.configure(yscrollcommand=scrollbar_tags.set) # 创建标签按钮 for category, tags in self.preset_tags.items(): cat_frame = ttk.LabelFrame(scrollable_tags, text=category) cat_frame.pack(fill=tk.X, padx=5, pady=5) # 按钮网格 for i, tag in enumerate(tags): row = i // 3 col = i % 3 btn = ttk.Button(cat_frame, text=tag, width=12, command=lambda t=tag: self.add_tag_to_selected(t)) btn.grid(row=row, column=col, padx=3, pady=3, sticky='ew') # 设置列权重以实现均匀分布 cat_frame.grid_columnconfigure(col, weight=1) canvas.pack(side="left", fill="both", expand=True, padx=5, pady=5) scrollbar_tags.pack(side="right", fill="y") def create_custom_tags(self, parent): """创建自定义标签区域""" custom_frame = ttk.LabelFrame(parent, text="✏️ 自定义标签", style='Mac.TLabelFrame') custom_frame.pack(fill=tk.X, padx=15, pady=(0, 10)) # 标签输入 input_frame = ttk.Frame(custom_frame) input_frame.pack(fill=tk.X, padx=10, pady=10) self.custom_tag_var = tk.StringVar() tag_entry = ttk.Entry(input_frame, textvariable=self.custom_tag_var, font=('SF Pro Display', 11)) tag_entry.pack(side=tk.LEFT, fill=tk.X, expand=True) tag_entry.bind('<Return>', lambda e: self.add_custom_tag()) ttk.Button(input_frame, text="➕ 添加", style='Mac.TButton', command=self.add_custom_tag).pack(side=tk.RIGHT, padx=(10, 0)) def create_current_tags(self, parent): """创建当前选中文件标签显示区域""" current_frame = ttk.LabelFrame(parent, text="📋 当前标签", style='Mac.TLabelFrame') current_frame.pack(fill=tk.BOTH, expand=True, padx=15, pady=(0, 10)) # 标签列表 self.current_tags_listbox = tk.Listbox(current_frame, height=6, font=('SF Pro Display', 10)) self.current_tags_listbox.pack(fill=tk.BOTH, expand=True, padx=10, pady=10) # 删除按钮 ttk.Button(current_frame, text="🗑️ 删除选中标签", style='Mac.TButton', command=self.remove_selected_tag).pack(pady=(0, 10)) def create_batch_operations(self, parent): """创建批量操作区域""" batch_frame = ttk.LabelFrame(parent, text="⚡ 批量操作", style='Mac.TLabelFrame') batch_frame.pack(fill=tk.X, padx=15, pady=(0, 10)) button_frame = ttk.Frame(batch_frame) button_frame.pack(fill=tk.X, padx=10, pady=10) ttk.Button(button_frame, text="🏷️ 批量添加标签", style='Mac.TButton', command=self.batch_add_tags).pack(fill=tk.X, pady=2) ttk.Button(button_frame, text="🗑️ 批量删除标签", style='Mac.TButton', command=self.batch_remove_tags).pack(fill=tk.X, pady=2) ttk.Button(button_frame, text="📋 复制标签", style='Mac.TButton', command=self.copy_tags).pack(fill=tk.X, pady=2) def create_status_bar(self, parent): """创建状态栏""" status_frame = ttk.Frame(parent) status_frame.pack(side=tk.BOTTOM, fill=tk.X, pady=(15, 0)) self.status_bar = ttk.Label(status_frame, text="✅ 就绪", font=('SF Pro Display', 10)) self.status_bar.pack(side=tk.LEFT) # 添加帮助提示 help_text = "💡 提示: Cmd+=/- 缩放 | 空格预览 | Cmd+A全选 | 拖拽多选" ttk.Label(status_frame, text=help_text, font=('SF Pro Display', 9), foreground='#666666').pack(side=tk.RIGHT) def generate_thumbnail(self, video_path): """生成视频缩略图""" thumbnail_path = self.thumbnails_dir / f"{Path(video_path).stem}.jpg" if thumbnail_path.exists(): return str(thumbnail_path) try: # 使用OpenCV生成缩略图 cap = cv2.VideoCapture(str(video_path)) if not cap.isOpened(): return None # 跳到视频的10%位置 frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) cap.set(cv2.CAP_PROP_POS_FRAMES, frame_count // 10) ret, frame = cap.read() cap.release() if ret: # 调整大小并保存 height, width = frame.shape[:2] aspect_ratio = width / height if aspect_ratio > 1: new_width = self.thumbnail_size[0] new_height = int(new_width / aspect_ratio) else: new_height = self.thumbnail_size[1] new_width = int(new_height * aspect_ratio) resized = cv2.resize(frame, (new_width, new_height)) cv2.imwrite(str(thumbnail_path), resized) return str(thumbnail_path) except Exception as e: print(f"生成缩略图失败: {e}") return None def generate_thumbnails_async(self): """异步生成所有缩略图""" if not self.current_dir: messagebox.showwarning("警告", "请先选择目录") return self.update_status("🔄 正在生成缩略图...") def generate_all(): video_extensions = {'.mp4', '.mov', '.avi', '.mkv', '.m4v', '.wmv', '.flv'} video_files = [] for file_path in Path(self.current_dir).rglob('*'): if file_path.is_file() and file_path.suffix.lower() in video_extensions: video_files.append(file_path) for i, video_path in enumerate(video_files): self.generate_thumbnail(video_path) progress = f"📊 生成进度: {i+1}/{len(video_files)}" self.root.after(0, lambda p=progress: self.update_status(p)) self.root.after(0, lambda: self.update_status("✅ 缩略图生成完成")) self.root.after(0, self.refresh_file_list) threading.Thread(target=generate_all, daemon=True).start() def create_file_item(self, file_path, metadata_info): """创建单个文件项目""" relative_path = str(file_path.relative_to(self.current_dir)) # 主容器 item_frame = ttk.Frame(self.scrollable_frame, relief='solid', borderwidth=1) item_frame.pack(fill=tk.X, padx=5, pady=5) # 选择状态变量 selected_var = tk.BooleanVar() # 左侧缩略图和选择框 left_frame = ttk.Frame(item_frame) left_frame.pack(side=tk.LEFT, padx=10, pady=10) # 选择框 check_btn = ttk.Checkbutton(left_frame, variable=selected_var, command=lambda: self.toggle_file_selection(relative_path, selected_var.get())) check_btn.pack(anchor='nw') # 缩略图 thumbnail_label = ttk.Label(left_frame, text="📹") thumbnail_label.pack(pady=(5, 0)) # 尝试加载缩略图 thumbnail_path = self.thumbnails_dir / f"{file_path.stem}.jpg" if thumbnail_path.exists(): try: img = Image.open(thumbnail_path) img = img.resize(self.thumbnail_size, Image.Resampling.LANCZOS) photo = ImageTk.PhotoImage(img) thumbnail_label.configure(image=photo, text="") thumbnail_label.image = photo # 保持引用 except Exception as e: print(f"加载缩略图失败: {e}") # 右侧文件信息 right_frame = ttk.Frame(item_frame) right_frame.pack(side=tk.LEFT, fill=tk.BOTH, expand=True, padx=10, pady=10) # 文件名 name_label = ttk.Label(right_frame, text=file_path.name, font=('SF Pro Display', 12, 'bold')) name_label.pack(anchor='w') # 文件大小 try: size = file_path.stat().st_size size_str = self.format_file_size(size) except: size_str = "未知" size_label = ttk.Label(right_frame, text=f"📏 大小: {size_str}", font=('SF Pro Display', 10)) size_label.pack(anchor='w', pady=(2, 0)) # 标签 tags = metadata_info.get('tags', []) if tags: tags_text = f"🏷️ 标签: {', '.join(tags)}" else: tags_text = "🏷️ 暂无标签" tags_label = ttk.Label(right_frame, text=tags_text, font=('SF Pro Display', 10), foreground='#007AFF' if tags else '#999999') tags_label.pack(anchor='w', pady=(2, 0)) # 绑定双击预览事件 for widget in [item_frame, left_frame, right_frame, name_label, size_label, tags_label]: widget.bind('<Double-Button-1>', lambda e, fp=file_path: self.preview_file(fp)) # 存储项目信息 self.file_items[relative_path] = { 'frame': item_frame, 'selected_var': selected_var, 'file_path': file_path, 'tags_label': tags_label } def toggle_file_selection(self, relative_path, selected): """切换文件选择状态""" if selected: self.selected_files.add(relative_path) else: self.selected_files.discard(relative_path) self.update_selection_display() def update_selection_display(self): """更新选中文件显示""" count = len(self.selected_files) if count == 0: self.selected_count_label.config(text="未选择文件") elif count == 1: filename = Path(list(self.selected_files)[0]).name self.selected_count_label.config(text=f"选中: {filename}") else: self.selected_count_label.config(text=f"选中 {count} 个文件") # 更新当前标签显示 if count == 1: self.update_current_tags_display(list(self.selected_files)[0]) else: self.current_tags_listbox.delete(0, tk.END) def refresh_file_list(self): """刷新文件列表""" if not self.current_dir or not os.path.exists(self.current_dir): return # 清空现有项目 for item_info in self.file_items.values(): item_info['frame'].destroy() self.file_items.clear() self.selected_files.clear() # 加载元数据 metadata = self.load_metadata() # 扫描视频文件 video_extensions = {'.mp4', '.mov', '.avi', '.mkv', '.m4v', '.wmv', '.flv'} video_files = [] for file_path in Path(self.current_dir).rglob('*'): if file_path.is_file() and file_path.suffix.lower() in video_extensions: video_files.append(file_path) # 按文件名排序 video_files.sort(key=lambda x: x.name.lower()) # 创建文件项目 for file_path in video_files: relative_path = str(file_path.relative_to(self.current_dir)) file_info = metadata.get(relative_path, {}) self.create_file_item(file_path, file_info) self.update_status(f"📊 已加载 {len(video_files)} 个视频文件") self.update_selection_display() def on_search(self, event): """搜索功能""" search_text = self.search_var.get().strip().lower() for relative_path, item_info in self.file_items.items(): file_path = item_info['file_path'] if not search_text: # 显示所有文件 item_info['frame'].pack(fill=tk.X, padx=5, pady=5) else: # 检查文件名或标签是否匹配 metadata = self.load_metadata() file_metadata = metadata.get(relative_path, {}) tags = file_metadata.get('tags', []) filename_match = search_text in file_path.name.lower() tag_match = any(search_text in tag.lower() for tag in tags) if filename_match or tag_match: item_info['frame'].pack(fill=tk.X, padx=5, pady=5) else: item_info['frame'].pack_forget() def clear_search(self): """清除搜索""" self.search_var.set("") self.on_search(None) def zoom_in(self, event=None): """放大缩略图""" current_width, current_height = self.thumbnail_size new_width = min(current_width + 50, 400) new_height = int(new_width * 3/4) self.thumbnail_size = (new_width, new_height) self.refresh_file_list() self.update_status(f"🔍 缩略图大小: {new_width}x{new_height}") def zoom_out(self, event=None): """缩小缩略图""" current_width, current_height = self.thumbnail_size new_width = max(current_width - 50, 100) new_height = int(new_width * 3/4) self.thumbnail_size = (new_width, new_height) self.refresh_file_list() self.update_status(f"🔍 缩略图大小: {new_width}x{new_height}") def select_all(self, event=None): """全选文件""" for relative_path, item_info in self.file_items.items(): if item_info['frame'].winfo_viewable(): # 只选择可见的文件 item_info['selected_var'].set(True) self.selected_files.add(relative_path) self.update_selection_display() self.update_status(f"✅ 已选择 {len(self.selected_files)} 个文件") def preview_selected(self, event=None): """预览选中的文件""" if len(self.selected_files) == 1: relative_path = list(self.selected_files)[0] item_info = self.file_items.get(relative_path) if item_info: self.preview_file(item_info['file_path']) def preview_file(self, file_path): """预览文件""" try: subprocess.run(['open', str(file_path)], check=True) self.update_status(f"🎬 正在预览: {file_path.name}") except subprocess.CalledProcessError: messagebox.showerror("错误", f"无法预览文件: {file_path.name}") def add_tag_to_selected(self, tag): """为选中文件添加标签""" if not self.selected_files: messagebox.showwarning("警告", "请先选择文件") return metadata = self.load_metadata() updated_files = [] for relative_path in self.selected_files: # 获取或创建文件信息 if relative_path not in metadata: file_path = Path(self.current_dir) / relative_path metadata[relative_path] = {'tags': [], 'filename': file_path.name} # 添加标签（避免重复） if tag not in metadata[relative_path]['tags']: metadata[relative_path]['tags'].append(tag) updated_files.append(relative_path) if updated_files: # 保存元数据 self.save_metadata(metadata) # 更新界面显示 self.update_file_displays(updated_files) count = len(updated_files) self.update_status(f"✅ 已为 {count} 个文件添加标签: {tag}") else: self.update_status(f"ℹ️ 选中文件已包含标签: {tag}") def add_custom_tag(self): """添加自定义标签""" tag = self.custom_tag_var.get().strip() if not tag: return self.add_tag_to_selected(tag) self.custom_tag_var.set("") # 清空输入框 def remove_selected_tag(self): """删除选中的标签""" # 获取选中的标签 tag_selection = self.current_tags_listbox.curselection() if not tag_selection: messagebox.showwarning("警告", "请先选择要删除的标签") return tag_to_remove = self.current_tags_listbox.get(tag_selection[0]) if not self.selected_files: messagebox.showwarning("警告", "请先选择文件") return metadata = self.load_metadata() updated_files = [] for relative_path in self.selected_files: if relative_path in metadata and tag_to_remove in metadata[relative_path]['tags']: metadata[relative_path]['tags'].remove(tag_to_remove) updated_files.append(relative_path) if updated_files: # 保存元数据 self.save_metadata(metadata) # 更新界面显示 self.update_file_displays(updated_files) count = len(updated_files) self.update_status(f"🗑️ 已从 {count} 个文件删除标签: {tag_to_remove}") # 更新当前标签显示 if len(self.selected_files) == 1: self.update_current_tags_display(list(self.selected_files)[0]) def batch_add_tags(self): """批量添加标签""" if not self.selected_files: messagebox.showwarning("警告", "请先选择文件") return # 创建标签选择对话框 self.create_tag_selection_dialog("添加") def batch_remove_tags(self): """批量删除标签""" if not self.selected_files: messagebox.showwarning("警告", "请先选择文件") return # 创建标签选择对话框 self.create_tag_selection_dialog("删除") def create_tag_selection_dialog(self, operation): """创建标签选择对话框""" dialog = tk.Toplevel(self.root) dialog.title(f"批量{operation}标签") dialog.geometry("400x500") dialog.transient(self.root) dialog.grab_set() # 设置对话框样式 dialog.configure(bg='#f0f0f0') # 主框架 main_frame = ttk.Frame(dialog) main_frame.pack(fill=tk.BOTH, expand=True, padx=20, pady=20) # 标题 title_label = ttk.Label(main_frame, text=f"选择要{operation}的标签", style='Title.TLabel') title_label.pack(pady=(0, 15)) # 获取所有可用标签 all_tags = set() metadata = self.load_metadata() if operation == "删除": # 删除操作：只显示选中文件已有的标签 for relative_path in self.selected_files: if relative_path in metadata: all_tags.update(metadata[relative_path].get('tags', [])) else: # 添加操作：显示所有预设标签 for category_tags in self.preset_tags.values(): all_tags.update(category_tags) # 添加所有已使用的标签 for file_info in metadata.values(): all_tags.update(file_info.get('tags', [])) # 标签选择列表 tags_frame = ttk.LabelFrame(main_frame, text="可用标签") tags_frame.pack(fill=tk.BOTH, expand=True, pady=(0, 15)) # 创建带滚动条的标签列表 canvas = tk.Canvas(tags_frame, bg='white') scrollbar = ttk.Scrollbar(tags_frame, orient="vertical", command=canvas.yview) scrollable_frame = ttk.Frame(canvas) scrollable_frame.bind( "<Configure>", lambda e: canvas.configure(scrollregion=canvas.bbox("all")) ) canvas.create_window((0, 0), window=scrollable_frame, anchor="nw") canvas.configure(yscrollcommand=scrollbar.set) # 标签复选框 tag_vars = {} for tag in sorted(all_tags): var = tk.BooleanVar() tag_vars[tag] = var ttk.Checkbutton(scrollable_frame, text=tag, variable=var).pack(anchor='w', padx=10, pady=2) canvas.pack(side="left", fill="both", expand=True, padx=10, pady=10) scrollbar.pack(side="right", fill="y", pady=10) # 按钮框架 button_frame = ttk.Frame(main_frame) button_frame.pack(fill=tk.X) def apply_batch_operation(): selected_tags = [tag for tag, var in tag_vars.items() if var.get()] if not selected_tags: messagebox.showwarning("警告", "请选择至少一个标签") return metadata = self.load_metadata() updated_files = [] for relative_path in self.selected_files: if relative_path not in metadata: file_path = Path(self.current_dir) / relative_path metadata[relative_path] = {'tags': [], 'filename': file_path.name} file_tags = metadata[relative_path]['tags'] changed = False for tag in selected_tags: if operation == "添加": if tag not in file_tags: file_tags.append(tag) changed = True else: # 删除 if tag in file_tags: file_tags.remove(tag) changed = True if changed: updated_files.append(relative_path) if updated_files: self.save_metadata(metadata) self.update_file_displays(updated_files) count = len(updated_files) tag_list = ', '.join(selected_tags) self.update_status(f"✅ 已为 {count} 个文件{operation}标签: {tag_list}") dialog.destroy() ttk.Button(button_frame, text=f"✅ {operation}标签", style='Mac.TButton', command=apply_batch_operation).pack(side=tk.LEFT) ttk.Button(button_frame, text="❌ 取消", style='Mac.TButton', command=dialog.destroy).pack(side=tk.RIGHT) def copy_tags(self): """复制标签到其他文件""" if len(self.selected_files) < 2: messagebox.showwarning("警告", "请选择至少两个文件（第一个作为源文件）") return source_file = list(self.selected_files)[0] target_files = list(self.selected_files)[1:] metadata = self.load_metadata() source_tags = metadata.get(source_file, {}).get('tags', []) if not source_tags: messagebox.showwarning("警告", "源文件没有标签") return # 复制标签 updated_files = [] for relative_path in target_files: if relative_path not in metadata: file_path = Path(self.current_dir) / relative_path metadata[relative_path] = {'tags': [], 'filename': file_path.name} # 添加源文件的标签（避免重复） file_tags = metadata[relative_path]['tags'] for tag in source_tags: if tag not in file_tags: file_tags.append(tag) updated_files.append(relative_path) # 保存元数据 self.save_metadata(metadata) self.update_file_displays(updated_files) count = len(target_files) source_name = Path(source_file).name self.update_status(f"📋 已将 {source_name} 的标签复制到 {count} 个文件") def update_file_displays(self, updated_files): """更新文件显示""" metadata = self.load_metadata() for relative_path in updated_files: if relative_path in self.file_items: item_info = self.file_items[relative_path] file_metadata = metadata.get(relative_path, {}) tags = file_metadata.get('tags', []) if tags: tags_text = f"🏷️ 标签: {', '.join(tags)}" color = '#007AFF' else: tags_text = "🏷️ 暂无标签" color = '#999999' item_info['tags_label'].config(text=tags_text, foreground=color) # 更新当前标签显示 if len(self.selected_files) == 1: self.update_current_tags_display(list(self.selected_files)[0]) def update_current_tags_display(self, relative_path): """更新当前文件标签显示""" self.current_tags_listbox.delete(0, tk.END) metadata = self.load_metadata() file_info = metadata.get(relative_path, {}) tags = file_info.get('tags', []) for tag in tags: self.current_tags_listbox.insert(tk.END, tag) def remove_tags_from_selected(self, event=None): """删除选中文件的所有标签""" if not self.selected_files: return result = messagebox.askyesno("确认", "确定要删除选中文件的所有标签吗？") if not result: return metadata = self.load_metadata() updated_files = [] for relative_path in self.selected_files: if relative_path in metadata and metadata[relative_path].get('tags'): metadata[relative_path]['tags'] = [] updated_files.append(relative_path) if updated_files: self.save_metadata(metadata) self.update_file_displays(updated_files) count = len(updated_files) self.update_status(f"🗑️ 已清除 {count} 个文件的所有标签") def on_mousewheel(self, event): """鼠标滚轮事件""" self.canvas.yview_scroll(int(-1 * (event.delta / 120)), "units") def on_canvas_click(self, event): """画布点击事件 - 用于取消选择""" # 如果点击的不是文件项目，则清除选择 widget = event.widget while widget and widget != self.root: if hasattr(widget, 'master') and any( widget == item_info['frame'] for item_info in self.file_items.values() ): return widget = widget.master # 清除所有选择 for item_info in self.file_items.values(): item_info['selected_var'].set(False) self.selected_files.clear() self.update_selection_display() def select_directory(self): """选择工作目录""" directory = filedialog.askdirectory(title="选择素材目录") if directory: self.save_current_dir(directory) self.dir_label.config(text=f"📂 {directory}") self.refresh_file_list() self.update_status(f"📁 已切换到目录: {Path(directory).name}") def format_file_size(self, size_bytes): """格式化文件大小""" if size_bytes == 0: return "0 B" size_names = ["B", "KB", "MB", "GB"] i = 0 while size_bytes >= 1024 and i < len(size_names) - 1: size_bytes /= 1024.0 i += 1 return f"{size_bytes:.1f} {size_names[i]}" def load_metadata(self): """加载元数据""" try: with open(self.metadata_file, 'r', encoding='utf-8') as f: return json.load(f) except: return {} def save_metadata(self, metadata): """保存元数据""" with open(self.metadata_file, 'w', encoding='utf-8') as f: json.dump(metadata, f, ensure_ascii=False, indent=2) def update_status(self, message): """更新状态栏""" self.status_bar.config(text=message) # 3秒后恢复默认状态 self.root.after(3000, lambda: self.status_bar.config(text="✅ 就绪")) def run(self): """运行应用程序""" self.root.mainloop()
def main():
"""主函数"""
try:
app = VideoAssetManager()
app.run()
except ImportError as e:
if 'cv2' in str(e):
print("错误: 缺少 opencv-python 依赖")
print("请运行: pip install opencv-python")
elif 'PIL' in str(e):
print("错误: 缺少 Pillow 依赖")
print("请运行: pip install Pillow")
else:
print(f"导入错误: {e}")
except Exception as e:
print(f"程序运行错误: {e}")
if name == "main":
main()
Gedanke
Antwort
下面这段代码直接贴进 VideoAssetManager 类里（建议放在 update_status 之后），就能把工具栏里尚未实现的 “📊 统计报告” 和 “💾 导出数据” 按钮补完——不改其他逻辑，也不需要再引入新第三方库。
python# ───────────────── 统计 & 导出 ───────────────── def show_statistics(self): """统计当前目录的视频数、已打标签文件数，以及 TOP10 标签使用情况""" if not self.current_dir: messagebox.showwarning("警告", "请先选择目录") return metadata = self.load_metadata() total_files = len(self.file_items) tagged_files = sum(1 for v in metadata.values() if v.get("tags")) untagged = total_files - tagged_files # 计算标签出现次数 from collections import Counter counter = Counter() for info in metadata.values(): counter.update(info.get("tags", [])) hot_tags = counter.most_common(10) hot_lines = [f"{tag}: {cnt}" for tag, cnt in hot_tags] or ["（暂无标签数据）"] report = ( f"📁 文件总数: {total_files}\n" f"🏷️ 已打标签: {tagged_files}\n" f"🚫 无标签文件: {untagged}\n\n" f"🔥 标签使用 TOP10:\n" + "\n".join(hot_lines) ) messagebox.showinfo("统计报告", report) def export_metadata(self): """把 metadata.json 复制导出到任意位置""" if not self.current_dir: messagebox.showwarning("警告", "请先选择目录") return dest = filedialog.asksaveasfilename( title="导出元数据", defaultextension=".json", filetypes=[("JSON 文件", "*.json")] ) if not dest: return # 用户取消 try: with open(dest, "w", encoding="utf-8") as f_out: json.dump(self.load_metadata(), f_out, ensure_ascii=False, indent=2) messagebox.showinfo("导出成功", f"✅ 已保存到: {dest}") self.update_status(f"💾 元数据已导出 → {Path(dest).name}") except Exception as e: messagebox.showerror("导出失败", f"无法写入文件:\n{e}")
功能说明
| 功能 | 触发方式 | 效果 |
|---|---|---|
| 统计报告 | 点击工具栏 📊 统计报告 | 弹窗展示①文件总数②已/未打标签数③最常用标签前十 |
| 导出数据 | 点击工具栏 💾 导出数据 | 让用户选择保存路径，将当前 metadata.json 拷贝为外部文件 |
加上这两段后，你的 “优化版本” 代码已完整可用：缩略图生成、Mac 风格 UI、批量标签、搜索/缩放、多选操作等都能正常工作。如果还想要图形化统计、CSV 导出、撤销重做等高级特性，随时告诉我 😉