1 from re import compile as re_compile
2 from os import path as os_path, listdir
3 from MenuList import MenuList
4 from Components.Harddisk import harddiskmanager
6 from Tools.Directories import SCOPE_SKIN_IMAGE, resolveFilename
8 from enigma import RT_HALIGN_LEFT, eListboxPythonMultiContent, \
9 eServiceReference, eServiceCenter, gFont
10 from Tools.LoadPixmap import LoadPixmap
34 def FileEntryComponent(name, absolute = None, isDir = False):
35 res = [ (absolute, isDir) ]
36 res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
38 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
40 extension = name.split('.')
41 extension = extension[-1].lower()
42 if EXTENSIONS.has_key(extension):
43 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
47 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
51 class FileList(MenuList):
52 def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
53 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
54 self.additional_extensions = additionalExtensions
56 self.current_directory = None
57 self.current_mountpoint = None
58 self.useServiceRef = useServiceRef
59 self.showDirectories = showDirectories
60 self.showMountpoints = showMountpoints
61 self.showFiles = showFiles
63 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
64 self.matchingPattern = matchingPattern
65 self.inhibitDirs = inhibitDirs or []
66 self.inhibitMounts = inhibitMounts or []
68 self.refreshMountpoints()
69 self.changeDir(directory)
70 self.l.setFont(0, gFont("Regular", 18))
71 self.l.setItemHeight(23)
72 self.serviceHandler = eServiceCenter.getInstance()
74 def refreshMountpoints(self):
75 self.mountpoints = [os_path.join(p.mountpoint, "") for p in harddiskmanager.getMountedPartitions()]
76 self.mountpoints.sort(reverse = True)
78 def getMountpoint(self, file):
79 file = os_path.join(os_path.realpath(file), "")
80 for m in self.mountpoints:
81 if file.startswith(m):
85 def getMountpointLink(self, file):
86 if os_path.realpath(file) == file:
87 return self.getMountpoint(file)
91 mp = self.getMountpoint(file)
93 file = os_path.dirname(file)
94 while last != "/" and mp == self.getMountpoint(file):
96 file = os_path.dirname(file)
97 return os_path.join(last, "")
99 def getSelection(self):
100 if self.l.getCurrentSelection() is None:
102 return self.l.getCurrentSelection()[0]
104 def getCurrentEvent(self):
105 l = self.l.getCurrentSelection()
106 if not l or l[0][1] == True:
109 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
111 def getFileList(self):
114 def inParentDirs(self, dir, parents):
115 dir = os_path.realpath(dir)
117 if dir.startswith(p):
121 def changeDir(self, directory, select = None):
124 # if we are just entering from the list of mount points:
125 if self.current_directory is None:
126 if directory and self.showMountpoints:
127 self.current_mountpoint = self.getMountpointLink(directory)
129 self.current_mountpoint = None
130 self.current_directory = directory
134 if directory is None and self.showMountpoints: # present available mountpoints
135 for p in harddiskmanager.getMountedPartitions():
136 path = os_path.join(p.mountpoint, "")
137 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
138 self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
141 elif directory is None:
144 elif self.useServiceRef:
145 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
146 if self.additional_extensions:
147 root.setName(self.additional_extensions)
148 serviceHandler = eServiceCenter.getInstance()
149 list = serviceHandler.list(root)
156 if s.flags & s.mustDescent:
157 directories.append(s.getPath())
163 if os_path.exists(directory):
164 files = listdir(directory)
168 if os_path.isdir(directory + x):
169 directories.append(directory + x + "/")
172 if directory is not None and self.showDirectories and not self.isTop:
173 if directory == self.current_mountpoint and self.showMountpoints:
174 self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
175 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
176 self.list.append(FileEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
178 if self.showDirectories:
179 for x in directories:
180 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
181 name = x.split('/')[-2]
182 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
186 if self.useServiceRef:
188 name = path.split('/')[-1]
193 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
194 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
196 self.l.setList(self.list)
198 if select is not None:
204 if isinstance(p, eServiceReference):
211 def getCurrentDirectory(self):
212 return self.current_directory
214 def canDescent(self):
215 if self.getSelection() is None:
217 return self.getSelection()[1]
220 if self.getSelection() is None:
222 self.changeDir(self.getSelection()[0], select = self.current_directory)
224 def getFilename(self):
225 if self.getSelection() is None:
227 x = self.getSelection()[0]
228 if isinstance(x, eServiceReference):
232 def getServiceRef(self):
233 if self.getSelection() is None:
235 x = self.getSelection()[0]
236 if isinstance(x, eServiceReference):
241 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
244 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
247 self.changeDir(self.current_directory, self.getFilename())
249 def partitionListChanged(self, action, device):
250 self.refreshMountpoints()
251 if self.current_directory is None:
255 def MultiFileSelectEntryComponent(name, absolute = None, isDir = False, selected = False):
256 res = [ (absolute, isDir, selected, name) ]
257 res.append((eListboxPythonMultiContent.TYPE_TEXT, 55, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
259 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
261 extension = name.split('.')
262 extension = extension[-1].lower()
263 if EXTENSIONS.has_key(extension):
264 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
268 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 30, 2, 20, 20, png))
270 if not name.startswith('<'):
271 if selected is False:
272 icon = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "skin_default/icons/lock_off.png"))
273 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
275 icon = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "skin_default/icons/lock_on.png"))
276 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
281 class MultiFileSelectList(FileList):
282 def __init__(self, preselectedFiles, directory, showMountpoints = False, matchingPattern = None, showDirectories = True, showFiles = True, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
283 self.selectedFiles = preselectedFiles
284 if self.selectedFiles is None:
285 self.selectedFiles = []
286 FileList.__init__(self, directory, showMountpoints = showMountpoints, matchingPattern = matchingPattern, showDirectories = showDirectories, showFiles = showFiles, useServiceRef = useServiceRef, inhibitDirs = inhibitDirs, inhibitMounts = inhibitMounts, isTop = isTop, enableWrapAround = enableWrapAround, additionalExtensions = additionalExtensions)
287 self.changeDir(directory)
288 self.l.setItemHeight(25)
289 self.l.setFont(0, gFont("Regular", 20))
290 self.onSelectionChanged = [ ]
292 def selectionChanged(self):
293 for f in self.onSelectionChanged:
296 def changeSelectionState(self):
297 idx = self.l.getCurrentSelectionIndex()
302 if x[0][3].startswith('<'):
306 realPathname = x[0][0]
308 realPathname = self.current_directory + x[0][0]
311 for entry in self.selectedFiles:
312 if entry == realPathname:
313 self.selectedFiles.remove(entry)
317 alreadyinList = False
318 for entry in self.selectedFiles:
319 if entry == realPathname:
321 if not alreadyinList:
322 self.selectedFiles.append(realPathname)
323 newList.append(MultiFileSelectEntryComponent(name = x[0][3], absolute = x[0][0], isDir = x[0][1], selected = SelectState ))
330 self.l.setList(self.list)
332 def getSelectedList(self):
333 return self.selectedFiles
335 def changeDir(self, directory, select = None):
338 # if we are just entering from the list of mount points:
339 if self.current_directory is None:
340 if directory and self.showMountpoints:
341 self.current_mountpoint = self.getMountpointLink(directory)
343 self.current_mountpoint = None
344 self.current_directory = directory
348 if directory is None and self.showMountpoints: # present available mountpoints
349 for p in harddiskmanager.getMountedPartitions():
350 path = os_path.join(p.mountpoint, "")
351 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
352 self.list.append(MultiFileSelectEntryComponent(name = p.description, absolute = path, isDir = True))
355 elif directory is None:
358 elif self.useServiceRef:
359 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
360 if self.additional_extensions:
361 root.setName(self.additional_extensions)
362 serviceHandler = eServiceCenter.getInstance()
363 list = serviceHandler.list(root)
370 if s.flags & s.mustDescent:
371 directories.append(s.getPath())
377 if os_path.exists(directory):
378 files = listdir(directory)
382 if os_path.isdir(directory + x):
383 directories.append(directory + x + "/")
386 if directory is not None and self.showDirectories and not self.isTop:
387 if directory == self.current_mountpoint and self.showMountpoints:
388 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
389 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
390 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
392 if self.showDirectories:
393 for x in directories:
394 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
395 name = x.split('/')[-2]
396 alreadySelected = False
397 for entry in self.selectedFiles:
399 alreadySelected = True
401 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = True))
403 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = False))
407 if self.useServiceRef:
409 name = path.split('/')[-1]
414 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
415 alreadySelected = False
416 for entry in self.selectedFiles:
417 if os_path.basename(entry) == x:
418 alreadySelected = True
420 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = True))
422 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = False))
424 self.l.setList(self.list)
426 if select is not None:
432 if isinstance(p, eServiceReference):