From 4530a9ff09ddf4aebd3b4d82b28ccb161b9f2792 Mon Sep 17 00:00:00 2001 From: Simon Legner Date: Sun, 25 Dec 2016 11:38:32 +0100 Subject: [PATCH] Use iterfiles in BaseTreeView#mimeData --- picard/ui/itemviews.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/picard/ui/itemviews.py b/picard/ui/itemviews.py index 8823876ee..093a160de 100644 --- a/picard/ui/itemviews.py +++ b/picard/ui/itemviews.py @@ -446,14 +446,8 @@ class BaseTreeView(QtGui.QTreeWidget): obj = item.obj if isinstance(obj, Album): album_ids.append(str(obj.id)) - elif isinstance(obj, Track): - files.extend(url(file.filename) for file in obj.linked_files) - elif isinstance(obj, File): - files.append(url(obj.filename)) - elif isinstance(obj, Cluster): - files.extend(url(file.filename) for file in obj.files) - elif isinstance(obj, ClusterList): - files.extend(url(file.filename) for cluster in obj for file in cluster.files) + elif obj.iterfiles: + files.extend([url(f.filename) for f in obj.iterfiles()]) mimeData = QtCore.QMimeData() mimeData.setData("application/picard.album-list", "\n".join(album_ids)) if files: