Add download history feature with UI and localization
Introduces a download history dialog with thumbnail previews, redownload, and remove options. Updates all language files to support history-related UI strings. Adds HistoryManager utility and integrates history dialog into the main GUI. Enhances downloader to better track final downloaded files and merged outputs.
This commit is contained in:
oop7
@@ -15,6 +15,7 @@ This package contains all dialog classes organized by functionality:
|
||||
from src.gui.ytsage_gui_dialogs.ytsage_dialogs_base import AboutDialog, LogWindow
|
||||
from src.gui.ytsage_gui_dialogs.ytsage_dialogs_custom import CustomOptionsDialog, TimeRangeDialog
|
||||
from src.gui.ytsage_gui_dialogs.ytsage_dialogs_ffmpeg import FFmpegCheckDialog, FFmpegInstallThread
|
||||
from src.gui.ytsage_gui_dialogs.ytsage_dialogs_history import HistoryDialog
|
||||
from src.gui.ytsage_gui_dialogs.ytsage_dialogs_selection import (
|
||||
PlaylistSelectionDialog,
|
||||
SponsorBlockCategoryDialog,
|
||||
@@ -50,4 +51,7 @@ __all__ = [
|
||||
# Custom functionality dialogs
|
||||
"CustomOptionsDialog",
|
||||
"TimeRangeDialog",
|
||||
|
||||
# History dialog
|
||||
"HistoryDialog",
|
||||
]
|
||||
|
||||
@@ -0,0 +1,581 @@
|
||||
"""
|
||||
History Dialog for YTSage application.
|
||||
Displays download history with thumbnails and provides options to redownload or remove entries.
|
||||
"""
|
||||
|
||||
import os
|
||||
import subprocess
|
||||
from datetime import datetime
|
||||
from io import BytesIO
|
||||
from pathlib import Path
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
|
||||
import requests
|
||||
from PIL import Image
|
||||
from PySide6.QtCore import Qt, QSize, Signal
|
||||
from PySide6.QtGui import QPixmap, QIcon
|
||||
from PySide6.QtWidgets import (
|
||||
QDialog,
|
||||
QVBoxLayout,
|
||||
QHBoxLayout,
|
||||
QLabel,
|
||||
QLineEdit,
|
||||
QPushButton,
|
||||
QScrollArea,
|
||||
QWidget,
|
||||
QFrame,
|
||||
QMenu,
|
||||
QMessageBox,
|
||||
QSizePolicy,
|
||||
)
|
||||
|
||||
from src.utils.ytsage_history_manager import HistoryManager
|
||||
from src.utils.ytsage_constants import APP_THUMBNAILS_DIR, SUBPROCESS_CREATIONFLAGS
|
||||
from src.utils.ytsage_localization import _
|
||||
from src.utils.ytsage_logger import logger
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from src.gui.ytsage_gui_main import YTSageApp
|
||||
|
||||
|
||||
class HistoryEntryWidget(QFrame):
|
||||
"""Widget representing a single history entry."""
|
||||
|
||||
remove_requested = Signal(str) # Emit entry ID when remove is requested
|
||||
redownload_requested = Signal(dict) # Emit entry data when redownload is requested
|
||||
|
||||
def __init__(self, entry: dict, parent=None):
|
||||
super().__init__(parent)
|
||||
self.entry = entry
|
||||
self.entry_id = entry.get("id", "")
|
||||
|
||||
self.setup_ui()
|
||||
|
||||
def setup_ui(self):
|
||||
"""Setup the UI for this history entry."""
|
||||
self.setFrameStyle(QFrame.Shape.StyledPanel | QFrame.Shadow.Raised)
|
||||
self.setStyleSheet("""
|
||||
QFrame {
|
||||
background-color: #1d1e22;
|
||||
border: 1px solid #2a2d36;
|
||||
border-radius: 8px;
|
||||
padding: 10px;
|
||||
margin: 5px;
|
||||
}
|
||||
QFrame:hover {
|
||||
background-color: #252830;
|
||||
border-color: #3a3d46;
|
||||
}
|
||||
""")
|
||||
|
||||
main_layout = QHBoxLayout(self)
|
||||
main_layout.setSpacing(15)
|
||||
main_layout.setContentsMargins(10, 10, 10, 10)
|
||||
|
||||
# Thumbnail - Balanced size for better visibility without taking too much space
|
||||
self.thumbnail_label = QLabel()
|
||||
self.thumbnail_label.setFixedSize(240, 135) # 16:9 ratio, balanced size
|
||||
self.thumbnail_label.setStyleSheet("""
|
||||
QLabel {
|
||||
border: 2px solid #3d3d3d;
|
||||
border-radius: 6px;
|
||||
background-color: #15181b;
|
||||
}
|
||||
""")
|
||||
self.thumbnail_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
||||
self.thumbnail_label.setScaledContents(True)
|
||||
|
||||
# Load thumbnail
|
||||
self.load_thumbnail()
|
||||
|
||||
main_layout.addWidget(self.thumbnail_label, alignment=Qt.AlignmentFlag.AlignTop)
|
||||
|
||||
# Info section
|
||||
info_layout = QVBoxLayout()
|
||||
info_layout.setSpacing(5)
|
||||
|
||||
# Title
|
||||
title = self.entry.get("title", _("video_info.unknown_title"))
|
||||
self.title_label = QLabel(title)
|
||||
self.title_label.setWordWrap(True)
|
||||
self.title_label.setStyleSheet("""
|
||||
QLabel {
|
||||
font-size: 14px;
|
||||
font-weight: bold;
|
||||
color: #ffffff;
|
||||
}
|
||||
""")
|
||||
info_layout.addWidget(self.title_label)
|
||||
|
||||
# Channel (if available)
|
||||
channel = self.entry.get("channel")
|
||||
if channel:
|
||||
channel_label = QLabel(f"{_('video_info.channel')}: {channel}")
|
||||
channel_label.setStyleSheet("color: #cccccc; font-size: 12px;")
|
||||
info_layout.addWidget(channel_label)
|
||||
|
||||
# Download date
|
||||
download_date = self.entry.get("download_date", "")
|
||||
if download_date:
|
||||
try:
|
||||
dt = datetime.fromisoformat(download_date)
|
||||
date_str = dt.strftime("%Y-%m-%d %H:%M")
|
||||
date_label = QLabel(_("history.downloaded_on", date=date_str))
|
||||
date_label.setStyleSheet("color: #aaaaaa; font-size: 11px;")
|
||||
info_layout.addWidget(date_label)
|
||||
except Exception as e:
|
||||
logger.debug(f"Error parsing date: {e}")
|
||||
|
||||
# File size and type
|
||||
file_size = self.entry.get("file_size", 0)
|
||||
is_audio = self.entry.get("is_audio_only", False)
|
||||
|
||||
size_type_layout = QHBoxLayout()
|
||||
|
||||
# File type badge
|
||||
type_badge = QLabel(_("history.audio_download") if is_audio else _("history.video_download"))
|
||||
type_badge.setStyleSheet(f"""
|
||||
QLabel {{
|
||||
background-color: {'#c90000' if not is_audio else '#0066cc'};
|
||||
color: white;
|
||||
padding: 2px 8px;
|
||||
border-radius: 3px;
|
||||
font-size: 10px;
|
||||
font-weight: bold;
|
||||
}}
|
||||
""")
|
||||
size_type_layout.addWidget(type_badge)
|
||||
|
||||
# File size
|
||||
if file_size > 0:
|
||||
size_str = self.format_file_size(file_size)
|
||||
size_label = QLabel(_("history.file_size", size=size_str))
|
||||
size_label.setStyleSheet("color: #aaaaaa; font-size: 11px;")
|
||||
size_type_layout.addWidget(size_label)
|
||||
|
||||
size_type_layout.addStretch()
|
||||
info_layout.addLayout(size_type_layout)
|
||||
|
||||
info_layout.addStretch()
|
||||
|
||||
main_layout.addLayout(info_layout, 1)
|
||||
|
||||
# Three-dot menu button
|
||||
self.menu_button = QPushButton("⋮")
|
||||
self.menu_button.setFixedSize(40, 40)
|
||||
self.menu_button.setStyleSheet("""
|
||||
QPushButton {
|
||||
background-color: #2a2d36;
|
||||
border: none;
|
||||
border-radius: 20px;
|
||||
color: white;
|
||||
font-size: 24px;
|
||||
font-weight: bold;
|
||||
}
|
||||
QPushButton:hover {
|
||||
background-color: #3a3d46;
|
||||
}
|
||||
QPushButton:pressed {
|
||||
background-color: #c90000;
|
||||
}
|
||||
""")
|
||||
self.menu_button.clicked.connect(self.show_menu)
|
||||
|
||||
main_layout.addWidget(self.menu_button, alignment=Qt.AlignmentFlag.AlignTop)
|
||||
|
||||
def load_thumbnail(self):
|
||||
"""Load and display the thumbnail."""
|
||||
thumbnail_url = self.entry.get("thumbnail_url")
|
||||
|
||||
if not thumbnail_url:
|
||||
self.set_placeholder_thumbnail()
|
||||
return
|
||||
|
||||
# Check if thumbnail is cached
|
||||
thumbnail_filename = f"{self.entry_id}.jpg"
|
||||
thumbnail_path = APP_THUMBNAILS_DIR / thumbnail_filename
|
||||
|
||||
if thumbnail_path.exists():
|
||||
try:
|
||||
pixmap = QPixmap(str(thumbnail_path))
|
||||
if not pixmap.isNull():
|
||||
# Don't scale here, let setScaledContents handle it
|
||||
self.thumbnail_label.setPixmap(pixmap)
|
||||
return
|
||||
except Exception as e:
|
||||
logger.debug(f"Error loading cached thumbnail: {e}")
|
||||
|
||||
# Download thumbnail
|
||||
try:
|
||||
response = requests.get(thumbnail_url, timeout=5)
|
||||
response.raise_for_status()
|
||||
|
||||
image = Image.open(BytesIO(response.content))
|
||||
|
||||
# Don't resize, keep original quality and just save at higher quality
|
||||
# The QPixmap scaling will handle the display size with high quality
|
||||
|
||||
# Save to cache with higher quality
|
||||
try:
|
||||
APP_THUMBNAILS_DIR.mkdir(parents=True, exist_ok=True)
|
||||
image.save(thumbnail_path, "JPEG", quality=95, optimize=True)
|
||||
except Exception as e:
|
||||
logger.debug(f"Error caching thumbnail: {e}")
|
||||
|
||||
# Convert to QPixmap
|
||||
image_bytes = BytesIO()
|
||||
image.save(image_bytes, format="JPEG", quality=95)
|
||||
image_bytes.seek(0)
|
||||
|
||||
pixmap = QPixmap()
|
||||
pixmap.loadFromData(image_bytes.read())
|
||||
|
||||
if not pixmap.isNull():
|
||||
# Don't scale here, let setScaledContents handle it
|
||||
self.thumbnail_label.setPixmap(pixmap)
|
||||
else:
|
||||
self.set_placeholder_thumbnail()
|
||||
|
||||
except Exception as e:
|
||||
logger.debug(f"Error downloading thumbnail: {e}")
|
||||
self.set_placeholder_thumbnail()
|
||||
|
||||
def set_placeholder_thumbnail(self):
|
||||
"""Set a placeholder when thumbnail is not available."""
|
||||
self.thumbnail_label.setText("📹" if not self.entry.get("is_audio_only") else "🎵")
|
||||
self.thumbnail_label.setStyleSheet("""
|
||||
QLabel {
|
||||
border: 1px solid #3d3d3d;
|
||||
border-radius: 4px;
|
||||
background-color: #15181b;
|
||||
color: #666666;
|
||||
font-size: 48px;
|
||||
}
|
||||
""")
|
||||
|
||||
def format_file_size(self, size_bytes: int) -> str:
|
||||
"""Format file size in human-readable format."""
|
||||
for unit in ['B', 'KB', 'MB', 'GB']:
|
||||
if size_bytes < 1024.0:
|
||||
return f"{size_bytes:.1f} {unit}"
|
||||
size_bytes /= 1024.0
|
||||
return f"{size_bytes:.1f} TB"
|
||||
|
||||
def show_menu(self):
|
||||
"""Show the context menu with options."""
|
||||
menu = QMenu(self)
|
||||
menu.setStyleSheet("""
|
||||
QMenu {
|
||||
background-color: #2a2d36;
|
||||
border: 1px solid #3a3d46;
|
||||
color: white;
|
||||
padding: 5px;
|
||||
}
|
||||
QMenu::item {
|
||||
padding: 8px 20px;
|
||||
border-radius: 4px;
|
||||
}
|
||||
QMenu::item:selected {
|
||||
background-color: #c90000;
|
||||
}
|
||||
""")
|
||||
|
||||
# Open file location
|
||||
open_action = menu.addAction("📁 " + _("history.open_location"))
|
||||
open_action.triggered.connect(self.open_file_location)
|
||||
|
||||
# Redownload
|
||||
redownload_action = menu.addAction("⬇️ " + _("history.redownload"))
|
||||
redownload_action.triggered.connect(self.redownload)
|
||||
|
||||
menu.addSeparator()
|
||||
|
||||
# Remove from history
|
||||
remove_action = menu.addAction("🗑️ " + _("history.remove"))
|
||||
remove_action.triggered.connect(self.remove_from_history)
|
||||
|
||||
# Show menu at button position
|
||||
menu.exec(self.menu_button.mapToGlobal(self.menu_button.rect().bottomLeft()))
|
||||
|
||||
def open_file_location(self):
|
||||
"""Open the file location in the system file explorer."""
|
||||
file_path = Path(self.entry.get("file_path", ""))
|
||||
|
||||
if not file_path.exists():
|
||||
QMessageBox.warning(
|
||||
self,
|
||||
_("history.file_not_found"),
|
||||
_("history.file_not_found_message", path=str(file_path))
|
||||
)
|
||||
return
|
||||
|
||||
try:
|
||||
# On Windows, use explorer with /select to highlight the file
|
||||
if os.name == "nt":
|
||||
subprocess.run(['explorer', '/select,', str(file_path)], creationflags=SUBPROCESS_CREATIONFLAGS)
|
||||
# On macOS, use open with -R to reveal in Finder
|
||||
elif subprocess.sys.platform == "darwin":
|
||||
subprocess.run(['open', '-R', str(file_path)])
|
||||
# On Linux, try to open the folder
|
||||
else:
|
||||
folder_path = file_path.parent
|
||||
subprocess.run(['xdg-open', str(folder_path)])
|
||||
|
||||
logger.info(f"Opened file location: {file_path}")
|
||||
except Exception as e:
|
||||
logger.exception(f"Error opening file location: {e}")
|
||||
QMessageBox.warning(self, "Error", f"Could not open file location: {str(e)}")
|
||||
|
||||
def redownload(self):
|
||||
"""Request redownload of this entry."""
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
_("history.redownload_confirm_title"),
|
||||
_("history.redownload_confirm_message", title=self.entry.get("title", "")),
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No
|
||||
)
|
||||
|
||||
if reply == QMessageBox.StandardButton.Yes:
|
||||
self.redownload_requested.emit(self.entry)
|
||||
|
||||
def remove_from_history(self):
|
||||
"""Request removal of this entry from history."""
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
_("history.remove_confirm_title"),
|
||||
_("history.remove_confirm_message", title=self.entry.get("title", "")),
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No
|
||||
)
|
||||
|
||||
if reply == QMessageBox.StandardButton.Yes:
|
||||
self.remove_requested.emit(self.entry_id)
|
||||
|
||||
|
||||
class HistoryDialog(QDialog):
|
||||
"""Dialog to display and manage download history."""
|
||||
|
||||
redownload_requested = Signal(dict) # Signal to request redownload in main window
|
||||
|
||||
def __init__(self, parent: Optional["YTSageApp"] = None):
|
||||
super().__init__(parent)
|
||||
self.parent_app = parent
|
||||
self.entry_widgets = []
|
||||
|
||||
self.setup_ui()
|
||||
self.load_history()
|
||||
|
||||
def setup_ui(self):
|
||||
"""Setup the dialog UI."""
|
||||
self.setWindowTitle(_("history.title"))
|
||||
self.setMinimumSize(700, 500)
|
||||
self.resize(850, 600)
|
||||
|
||||
# Set window flags
|
||||
self.setWindowFlags(Qt.WindowType.Dialog | Qt.WindowType.WindowCloseButtonHint)
|
||||
|
||||
layout = QVBoxLayout(self)
|
||||
layout.setSpacing(10)
|
||||
layout.setContentsMargins(20, 20, 20, 20)
|
||||
|
||||
# Header with title and buttons
|
||||
header_layout = QHBoxLayout()
|
||||
|
||||
title_label = QLabel(_("history.title"))
|
||||
title_label.setStyleSheet("""
|
||||
QLabel {
|
||||
font-size: 18px;
|
||||
font-weight: bold;
|
||||
color: white;
|
||||
}
|
||||
""")
|
||||
header_layout.addWidget(title_label)
|
||||
|
||||
header_layout.addStretch()
|
||||
|
||||
# Clear all button
|
||||
self.clear_all_btn = QPushButton(_("history.clear_all"))
|
||||
self.clear_all_btn.setStyleSheet("""
|
||||
QPushButton {
|
||||
background-color: #c90000;
|
||||
color: white;
|
||||
padding: 8px 16px;
|
||||
border: none;
|
||||
border-radius: 4px;
|
||||
font-weight: bold;
|
||||
}
|
||||
QPushButton:hover {
|
||||
background-color: #a50000;
|
||||
}
|
||||
QPushButton:pressed {
|
||||
background-color: #800000;
|
||||
}
|
||||
""")
|
||||
self.clear_all_btn.clicked.connect(self.clear_all_history)
|
||||
header_layout.addWidget(self.clear_all_btn)
|
||||
|
||||
layout.addLayout(header_layout)
|
||||
|
||||
# Search bar
|
||||
self.search_input = QLineEdit()
|
||||
self.search_input.setPlaceholderText(_("history.search_placeholder"))
|
||||
self.search_input.setStyleSheet("""
|
||||
QLineEdit {
|
||||
padding: 10px;
|
||||
border: 2px solid #1b2021;
|
||||
border-radius: 4px;
|
||||
background-color: #1b2021;
|
||||
color: #ffffff;
|
||||
font-size: 13px;
|
||||
}
|
||||
""")
|
||||
self.search_input.textChanged.connect(self.filter_history)
|
||||
layout.addWidget(self.search_input)
|
||||
|
||||
# Scroll area for history entries
|
||||
scroll_area = QScrollArea()
|
||||
scroll_area.setWidgetResizable(True)
|
||||
scroll_area.setStyleSheet("""
|
||||
QScrollArea {
|
||||
border: none;
|
||||
background-color: transparent;
|
||||
}
|
||||
""")
|
||||
|
||||
# Container for history entries
|
||||
self.history_container = QWidget()
|
||||
self.history_layout = QVBoxLayout(self.history_container)
|
||||
self.history_layout.setSpacing(10)
|
||||
self.history_layout.setContentsMargins(0, 0, 0, 0)
|
||||
self.history_layout.setAlignment(Qt.AlignmentFlag.AlignTop)
|
||||
|
||||
scroll_area.setWidget(self.history_container)
|
||||
layout.addWidget(scroll_area)
|
||||
|
||||
# Status label
|
||||
self.status_label = QLabel()
|
||||
self.status_label.setStyleSheet("color: #aaaaaa; font-size: 12px;")
|
||||
layout.addWidget(self.status_label)
|
||||
|
||||
# Apply dark theme
|
||||
self.setStyleSheet("""
|
||||
QDialog {
|
||||
background-color: #15181b;
|
||||
}
|
||||
QLabel {
|
||||
color: #ffffff;
|
||||
}
|
||||
""")
|
||||
|
||||
def load_history(self):
|
||||
"""Load and display history entries."""
|
||||
# Clear existing widgets
|
||||
for widget in self.entry_widgets:
|
||||
widget.deleteLater()
|
||||
self.entry_widgets.clear()
|
||||
|
||||
# Get history entries
|
||||
entries = HistoryManager.get_all_entries()
|
||||
|
||||
if not entries:
|
||||
self.show_empty_state()
|
||||
return
|
||||
|
||||
# Create widgets for each entry
|
||||
for entry in entries:
|
||||
widget = HistoryEntryWidget(entry, self.history_container)
|
||||
widget.remove_requested.connect(self.remove_entry)
|
||||
widget.redownload_requested.connect(self.handle_redownload)
|
||||
|
||||
self.history_layout.addWidget(widget)
|
||||
self.entry_widgets.append(widget)
|
||||
|
||||
# Update status
|
||||
count = len(entries)
|
||||
if count == 1:
|
||||
status_text = _("history.one_entry")
|
||||
else:
|
||||
status_text = _("history.entries_count", count=count)
|
||||
self.status_label.setText(status_text)
|
||||
|
||||
def show_empty_state(self):
|
||||
"""Show empty state when there's no history."""
|
||||
empty_widget = QWidget()
|
||||
empty_layout = QVBoxLayout(empty_widget)
|
||||
empty_layout.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
||||
|
||||
icon_label = QLabel("📂")
|
||||
icon_label.setStyleSheet("font-size: 64px; color: #555555;")
|
||||
icon_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
||||
empty_layout.addWidget(icon_label)
|
||||
|
||||
title_label = QLabel(_("history.no_history"))
|
||||
title_label.setStyleSheet("font-size: 16px; color: #888888; font-weight: bold;")
|
||||
title_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
||||
empty_layout.addWidget(title_label)
|
||||
|
||||
desc_label = QLabel(_("history.no_history_description"))
|
||||
desc_label.setStyleSheet("font-size: 13px; color: #666666;")
|
||||
desc_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
||||
empty_layout.addWidget(desc_label)
|
||||
|
||||
self.history_layout.addWidget(empty_widget)
|
||||
self.entry_widgets.append(empty_widget)
|
||||
|
||||
self.status_label.setText("")
|
||||
self.clear_all_btn.setEnabled(False)
|
||||
|
||||
def filter_history(self, query: str):
|
||||
"""Filter history entries based on search query."""
|
||||
if not query:
|
||||
# Show all entries
|
||||
for widget in self.entry_widgets:
|
||||
widget.show()
|
||||
return
|
||||
|
||||
# Hide/show based on query
|
||||
query_lower = query.lower()
|
||||
visible_count = 0
|
||||
|
||||
for widget in self.entry_widgets:
|
||||
if isinstance(widget, HistoryEntryWidget):
|
||||
title = (widget.entry.get("title") or "").lower()
|
||||
channel = (widget.entry.get("channel") or "").lower()
|
||||
|
||||
if query_lower in title or query_lower in channel:
|
||||
widget.show()
|
||||
visible_count += 1
|
||||
else:
|
||||
widget.hide()
|
||||
|
||||
def remove_entry(self, entry_id: str):
|
||||
"""Remove an entry from history."""
|
||||
success = HistoryManager.remove_entry(entry_id)
|
||||
|
||||
if success:
|
||||
# Reload history
|
||||
self.load_history()
|
||||
logger.info(f"Removed entry from history: {entry_id}")
|
||||
else:
|
||||
QMessageBox.warning(self, "Error", "Failed to remove entry from history")
|
||||
|
||||
def clear_all_history(self):
|
||||
"""Clear all history entries."""
|
||||
reply = QMessageBox.question(
|
||||
self,
|
||||
_("history.clear_confirm_title"),
|
||||
_("history.clear_confirm_message"),
|
||||
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No
|
||||
)
|
||||
|
||||
if reply == QMessageBox.StandardButton.Yes:
|
||||
count = HistoryManager.clear_history()
|
||||
self.load_history()
|
||||
logger.info(f"Cleared all history: {count} entries")
|
||||
|
||||
def handle_redownload(self, entry: dict):
|
||||
"""Handle redownload request."""
|
||||
# Emit signal to parent window
|
||||
self.redownload_requested.emit(entry)
|
||||
|
||||
# Close dialog
|
||||
self.accept()
|
||||
Reference in New Issue
Block a user