1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
|
from gtcache import gettext as _
import app
import dialogs
import indexes
import menu
import playlist
import util
import views
from database import DDBObject
from databasehelper import makeSimpleGetSet
class FolderBase(DDBObject):
"""Base class for ChannelFolder and Playlist folder classes."""
def __init__(self, title):
self.title = title
self.expanded = True
DDBObject.__init__(self)
getTitle, setTitle = makeSimpleGetSet('title')
def getExpanded(self):
self.confirmDBThread()
return self.expanded
def setExpanded(self, newExpanded):
self.confirmDBThread()
self.expanded = newExpanded
self.signalChange()
for child in self.getChildrenView():
child.signalChange(needsSave=False)
def getNextTab(self):
"""Get the first tab that isn't in this folder our. If there are no
items afterwards, return None.
"""
anchorItem = None
seenSelf = False
# Find the tab directly after this folder and move the tabs above
# that one.
for tab in self.getTabOrder().getView():
if not seenSelf and tab.obj is self:
seenSelf = True
elif seenSelf and tab.obj.getFolder() is not self:
return tab.obj
return None
def handleDNDAppend(self, draggedIDs):
tabOrder = self.getTabOrder()
for id in draggedIDs:
tab = tabOrder.tabView.getObjectByID(id)
tab.obj.setFolder(self)
tabOrder.moveTabs(self.getNextTab(), draggedIDs)
selection = app.controller.selection.tabListSelection
if len(selection.currentSelection) == 0:
# we appended tabs to a non-expanded folder and now nothing is
# selected. Select that folder.
app.controller.selection.selectItem('tablist', tabOrder.tabView,
self.getID(), False, False)
self.signalChange()
def rename(self):
def callback(dialog):
if self.idExists() and dialog.choice == dialogs.BUTTON_OK:
self.setTitle(dialog.value)
dialogs.TextEntryDialog(self.renameTitle(), self.renameText(),
dialogs.BUTTON_OK, dialogs.BUTTON_CANCEL).run(callback)
def remove(self):
children = [child for child in self.getChildrenView()]
for child in children:
child.remove()
DDBObject.remove(self)
# getFolder and setFolder are here so that channels/playlists and folders
# have a consistent API. They don't do much since we don't allow nested
# folders.
def getFolder(self):
return None
def setFolder(self, newFolder):
if newFolder is not None:
raise TypeError("Nested folders not allowed")
def renameTitle(self):
"""Return the title to use for the rename dialog"""
raise NotImplementedError()
def renameText(self):
"""Return the description text to use for the rename dialog"""
raise NotImplementedError()
def getTabOrder(self):
"""Return the TabOrder object that this folder belongs to."""
raise NotImplementedError()
def getChildrenView(self):
"""Return the children of this folder."""
raise NotImplementedError()
class ChannelFolder(FolderBase):
def renameTitle(self):
return _("Rename Channel Folder")
def renameText(self):
return _("Enter a new name for the channel folder %s" %
self.getTitle())
def getTabOrder(self):
return util.getSingletonDDBObject(views.channelTabOrder)
def getChildrenView(self):
return views.feeds.filterWithIndex(indexes.byFolder, self)
def hasDownloadedItems(self):
for feed in self.getChildrenView():
if feed.hasDownloadedItems():
return True
return False
def makeContextMenu(self, templateName, view):
return menu.makeMenu([
(self.rename, _('Rename Channel Folder')),
(lambda: app.controller.removeFeed(self), _('Remove')),
])
# Returns true iff unwatched should be shown
def showU(self):
return self.numUnwatched() > 0
# Returns string with number of unwatched videos in feed
def numUnwatched(self):
unwatched = 0
for child in self.getChildrenView():
if child.showU():
unwatched += child.unwatched
return unwatched
# Returns true iff unwatched should be shown
def showA(self):
return self.numAvailable() > 0
# Returns string with number of unwatched videos in feed
def numAvailable(self):
available = 0
for child in self.getChildrenView():
if child.showA():
available += child.available
return available
class PlaylistFolder(FolderBase, playlist.PlaylistMixin):
def __init__(self, title):
self.item_ids = []
self.setupTrackedItemView()
FolderBase.__init__(self, title)
def onRestore(self):
self.setupTrackedItemView()
def handleDNDAppend(self, draggedIDs):
FolderBase.handleDNDAppend(self, draggedIDs)
for id in draggedIDs:
tab = self.getTabOrder().tabView.getObjectByID(id)
for item in tab.obj.getView():
if item.getID() not in self.trackedItems:
self.trackedItems.appendID(item.getID())
self.signalChange()
def checkItemIDRemoved(self, id):
index = indexes.playlistsByItemAndFolderID
value = (id, self.getID())
view = views.playlists.filterWithIndex(index, value)
if view.len() == 0 and id in self.trackedItems:
self.removeID(id)
def makeContextMenu(self, templateName, view):
return menu.makeMenu([
(self.rename, _('Rename Playlist Folder')),
(lambda: app.controller.removePlaylist(self), _('Remove')),
])
def renameTitle(self):
return _("Rename Playlist Folder")
def renameText(self):
return _("Enter a new name for the playlist folder %s" %
self.getTitle())
def getTabOrder(self):
return util.getSingletonDDBObject(views.playlistTabOrder)
def getChildrenView(self):
return views.playlists.filterWithIndex(indexes.byFolder, self)
def createNewChannelFolder(childIDs=None):
title = _("Create Channel Folder")
description = _("Enter a name for the new channel folder")
def callback(dialog):
if dialog.choice == dialogs.BUTTON_CREATE:
folder = ChannelFolder(dialog.value)
app.controller.selection.selectTabByObject(folder)
if childIDs:
folder.handleDNDAppend(childIDs)
dialogs.TextEntryDialog(title, description, dialogs.BUTTON_CREATE,
dialogs.BUTTON_CANCEL).run(callback)
def createNewPlaylistFolder(childIDs=None):
title = _("Create Playlist Folder")
description = _("Enter a name for the new playlist folder")
def callback(dialog):
if dialog.choice == dialogs.BUTTON_CREATE:
folder = PlaylistFolder(dialog.value)
app.controller.selection.selectTabByObject(folder)
if childIDs:
folder.handleDNDAppend(childIDs)
dialogs.TextEntryDialog(title, description, dialogs.BUTTON_CREATE,
dialogs.BUTTON_CANCEL).run(callback)
|
