latest pushes

gui/file_explorer.py

@@ -0,0 +1,112 @@
+# Copyright (C) 2024 Bellande Application Interoperability Xenogen Research Innovation Center, Ronaldson Bellande
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+import os
+from PyQt6.QtWidgets import QTreeView, QAbstractItemView
+from PyQt6.QtCore import Qt, pyqtSignal
+from PyQt6.QtGui import QStandardItemModel, QStandardItem
+from typing import Optional
+
+class FileExplorer(QTreeView):
+    file_selected = pyqtSignal(str)
+    _model: Optional[QStandardItemModel]
+
+    def __init__(self):
+        super().__init__()
+        self._model = None
+        self.current_path = ""
+        self.setup_ui()
+
+    def setup_ui(self):
+        # Create and set up the model
+        self._model = QStandardItemModel(parent=self)
+        self._model.setHorizontalHeaderLabels(['Name'])
+        
+        # Set up the tree view
+        super().setModel(self._model)
+        self.setAnimated(False)
+        self.setIndentation(20)
+        self.setSortingEnabled(True)
+        self.setSelectionMode(QAbstractItemView.SelectionMode.SingleSelection)
+        
+        # Connect signals
+        self.clicked.connect(self.on_item_clicked)
+
+    def populate_tree(self, path: str, parent_item: Optional[QStandardItem] = None) -> None:
+        if parent_item is None and self._model is not None:
+            self._model.clear()
+            self._model.setHorizontalHeaderLabels(['Name'])
+            parent_item = self._model.invisibleRootItem()
+            
+        try:
+            # List directory contents
+            items = os.listdir(path)
+            
+            # Sort items: directories first, then files
+            dirs = sorted([item for item in items if os.path.isdir(os.path.join(path, item))])
+            files = sorted([item for item in items if os.path.isfile(os.path.join(path, item))])
+            
+            # Add directories
+            for dir_name in dirs:
+                dir_path = os.path.join(path, dir_name)
+                try:
+                    item = QStandardItem(dir_name)
+                    item.setData(dir_path, Qt.ItemDataRole.UserRole)
+                    if parent_item:
+                        parent_item.appendRow(item)
+                except Exception:
+                    continue
+            
+            # Add files
+            for file_name in files:
+                file_path = os.path.join(path, file_name)
+                try:
+                    item = QStandardItem(file_name)
+                    item.setData(file_path, Qt.ItemDataRole.UserRole)
+                    if parent_item:
+                        parent_item.appendRow(item)
+                except Exception:
+                    continue
+                    
+        except PermissionError:
+            pass
+        except Exception:
+            pass
+
+    def on_item_clicked(self, index):
+        if self._model is None:
+            return
+            
+        item = self._model.itemFromIndex(index)
+        if item:
+            path = item.data(Qt.ItemDataRole.UserRole)
+            if os.path.isdir(path):
+                if self.isExpanded(index):
+                    self.collapse(index)
+                else:
+                    item.removeRows(0, item.rowCount())
+                    self.populate_tree(path, item)
+                    self.expand(index)
+            else:
+                self.file_selected.emit(path)
+
+    def set_root_path(self, path: str) -> None:
+        if os.path.exists(path):
+            self.current_path = path
+            self.populate_tree(path)
+
+    def refresh(self) -> None:
+        if self.current_path:
+            self.populate_tree(self.current_path)