From 1eff33dbd9e342142092d0581f79e21560ecdf2c Mon Sep 17 00:00:00 2001
From: Andrew Park <hep003@ucsd.edu>
Date: Wed, 18 Dec 2024 09:37:31 -0800
Subject: [PATCH] Separate the video name and its filepath columns in
 `VideoTablesModel` (#2052)

* add option to show video names with filepath

* add doc

* new feature added successfully

* delete unnecessary code

* remove attributes from video object

* Update dataviews.py

* remove all properties

* delete toggle option

* remove video show

* fix the order of the columns

* remove options

* Update sleap/gui/dataviews.py

Co-authored-by: Liezl Maree <38435167+roomrys@users.noreply.github.com>

* Update sleap/gui/dataviews.py

Co-authored-by: Liezl Maree <38435167+roomrys@users.noreply.github.com>

* use pathlib instead of substrings

* Update dataviews.py

Co-authored-by: Liezl Maree <38435167+roomrys@users.noreply.github.com>

* Use Path instead of pathlib.Path
and sort imports and remove unused imports

* Use item.filename instead of getattr

---------

Co-authored-by: Liezl Maree <38435167+roomrys@users.noreply.github.com>
---
 sleap/gui/dataviews.py | 38 +++++++++++++++++++++++++-------------
 1 file changed, 25 insertions(+), 13 deletions(-)

diff --git a/sleap/gui/dataviews.py b/sleap/gui/dataviews.py
index f68dc0180..721bdc321 100644
--- a/sleap/gui/dataviews.py
+++ b/sleap/gui/dataviews.py
@@ -15,20 +15,17 @@
 
 """
 
-from qtpy import QtCore, QtWidgets, QtGui
-
-import numpy as np
 import os
-
 from operator import itemgetter
+from pathlib import Path
+from typing import Any, Callable, List, Optional
 
-from typing import Any, Callable, Dict, List, Optional, Type
+import numpy as np
+from qtpy import QtCore, QtGui, QtWidgets
 
-from sleap.gui.state import GuiState
 from sleap.gui.commands import CommandContext
-from sleap.gui.color import ColorManager
-from sleap.io.dataset import Labels
-from sleap.instance import LabeledFrame, Instance
+from sleap.gui.state import GuiState
+from sleap.instance import LabeledFrame
 from sleap.skeleton import Skeleton
 
 
@@ -386,10 +383,25 @@ def getSelectedRowItem(self) -> Any:
 
 
 class VideosTableModel(GenericTableModel):
-    properties = ("filename", "frames", "height", "width", "channels")
-
-    def item_to_data(self, obj, item):
-        return {key: getattr(item, key) for key in self.properties}
+    properties = (
+        "name",
+        "filepath",
+        "frames",
+        "height",
+        "width",
+        "channels",
+    )
+
+    def item_to_data(self, obj, item: "Video"):
+        data = {}
+        for property in self.properties:
+            if property == "name":
+                data[property] = Path(item.filename).name
+            elif property == "filepath":
+                data[property] = str(Path(item.filename).parent)
+            else:
+                data[property] = getattr(item, property)
+        return data
 
 
 class SkeletonNodesTableModel(GenericTableModel):