✨ Column added to display "date added" of list items. Fixes #146

2 files changed, 24 insertions, 3 deletions

diff --git a/tidal_dl_ng/gui.py b/tidal_dl_ng/gui.py
index 5837a6b..7aa2150 100644
--- a/tidal_dl_ng/gui.py
+++ b/tidal_dl_ng/gui.py
@@ -253,9 +253,11 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
         tree.customContextMenuRequested.connect(self.menu_context_tree_results)
 
     def _init_tree_results_model(self, model: QtGui.QStandardItemModel) -> None:
-        model.setColumnCount(7)
+        labels_column: [str] = ["#", "obj", "Artist", "Title", "Album", "Duration", "Quality", "Date Added"]
+
+        model.setColumnCount(len(labels_column))
         model.setRowCount(0)
-        model.setHorizontalHeaderLabels(["#", "obj", "Artist", "Title", "Album", "Duration", "Quality"])
+        model.setHorizontalHeaderLabels(labels_column)
 
     def _init_tree_queue(self, tree: QtWidgets.QTableWidget):
         tree.setColumnHidden(1, True)
@@ -480,6 +482,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
         child_album: QtGui.QStandardItem = QtGui.QStandardItem(item.album)
         child_duration: QtGui.QStandardItem = QtGui.QStandardItem(duration)
         child_quality: QtGui.QStandardItem = QtGui.QStandardItem(item.quality)
+        child_date_added: QtGui.QStandardItem = QtGui.QStandardItem(item.date_user_added)
 
         if isinstance(item.obj, Mix | Playlist | Album | Artist):
             # Add a disabled dummy child, so expansion arrow will appear. This Child will be replaced on expansion.
@@ -488,7 +491,16 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
             child_dummy.setEnabled(False)
             child_index.appendRow(child_dummy)
 
-        return child_index, child_obj, child_artist, child_title, child_album, child_duration, child_quality
+        return (
+            child_index,
+            child_obj,
+            child_artist,
+            child_title,
+            child_album,
+            child_duration,
+            child_quality,
+            child_date_added,
+        )
 
     def on_tr_results_add_top_level_item(self, item_child: Sequence[QtGui.QStandardItem]):
         self.model_tr_results.appendRow(item_child)
@@ -537,6 +549,8 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
             if isinstance(item, Track | Video | Album):
                 explicit = " 🅴" if item.explicit else ""
 
+            date_user_added: str = item.user_date_added.strftime("%Y-%m-%d_%H:%M") if item.user_date_added else ""
+
             if isinstance(item, Track):
                 result_item: ResultItem = ResultItem(
                     position=idx,
@@ -547,6 +561,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
                     obj=item,
                     quality=quality_audio_highest(item),
                     explicit=bool(item.explicit),
+                    date_user_added=date_user_added,
                 )
 
                 result.append(result_item)
@@ -560,6 +575,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
                     obj=item,
                     quality=item.video_quality,
                     explicit=bool(item.explicit),
+                    date_user_added=date_user_added,
                 )
 
                 result.append(result_item)
@@ -573,6 +589,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
                     obj=item,
                     quality="",
                     explicit=False,
+                    date_user_added=date_user_added,
                 )
 
                 result.append(result_item)
@@ -586,6 +603,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
                     obj=item,
                     quality=quality_audio_highest(item),
                     explicit=bool(item.explicit),
+                    date_user_added=date_user_added,
                 )
 
                 result.append(result_item)
@@ -600,6 +618,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
                     obj=item,
                     quality="",
                     explicit=False,
+                    date_user_added=date_user_added,
                 )
 
                 result.append(result_item)
@@ -613,6 +632,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
                     obj=item,
                     quality="",
                     explicit=False,
+                    date_user_added=date_user_added,
                 )
 
                 result.append(result_item)
diff --git a/tidal_dl_ng/model/gui_data.py b/tidal_dl_ng/model/gui_data.py
index 8c7c30e..186ed98 100644
--- a/tidal_dl_ng/model/gui_data.py
+++ b/tidal_dl_ng/model/gui_data.py
@@ -28,6 +28,7 @@ class ResultItem:
     obj: object
     quality: str
     explicit: bool
+    date_user_added: str
 
 
 @dataclass