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_CURRENT_SKIN, resolveFilename, fileExists
8 from enigma import RT_HALIGN_LEFT, eListboxPythonMultiContent, \
9 eServiceReference, eServiceCenter, gFont
10 from Tools.LoadPixmap import LoadPixmap
35 def FileEntryComponent(name, absolute = None, isDir = False):
36 res = [ (absolute, isDir) ]
37 res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
39 png = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "extensions/directory.png"))
41 extension = name.split('.')
42 extension = extension[-1].lower()
43 if EXTENSIONS.has_key(extension):
44 png = LoadPixmap(resolveFilename(SCOPE_CURRENT_SKIN, "extensions/" + EXTENSIONS[extension] + ".png"))
48 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
52 class FileList(MenuList):
53 def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
54 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
55 self.additional_extensions = additionalExtensions
57 self.current_directory = None
58 self.current_mountpoint = None
59 self.useServiceRef = useServiceRef
60 self.showDirectories = showDirectories
61 self.showMountpoints = showMountpoints
62 self.showFiles = showFiles
64 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
65 self.matchingPattern = matchingPattern
66 self.inhibitDirs = inhibitDirs or []
67 self.inhibitMounts = inhibitMounts or []
69 self.refreshMountpoints()
70 self.changeDir(directory)
71 self.l.setFont(0, gFont("Regular", 18))
72 self.l.setItemHeight(23)
73 self.serviceHandler = eServiceCenter.getInstance()
75 def refreshMountpoints(self):
76 self.mountpoints = [os_path.join(p.mountpoint, "") for p in harddiskmanager.getMountedPartitions()]
77 self.mountpoints.sort(reverse = True)
79 def getMountpoint(self, file):
80 file = os_path.join(os_path.realpath(file), "")
81 for m in self.mountpoints:
82 if file.startswith(m):
86 def getMountpointLink(self, file):
87 if os_path.realpath(file) == file:
88 return self.getMountpoint(file)
92 mp = self.getMountpoint(file)
94 file = os_path.dirname(file)
95 while last != "/" and mp == self.getMountpoint(file):
97 file = os_path.dirname(file)
98 return os_path.join(last, "")
100 def getSelection(self):
101 if self.l.getCurrentSelection() is None:
103 return self.l.getCurrentSelection()[0]
105 def getCurrentEvent(self):
106 l = self.l.getCurrentSelection()
107 if not l or l[0][1] == True:
110 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
112 def getFileList(self):
115 def inParentDirs(self, dir, parents):
116 dir = os_path.realpath(dir)
118 if dir.startswith(p):
122 def changeDir(self, directory, select = None):
125 # if we are just entering from the list of mount points:
126 if self.current_directory is None:
127 if directory and self.showMountpoints:
128 self.current_mountpoint = self.getMountpointLink(directory)
130 self.current_mountpoint = None
131 self.current_directory = directory
135 if directory is None and self.showMountpoints: # present available mountpoints
136 for p in harddiskmanager.getMountedPartitions():
137 path = os_path.join(p.mountpoint, "")
138 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
139 self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
142 elif directory is None:
145 elif self.useServiceRef:
146 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
147 if self.additional_extensions:
148 root.setName(self.additional_extensions)
149 serviceHandler = eServiceCenter.getInstance()
150 list = serviceHandler.list(root)
157 if s.flags & s.mustDescent:
158 directories.append(s.getPath())
164 if fileExists(directory):
166 files = listdir(directory)
172 if os_path.isdir(directory + x):
173 directories.append(directory + x + "/")
176 if directory is not None and self.showDirectories and not self.isTop:
177 if directory == self.current_mountpoint and self.showMountpoints:
178 self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
179 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
180 self.list.append(FileEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
182 if self.showDirectories:
183 for x in directories:
184 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
185 name = x.split('/')[-2]
186 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
190 if self.useServiceRef:
192 name = path.split('/')[-1]
197 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
198 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
200 if self.showMountpoints and len(self.list) == 0:
201 self.list.append(FileEntryComponent(name = _("nothing connected"), absolute = None, isDir = False))
203 self.l.setList(self.list)
205 if select is not None:
211 if isinstance(p, eServiceReference):
218 def getCurrentDirectory(self):
219 return self.current_directory
221 def canDescent(self):
222 if self.getSelection() is None:
224 return self.getSelection()[1]
227 if self.getSelection() is None:
229 self.changeDir(self.getSelection()[0], select = self.current_directory)
231 def getFilename(self):
232 if self.getSelection() is None:
234 x = self.getSelection()[0]
235 if isinstance(x, eServiceReference):
239 def getServiceRef(self):
240 if self.getSelection() is None:
242 x = self.getSelection()[0]
243 if isinstance(x, eServiceReference):
248 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
251 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
254 self.changeDir(self.current_directory, self.getFilename())
256 def partitionListChanged(self, action, device):
257 self.refreshMountpoints()
258 if self.current_directory is None:
262 def MultiFileSelectEntryComponent(name, absolute = None, isDir = False, selected = False):
263 res = [ (absolute, isDir, selected, name) ]
264 res.append((eListboxPythonMultiContent.TYPE_TEXT, 55, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
266 png = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "extensions/directory.png"))
268 extension = name.split('.')
269 extension = extension[-1].lower()
270 if EXTENSIONS.has_key(extension):
271 png = LoadPixmap(resolveFilename(SCOPE_CURRENT_SKIN, "extensions/" + EXTENSIONS[extension] + ".png"))
275 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 30, 2, 20, 20, png))
277 if not name.startswith('<'):
278 if selected is False:
279 icon = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "skin_default/icons/lock_off.png"))
280 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
282 icon = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "skin_default/icons/lock_on.png"))
283 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
288 class MultiFileSelectList(FileList):
289 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):
290 self.selectedFiles = preselectedFiles
291 if self.selectedFiles is None:
292 self.selectedFiles = []
293 FileList.__init__(self, directory, showMountpoints = showMountpoints, matchingPattern = matchingPattern, showDirectories = showDirectories, showFiles = showFiles, useServiceRef = useServiceRef, inhibitDirs = inhibitDirs, inhibitMounts = inhibitMounts, isTop = isTop, enableWrapAround = enableWrapAround, additionalExtensions = additionalExtensions)
294 self.changeDir(directory)
295 self.l.setItemHeight(25)
296 self.l.setFont(0, gFont("Regular", 20))
297 self.onSelectionChanged = [ ]
299 def selectionChanged(self):
300 for f in self.onSelectionChanged:
303 def changeSelectionState(self):
304 idx = self.l.getCurrentSelectionIndex()
309 if x[0][3].startswith('<'):
313 realPathname = x[0][0]
315 realPathname = self.current_directory + x[0][0]
318 for entry in self.selectedFiles:
319 if entry == realPathname:
320 self.selectedFiles.remove(entry)
324 alreadyinList = False
325 for entry in self.selectedFiles:
326 if entry == realPathname:
328 if not alreadyinList:
329 self.selectedFiles.append(realPathname)
330 newList.append(MultiFileSelectEntryComponent(name = x[0][3], absolute = x[0][0], isDir = x[0][1], selected = SelectState ))
337 self.l.setList(self.list)
339 def getSelectedList(self):
340 return self.selectedFiles
342 def changeDir(self, directory, select = None):
345 # if we are just entering from the list of mount points:
346 if self.current_directory is None:
347 if directory and self.showMountpoints:
348 self.current_mountpoint = self.getMountpointLink(directory)
350 self.current_mountpoint = None
351 self.current_directory = directory
355 if directory is None and self.showMountpoints: # present available mountpoints
356 for p in harddiskmanager.getMountedPartitions():
357 path = os_path.join(p.mountpoint, "")
358 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
359 self.list.append(MultiFileSelectEntryComponent(name = p.description, absolute = path, isDir = True))
362 elif directory is None:
365 elif self.useServiceRef:
366 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
367 if self.additional_extensions:
368 root.setName(self.additional_extensions)
369 serviceHandler = eServiceCenter.getInstance()
370 list = serviceHandler.list(root)
377 if s.flags & s.mustDescent:
378 directories.append(s.getPath())
384 if fileExists(directory):
386 files = listdir(directory)
392 if os_path.isdir(directory + x):
393 directories.append(directory + x + "/")
396 if directory is not None and self.showDirectories and not self.isTop:
397 if directory == self.current_mountpoint and self.showMountpoints:
398 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
399 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
400 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
402 if self.showDirectories:
403 for x in directories:
404 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
405 name = x.split('/')[-2]
406 alreadySelected = False
407 for entry in self.selectedFiles:
409 alreadySelected = True
411 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = True))
413 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = False))
417 if self.useServiceRef:
419 name = path.split('/')[-1]
424 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
425 alreadySelected = False
426 for entry in self.selectedFiles:
427 if os_path.basename(entry) == x:
428 alreadySelected = True
430 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = True))
432 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = False))
434 self.l.setList(self.list)
436 if select is not None:
442 if isinstance(p, eServiceReference):