Separate the video name and its filepath columns in VideoTablesModel

sleap/gui/dataviews.py

@@ -291,6 +291,7 @@ class GenericTableView(QtWidgets.QTableView):
     name_prefix: str = "selected_"
     is_activatable: bool = False
     is_sortable: bool = False
+    show_video_name: bool = False
 
     def __init__(
         self,
@@ -305,12 +306,17 @@ def __init__(
     ):
         super(GenericTableView, self).__init__()
 
+        model.show_video_name = self.show_video_name
+
         self.state = state or GuiState()
         self.row_name = row_name or self.row_name
         self.name_prefix = name_prefix if name_prefix is not None else self.name_prefix
         self.is_sortable = is_sortable or self.is_sortable
         self.is_activatable = is_activatable or self.is_activatable
         self.multiple_selection = multiple_selection
+        self.options = {
+            "Show Video Name": self.show_video_name,
+        }
 
         self.setModel(model)
 
@@ -384,12 +390,111 @@ def getSelectedRowItem(self) -> Any:
             return None
         return self.model().original_items[idx.row()]
 
+    def mousePressEvent(self, event) -> None:
+        """Only for right-click on VideosTableView.
+
+        Args:
+            event (QMouseEvent): The mouse event.
+        """
+        if event.button() == QtCore.Qt.RightButton and isinstance(
+            self.model(), VideosTableModel
+        ):
+            self.show_context_menu(event)
+        super().mousePressEvent(event)
+
+    def show_context_menu(self, event):
+        """Show context menu for VideosTableView.
+
+        Args:
+            event (QMouseEvent): The mouse event.
+        """
+        menu = QtWidgets.QMenu(self)
+
+        # Add actions to the menu
+        for option, is_checked in self.options.items():
+            action = QtWidgets.QAction(option, self)
+            action.setCheckable(True)
+            action.setChecked(is_checked)
+            action.triggered.connect(self.create_checked_lambda(option))
+            menu.addAction(action)
+
+        # Show the context menu
+        menu.exec_(event.globalPos())
+
+    def create_checked_lambda(self, option):
+        """Callback for context menu actions.
+
+        Args:
+            option (dict): The option to toggle.
+
+        Returns:
+            function: The callback function.
+        """
+        return lambda checked: self.toggle_option(option, checked)
+
+    def toggle_option(self, option, checked):
+        """Toggle the option in the context menu.
+
+        Args:
+            option (str): The option to toggle.
+            checked (bool): The new value for the option.
+        """
+        self.options[option] = checked
+        model = self.model()
+        if isinstance(model, VideosTableModel):
+            if option == "Show Video Name":
+                model.set_show_video_name(self.options["Show Video Name"])
+
 
+# TODO: Fix the error in the Pytest.
 class VideosTableModel(GenericTableModel):
-    properties = ("filename", "frames", "height", "width", "channels")
+    def __init__(self, items, show_video_name=False, **kwargs):
+        super().__init__(**kwargs)
+        self.items = items
+        self.show_video_name = show_video_name
+        self.properties = (
+            "filepath",
+            "name",
+            "frames",
+            "height",
+            "width",
+            "channels",
+        )
+        # TODO: Always with filepath and name (no filename)
+
+    def headerData(self, section, orientation, role=QtCore.Qt.DisplayRole):
+        if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
+            return self.properties[section].title()
+        return super().headerData(section, orientation, role)
+
+    def _get_properties(self):
+        """Return properties based on the show_video_name flag."""
+        if self.show_video_name:
+            return ["filepath", "name", "frames", "height", "width", "channels"]
+        return ["filename", "frames", "height", "width", "channels"]
+
+    def set_show_video_name(self, show_video_name: bool):
+        """Set whether to show video name in table."""
+        if self.show_video_name == show_video_name:
+            return
 
-    def item_to_data(self, obj, item):
-        return {key: getattr(item, key) for key in self.properties}
+        # Reset the table so that new columns are added
+        self.show_video_name = show_video_name
+        self.properties = self._get_properties()
+        self.beginResetModel()
+        self.endResetModel()
+
+    def item_to_data(self, obj, item: "Video"):
+        data = {}
+        for property in self.properties:
+            if property == "name":
+                data[property] = getattr(item, "filename").split("/")[-1]
+            elif property == "filepath":
+                splitted = getattr(item, "filename").split("/")[:-1]
+                data[property] = "/".join(splitted)
+            else:
+                data[property] = getattr(item, property)
+        return data
 
 
 class SkeletonNodesTableModel(GenericTableModel):