summaryrefslogtreecommitdiff
path: root/tidal_dl_ng/gui.py
diff options
context:
space:
mode:
authorRobert Honz <exislow+github@gmail.com>2024-03-29 12:34:55 +0100
committerRobert Honz <exislow+github@gmail.com>2024-03-29 12:34:55 +0100
commit349e1d79509ecf63122f2ed8d164d2ca6fdb7afd (patch)
treea1f7380a92b689a279df1e551cabe234c3373ac0 /tidal_dl_ng/gui.py
parent9dd8f6c856cf39cc1d228e171c1c81ff1d6e4b2f (diff)
downloadtidal-dl-ng-349e1d79509ecf63122f2ed8d164d2ca6fdb7afd.tar.gz
✨ Set table data using helper function.
Diffstat (limited to 'tidal_dl_ng/gui.py')
-rw-r--r--tidal_dl_ng/gui.py11
1 files changed, 8 insertions, 3 deletions
diff --git a/tidal_dl_ng/gui.py b/tidal_dl_ng/gui.py
index 9c13e3f..f988e97 100644
--- a/tidal_dl_ng/gui.py
+++ b/tidal_dl_ng/gui.py
@@ -6,7 +6,12 @@ from requests.exceptions import HTTPError
from tidal_dl_ng import __version__, update_available
from tidal_dl_ng.dialog import DialogLogin, DialogPreferences, DialogVersion
-from tidal_dl_ng.helper.gui import get_results_media_item, get_user_list_media_item
+from tidal_dl_ng.helper.gui import (
+ get_results_media_item,
+ get_user_list_media_item,
+ set_results_media,
+ set_user_list_media,
+)
from tidal_dl_ng.helper.path import get_format_template
from tidal_dl_ng.helper.tidal import (
get_tidal_media_id,
@@ -242,7 +247,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
info: str = item.sub_title
twi_child.setText(0, name)
- twi_child.setData(1, QtCore.Qt.ItemDataRole.UserRole, item)
+ set_user_list_media(twi_child, item)
twi_child.setText(2, info)
# Stop load spinner
@@ -368,7 +373,7 @@ class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):
# Populate child
child: QtWidgets.QTreeWidgetItem = QtWidgets.QTreeWidgetItem()
child.setText(0, index)
- child.setData(1, QtCore.Qt.ItemDataRole.UserRole, item.obj)
+ set_results_media(child, item.obj)
child.setText(2, item.artist)
child.setText(3, item.title)
child.setText(4, item.album)