# P2P Network Communication - File Transfer Widget """ 文件传输界面组件 实现文件选择、传输进度显示和图片预览 需求: 4.1, 4.3, 4.6, 5.2, 5.3, 5.4, 5.6 """ from __future__ import annotations import logging import os from typing import Optional, Dict, TYPE_CHECKING if TYPE_CHECKING: from typing import Type from PyQt6.QtWidgets import ( QWidget, QVBoxLayout, QHBoxLayout, QLabel, QPushButton, QProgressBar, QFileDialog, QScrollArea, QFrame, QMessageBox, QDialog, QSlider, QSizePolicy ) from PyQt6.QtCore import Qt, pyqtSignal, QSize from PyQt6.QtGui import QPixmap, QTransform from shared.models import TransferProgress, TransferStatus logger = logging.getLogger(__name__) class TransferItem(QFrame): """传输项组件""" cancel_requested = pyqtSignal(str) # file_id def __init__(self, file_id: str, file_name: str, total_size: int, is_upload: bool = True, parent=None): super().__init__(parent) self.file_id = file_id self.file_name = file_name self.total_size = total_size self.is_upload = is_upload self._setup_ui() def _setup_ui(self) -> None: """设置UI""" self.setFrameShape(QFrame.Shape.StyledPanel) self.setStyleSheet(""" QFrame { background-color: white; border: 1px solid #ddd; border-radius: 5px; margin: 2px; } """) layout = QVBoxLayout(self) layout.setContentsMargins(10, 10, 10, 10) layout.setSpacing(5) # 文件信息行 info_layout = QHBoxLayout() icon = "⬆️" if self.is_upload else "⬇️" self._name_label = QLabel(f"{icon} {self.file_name}") self._name_label.setStyleSheet("font-weight: bold;") info_layout.addWidget(self._name_label, 1) self._size_label = QLabel(self._format_size(self.total_size)) self._size_label.setStyleSheet("color: #666;") info_layout.addWidget(self._size_label) self._cancel_btn = QPushButton("✕") self._cancel_btn.setFixedSize(20, 20) self._cancel_btn.setStyleSheet(""" QPushButton { border: none; color: #999; } QPushButton:hover { color: red; } """) self._cancel_btn.clicked.connect(lambda: self.cancel_requested.emit(self.file_id)) info_layout.addWidget(self._cancel_btn) layout.addLayout(info_layout) # 进度条 self._progress_bar = QProgressBar() self._progress_bar.setRange(0, 100) self._progress_bar.setValue(0) self._progress_bar.setTextVisible(True) self._progress_bar.setStyleSheet(""" QProgressBar { border: 1px solid #ddd; border-radius: 3px; text-align: center; height: 20px; } QProgressBar::chunk { background-color: #4a90d9; border-radius: 2px; } """) layout.addWidget(self._progress_bar) # 状态行 status_layout = QHBoxLayout() self._status_label = QLabel("等待中...") self._status_label.setStyleSheet("color: #666; font-size: 12px;") status_layout.addWidget(self._status_label) status_layout.addStretch() self._speed_label = QLabel("") self._speed_label.setStyleSheet("color: #666; font-size: 12px;") status_layout.addWidget(self._speed_label) layout.addLayout(status_layout) def _format_size(self, size: int) -> str: """格式化文件大小""" if size < 1024: return f"{size} B" elif size < 1024 * 1024: return f"{size / 1024:.1f} KB" elif size < 1024 * 1024 * 1024: return f"{size / (1024 * 1024):.1f} MB" else: return f"{size / (1024 * 1024 * 1024):.2f} GB" def _format_speed(self, speed: float) -> str: """格式化传输速度""" return f"{self._format_size(int(speed))}/s" def update_progress(self, progress: TransferProgress) -> None: """更新进度""" percent = int(progress.progress_percent) self._progress_bar.setValue(percent) self._speed_label.setText(self._format_speed(progress.speed)) if progress.eta > 0: eta_str = f"剩余 {int(progress.eta)}秒" else: eta_str = "" self._status_label.setText(f"传输中... {eta_str}") def set_status(self, status: TransferStatus) -> None: """设置状态""" status_texts = { TransferStatus.PENDING: "等待中...", TransferStatus.IN_PROGRESS: "传输中...", TransferStatus.COMPLETED: "✓ 完成", TransferStatus.FAILED: "✕ 失败", TransferStatus.CANCELLED: "已取消", TransferStatus.PAUSED: "已暂停", } self._status_label.setText(status_texts.get(status, "未知")) if status == TransferStatus.COMPLETED: self._progress_bar.setValue(100) self._cancel_btn.hide() self._progress_bar.setStyleSheet(""" QProgressBar { border: 1px solid #ddd; border-radius: 3px; text-align: center; height: 20px; } QProgressBar::chunk { background-color: #4caf50; border-radius: 2px; } """) elif status == TransferStatus.FAILED: self._cancel_btn.hide() self._progress_bar.setStyleSheet(""" QProgressBar { border: 1px solid #ddd; border-radius: 3px; text-align: center; height: 20px; } QProgressBar::chunk { background-color: #f44336; border-radius: 2px; } """) class FileTransferWidget(QWidget): """ 文件传输界面组件 实现文件选择对话框 (需求 4.1) 实现传输进度显示 (需求 4.3) 实现图片预览和查看 (需求 5.2, 5.3, 5.4, 5.6) """ # 信号定义 send_file_requested = pyqtSignal(str, str) # peer_id, file_path cancel_transfer_requested = pyqtSignal(str) # file_id def __init__(self, parent=None): super().__init__(parent) self._transfers: Dict[str, TransferItem] = {} self._setup_ui() logger.info("FileTransferWidget initialized") def _setup_ui(self) -> None: """设置UI""" layout = QVBoxLayout(self) layout.setContentsMargins(0, 0, 0, 0) layout.setSpacing(0) # 标题栏 header = QFrame() header.setFixedHeight(40) header.setStyleSheet(""" QFrame { background-color: #f5f5f5; border-bottom: 1px solid #ddd; } """) header_layout = QHBoxLayout(header) header_layout.setContentsMargins(10, 0, 10, 0) title_label = QLabel("文件传输") title_label.setStyleSheet("font-weight: bold;") header_layout.addWidget(title_label) header_layout.addStretch() self._clear_btn = QPushButton("清除已完成") self._clear_btn.clicked.connect(self._clear_completed) header_layout.addWidget(self._clear_btn) layout.addWidget(header) # 传输列表 scroll_area = QScrollArea() scroll_area.setWidgetResizable(True) scroll_area.setHorizontalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOff) self._transfer_container = QWidget() self._transfer_layout = QVBoxLayout(self._transfer_container) self._transfer_layout.setContentsMargins(5, 5, 5, 5) self._transfer_layout.setSpacing(5) self._transfer_layout.addStretch() scroll_area.setWidget(self._transfer_container) layout.addWidget(scroll_area, 1) # 空状态提示 self._empty_label = QLabel("暂无传输任务") self._empty_label.setAlignment(Qt.AlignmentFlag.AlignCenter) self._empty_label.setStyleSheet("color: #999;") self._transfer_layout.insertWidget(0, self._empty_label) def _clear_completed(self) -> None: """清除已完成的传输""" to_remove = [] for file_id, item in self._transfers.items(): # 检查是否已完成（简化判断） if item._progress_bar.value() == 100: to_remove.append(file_id) for file_id in to_remove: self.remove_transfer(file_id) def _update_empty_state(self) -> None: """更新空状态显示""" self._empty_label.setVisible(len(self._transfers) == 0) # ==================== 公共方法 ==================== def add_transfer(self, file_id: str, file_name: str, total_size: int, is_upload: bool = True) -> None: """ 添加传输任务 Args: file_id: 文件ID file_name: 文件名 total_size: 文件大小 is_upload: 是否是上传 """ if file_id in self._transfers: return item = TransferItem(file_id, file_name, total_size, is_upload) item.cancel_requested.connect(self._on_cancel_requested) self._transfer_layout.insertWidget(self._transfer_layout.count() - 1, item) self._transfers[file_id] = item self._update_empty_state() logger.debug(f"Transfer added: {file_name}") def update_transfer_progress(self, file_id: str, progress: TransferProgress) -> None: """ 更新传输进度 实现传输进度显示 (需求 4.3) WHILE 文件传输进行中 THEN P2P_Client SHALL 显示传输进度百分比和传输速度 Args: file_id: 文件ID progress: 进度信息 """ if file_id in self._transfers: self._transfers[file_id].update_progress(progress) def update_transfer_status(self, file_id: str, status: TransferStatus) -> None: """ 更新传输状态 Args: file_id: 文件ID status: 状态 """ if file_id in self._transfers: self._transfers[file_id].set_status(status) def remove_transfer(self, file_id: str) -> None: """ 移除传输任务 Args: file_id: 文件ID """ if file_id not in self._transfers: return item = self._transfers.pop(file_id) self._transfer_layout.removeWidget(item) item.deleteLater() self._update_empty_state() def _on_cancel_requested(self, file_id: str) -> None: """处理取消请求""" self.cancel_transfer_requested.emit(file_id) def show_file_dialog(self, peer_id: str) -> Optional[str]: """ 显示文件选择对话框 实现文件选择对话框 (需求 4.1) WHEN 用户选择发送文件 THEN File_Transfer_Module SHALL 允许用户从本地选择任意类型的文件 Args: peer_id: 目标用户ID Returns: 选择的文件路径，如果取消则返回None """ file_path, _ = QFileDialog.getOpenFileName( self, "选择文件", "", "所有文件 (*.*)" ) if file_path: self.send_file_requested.emit(peer_id, file_path) return file_path return None def show_image_dialog(self, peer_id: str) -> Optional[str]: """ 显示图片选择对话框 Args: peer_id: 目标用户ID Returns: 选择的图片路径 """ file_path, _ = QFileDialog.getOpenFileName( self, "选择图片", "", "图片文件 (*.jpg *.jpeg *.png *.gif *.bmp);;所有文件 (*.*)" ) if file_path: self.send_file_requested.emit(peer_id, file_path) return file_path return None def show_save_dialog(self, file_name: str) -> Optional[str]: """ 显示保存文件对话框 实现用户选择保存位置 (需求 4.6) WHEN 接收到文件 THEN P2P_Client SHALL 允许用户选择保存位置 Args: file_name: 默认文件名 Returns: 保存路径 """ file_path, _ = QFileDialog.getSaveFileName( self, "保存文件", file_name, "所有文件 (*.*)" ) return file_path if file_path else None def show_image_preview(self, image_path: str) -> None: """ 显示图片预览 实现图片预览和查看 (需求 5.4) WHEN 用户点击图片缩略图 THEN P2P_Client SHALL 打开图片查看器显示完整图片 Args: image_path: 图片路径 """ dialog = ImagePreviewDialog(image_path, self) dialog.exec() def create_image_thumbnail(self, image_path: str) -> Optional["ImageThumbnail"]: """ 创建图片缩略图组件 实现图片缩略图预览 (需求 5.2, 5.3) Args: image_path: 图片路径 Returns: 缩略图组件 """ if not os.path.exists(image_path): return None thumbnail = ImageThumbnail(image_path) thumbnail.clicked.connect(self.show_image_preview) return thumbnail class ImagePreviewDialog(QDialog): """ 图片预览对话框 实现图片预览和查看 (需求 5.3, 5.4) WHEN 图片传输完成 THEN P2P_Client SHALL 在聊天窗口内直接显示图片缩略图 WHEN 用户点击图片缩略图 THEN P2P_Client SHALL 打开图片查看器显示完整图片 """ def __init__(self, image_path: str, parent=None): super().__init__(parent) self.image_path = image_path self._rotation = 0 self._zoom_level = 100 self._setup_ui() self._load_image() def _setup_ui(self) -> None: """设置UI""" self.setWindowTitle("图片预览") self.setMinimumSize(600, 500) self.resize(800, 600) layout = QVBoxLayout(self) layout.setContentsMargins(0, 0, 0, 0) layout.setSpacing(0) # 图片显示区域 scroll_area = QScrollArea() scroll_area.setWidgetResizable(True) scroll_area.setAlignment(Qt.AlignmentFlag.AlignCenter) scroll_area.setStyleSheet("background-color: #2d2d2d;") self._image_label = QLabel() self._image_label.setAlignment(Qt.AlignmentFlag.AlignCenter) scroll_area.setWidget(self._image_label) layout.addWidget(scroll_area, 1) # 工具栏 toolbar = QFrame() toolbar.setFixedHeight(50) toolbar.setStyleSheet(""" QFrame { background-color: #f5f5f5; border-top: 1px solid #ddd; } """) toolbar_layout = QHBoxLayout(toolbar) toolbar_layout.setContentsMargins(10, 5, 10, 5) toolbar_layout.setSpacing(10) # 缩放控制 zoom_out_btn = QPushButton("➖") zoom_out_btn.setFixedSize(30, 30) zoom_out_btn.clicked.connect(self._zoom_out) toolbar_layout.addWidget(zoom_out_btn) self._zoom_slider = QSlider(Qt.Orientation.Horizontal) self._zoom_slider.setRange(25, 400) self._zoom_slider.setValue(100) self._zoom_slider.setFixedWidth(150) self._zoom_slider.valueChanged.connect(self._on_zoom_changed) toolbar_layout.addWidget(self._zoom_slider) zoom_in_btn = QPushButton("➕") zoom_in_btn.setFixedSize(30, 30) zoom_in_btn.clicked.connect(self._zoom_in) toolbar_layout.addWidget(zoom_in_btn) self._zoom_label = QLabel("100%") self._zoom_label.setFixedWidth(50) toolbar_layout.addWidget(self._zoom_label) toolbar_layout.addStretch() # 旋转按钮 rotate_left_btn = QPushButton("↺") rotate_left_btn.setFixedSize(30, 30) rotate_left_btn.setToolTip("向左旋转") rotate_left_btn.clicked.connect(self._rotate_left) toolbar_layout.addWidget(rotate_left_btn) rotate_right_btn = QPushButton("↻") rotate_right_btn.setFixedSize(30, 30) rotate_right_btn.setToolTip("向右旋转") rotate_right_btn.clicked.connect(self._rotate_right) toolbar_layout.addWidget(rotate_right_btn) toolbar_layout.addStretch() # 关闭按钮 close_btn = QPushButton("关闭") close_btn.clicked.connect(self.close) toolbar_layout.addWidget(close_btn) layout.addWidget(toolbar) def _load_image(self) -> None: """加载图片""" if not os.path.exists(self.image_path): self._image_label.setText("图片不存在") return self._original_pixmap = QPixmap(self.image_path) if self._original_pixmap.isNull(): self._image_label.setText("无法加载图片") return self._update_display() def _update_display(self) -> None: """更新图片显示""" if not hasattr(self, '_original_pixmap') or self._original_pixmap.isNull(): return # 应用旋转 transform = QTransform() transform.rotate(self._rotation) rotated = self._original_pixmap.transformed(transform, Qt.TransformationMode.SmoothTransformation) # 应用缩放 scale = self._zoom_level / 100.0 scaled = rotated.scaled( int(rotated.width() * scale), int(rotated.height() * scale), Qt.AspectRatioMode.KeepAspectRatio, Qt.TransformationMode.SmoothTransformation ) self._image_label.setPixmap(scaled) def _zoom_in(self) -> None: """放大""" self._zoom_slider.setValue(min(400, self._zoom_level + 25)) def _zoom_out(self) -> None: """缩小""" self._zoom_slider.setValue(max(25, self._zoom_level - 25)) def _on_zoom_changed(self, value: int) -> None: """缩放变化""" self._zoom_level = value self._zoom_label.setText(f"{value}%") self._update_display() def _rotate_left(self) -> None: """向左旋转""" self._rotation = (self._rotation - 90) % 360 self._update_display() def _rotate_right(self) -> None: """向右旋转""" self._rotation = (self._rotation + 90) % 360 self._update_display() class ImageThumbnail(QFrame): """ 图片缩略图组件 实现图片缩略图预览 (需求 5.2) WHEN 图片发送前 THEN P2P_Client SHALL 显示图片缩略图预览 """ clicked = pyqtSignal(str) # image_path def __init__(self, image_path: str, thumbnail_size: tuple = (150, 150), parent=None): super().__init__(parent) self.image_path = image_path self.thumbnail_size = thumbnail_size self._setup_ui() def _setup_ui(self) -> None: """设置UI""" self.setFrameShape(QFrame.Shape.StyledPanel) self.setCursor(Qt.CursorShape.PointingHandCursor) self.setStyleSheet(""" QFrame { background-color: white; border: 1px solid #ddd; border-radius: 5px; } QFrame:hover { border-color: #4a90d9; } """) layout = QVBoxLayout(self) layout.setContentsMargins(5, 5, 5, 5) layout.setSpacing(5) # 缩略图 self._image_label = QLabel() self._image_label.setAlignment(Qt.AlignmentFlag.AlignCenter) self._image_label.setFixedSize(*self.thumbnail_size) if os.path.exists(self.image_path): pixmap = QPixmap(self.image_path) if not pixmap.isNull(): scaled = pixmap.scaled( *self.thumbnail_size, Qt.AspectRatioMode.KeepAspectRatio, Qt.TransformationMode.SmoothTransformation ) self._image_label.setPixmap(scaled) else: self._image_label.setText("无法加载") else: self._image_label.setText("图片不存在") layout.addWidget(self._image_label) # 文件名 file_name = os.path.basename(self.image_path) if len(file_name) > 20: file_name = file_name[:17] + "..." name_label = QLabel(file_name) name_label.setAlignment(Qt.AlignmentFlag.AlignCenter) name_label.setStyleSheet("font-size: 11px; color: #666;") layout.addWidget(name_label) def mousePressEvent(self, event) -> None: """鼠标点击事件""" if event.button() == Qt.MouseButton.LeftButton: self.clicked.emit(self.image_path) super().mousePressEvent(event)