-
Notifications
You must be signed in to change notification settings - Fork 9
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: Implement Video Manager with Import, Load, and Delete Features
- Developed a using PyQt5 to manage video imports for the project. - Added support for importing videos recursively from folders, including handling subfolders. - Integrated functionality to display imported videos in a table with columns for video name, path, load, and delete actions. - Implemented a check to prevent duplicate videos from being added to the table, using a set to track imported video paths. - Connected the Load button to emit a signal for loading videos, enabling seamless integration with the . - Enhanced delete functionality to remove videos from the table and allow re-importing if deleted.
- Loading branch information
1 parent
1337222
commit 7e69ad2
Showing
2 changed files
with
199 additions
and
80 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,92 @@ | ||
| import os | ||
| from qtpy.QtCore import Signal | ||
| from qtpy.QtWidgets import ( | ||
| QWidget, QVBoxLayout, QPushButton, QTableWidget, QTableWidgetItem, | ||
| QFileDialog, QMessageBox, QAbstractItemView | ||
| ) | ||
|
|
||
|
|
||
| class VideoManagerWidget(QWidget): | ||
| video_selected = Signal(str) # Signal to send the selected video path | ||
|
|
||
| def __init__(self, parent=None): | ||
| super().__init__(parent) | ||
|
|
||
| # Layouts | ||
| self.layout = QVBoxLayout(self) | ||
|
|
||
| # Set to track imported videos | ||
| self.imported_videos = set() | ||
|
|
||
| # Import Button | ||
| self.import_button = QPushButton("Import Videos") | ||
| self.import_button.clicked.connect(self.import_videos) | ||
| self.layout.addWidget(self.import_button) | ||
|
|
||
| # Video Table | ||
| # Rows: 0, Columns: 4 (Name, Path, Load, Delete) | ||
| self.video_table = QTableWidget(0, 4) | ||
| self.video_table.setHorizontalHeaderLabels( | ||
| ["Name", "Path", "Load", "Delete"]) | ||
| self.video_table.setSelectionBehavior(QAbstractItemView.SelectRows) | ||
| self.layout.addWidget(self.video_table) | ||
|
|
||
| def import_videos(self): | ||
| # Open folder dialog | ||
| folder_path = QFileDialog.getExistingDirectory( | ||
| self, "Select Video Folder") | ||
| if not folder_path: | ||
| return | ||
|
|
||
| # Get video files (recursively) | ||
| video_extensions = {'.mp4', '.avi', '.mov', '.mkv', | ||
| '.mpg'} # Add more extensions if needed | ||
| video_files = [] | ||
| for root, _, files in os.walk(folder_path): | ||
| for file in files: | ||
| if os.path.splitext(file)[1].lower() in video_extensions: | ||
| video_files.append(os.path.join(root, file)) | ||
|
|
||
| # Add videos to the table | ||
| for video in video_files: | ||
| # Check if video is already added | ||
| if video not in self.imported_videos: | ||
| self.add_video_to_table(video) | ||
|
|
||
| def add_video_to_table(self, video_path): | ||
| # Add the video to the imported videos set | ||
| self.imported_videos.add(video_path) | ||
| # Get video name | ||
| video_name = os.path.basename(video_path) | ||
|
|
||
| # Create a new row | ||
| row_position = self.video_table.rowCount() | ||
| self.video_table.insertRow(row_position) | ||
|
|
||
| # Add Name and Path | ||
| self.video_table.setItem(row_position, 0, QTableWidgetItem(video_name)) | ||
| self.video_table.setItem(row_position, 1, QTableWidgetItem(video_path)) | ||
|
|
||
| # Add Load Button | ||
| load_button = QPushButton("Load") | ||
| load_button.clicked.connect( | ||
| lambda: self.video_selected.emit(video_path)) | ||
| self.video_table.setCellWidget(row_position, 2, load_button) | ||
|
|
||
| # Add Delete Button | ||
| delete_button = QPushButton("Delete") | ||
| delete_button.clicked.connect( | ||
| lambda: self.delete_video(row_position, video_path)) | ||
| self.video_table.setCellWidget(row_position, 3, delete_button) | ||
|
|
||
| def delete_video(self, row, video_path): | ||
| # Confirm deletion | ||
| confirmation = QMessageBox.question( | ||
| self, "Delete Video", "Are you sure you want to delete this video from the list?", | ||
| QMessageBox.Yes | QMessageBox.No | ||
| ) | ||
| if confirmation == QMessageBox.Yes: | ||
| # Remove from the imported videos set | ||
| self.imported_videos.discard(video_path) | ||
|
|
||
| self.video_table.removeRow(row) |