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
40 def FileEntryComponent(name, absolute = None, isDir = False):
41 res = [ (absolute, isDir) ]
42 res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
44 png = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "extensions/directory.png"))
46 extension = name.split('.')
47 extension = extension[-1].lower()
48 if EXTENSIONS.has_key(extension):
49 png = LoadPixmap(resolveFilename(SCOPE_CURRENT_SKIN, "extensions/" + EXTENSIONS[extension] + ".png"))
53 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
57 class FileList(MenuList):
58 def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
59 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
60 self.additional_extensions = additionalExtensions
62 self.current_directory = None
63 self.current_mountpoint = None
64 self.useServiceRef = useServiceRef
65 self.showDirectories = showDirectories
66 self.showMountpoints = showMountpoints
67 self.showFiles = showFiles
69 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
70 self.matchingPattern = matchingPattern
71 self.inhibitDirs = inhibitDirs or []
72 self.inhibitMounts = inhibitMounts or []
74 self.refreshMountpoints()
75 self.changeDir(directory)
76 self.l.setFont(0, gFont("Regular", 18))
77 self.l.setItemHeight(23)
78 self.serviceHandler = eServiceCenter.getInstance()
80 def refreshMountpoints(self):
81 self.mountpoints = [os_path.join(p.mountpoint, "") for p in harddiskmanager.getMountedPartitions()]
82 self.mountpoints.sort(reverse = True)
84 def getMountpoint(self, file):
85 file = os_path.join(os_path.realpath(file), "")
86 for m in self.mountpoints:
87 if file.startswith(m):
91 def getMountpointLink(self, file):
92 if os_path.realpath(file) == file:
93 return self.getMountpoint(file)
97 mp = self.getMountpoint(file)
99 file = os_path.dirname(file)
100 while last != "/" and mp == self.getMountpoint(file):
102 file = os_path.dirname(file)
103 return os_path.join(last, "")
105 def getSelection(self):
106 if self.l.getCurrentSelection() is None:
108 return self.l.getCurrentSelection()[0]
110 def getCurrentEvent(self):
111 l = self.l.getCurrentSelection()
112 if not l or l[0][1] == True:
115 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
117 def getFileList(self):
120 def inParentDirs(self, dir, parents):
121 dir = os_path.realpath(dir)
123 if dir.startswith(p):
127 def changeDir(self, directory, select = None):
130 # if we are just entering from the list of mount points:
131 if self.current_directory is None:
132 if directory and self.showMountpoints:
133 self.current_mountpoint = self.getMountpointLink(directory)
135 self.current_mountpoint = None
136 self.current_directory = directory
140 if directory is None and self.showMountpoints: # present available mountpoints
141 for p in harddiskmanager.getMountedPartitions():
142 path = os_path.join(p.mountpoint, "")
143 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
144 self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
147 elif directory is None:
150 elif self.useServiceRef:
151 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
152 if self.additional_extensions:
153 root.setName(self.additional_extensions)
154 serviceHandler = eServiceCenter.getInstance()
155 list = serviceHandler.list(root)
162 if s.flags & s.mustDescent:
163 directories.append(s.getPath())
169 if fileExists(directory):
171 files = listdir(directory)
177 if os_path.isdir(directory + x):
178 directories.append(directory + x + "/")
181 if directory is not None and self.showDirectories and not self.isTop:
182 if directory == self.current_mountpoint and self.showMountpoints:
183 self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
184 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
185 self.list.append(FileEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
187 if self.showDirectories:
188 for x in directories:
189 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
190 name = x.split('/')[-2]
191 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
195 if self.useServiceRef:
197 name = path.split('/')[-1]
202 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
203 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
205 if self.showMountpoints and len(self.list) == 0:
206 self.list.append(FileEntryComponent(name = _("nothing connected"), absolute = None, isDir = False))
208 self.l.setList(self.list)
210 if select is not None:
216 if isinstance(p, eServiceReference):
223 def getCurrentDirectory(self):
224 return self.current_directory
226 def canDescent(self):
227 if self.getSelection() is None:
229 return self.getSelection()[1]
232 if self.getSelection() is None:
234 self.changeDir(self.getSelection()[0], select = self.current_directory)
236 def getFilename(self):
237 if self.getSelection() is None:
239 x = self.getSelection()[0]
240 if isinstance(x, eServiceReference):
244 def getServiceRef(self):
245 if self.getSelection() is None:
247 x = self.getSelection()[0]
248 if isinstance(x, eServiceReference):
253 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
256 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
259 self.changeDir(self.current_directory, self.getFilename())
261 def partitionListChanged(self, action, device):
262 self.refreshMountpoints()
263 if self.current_directory is None:
267 def MultiFileSelectEntryComponent(name, absolute = None, isDir = False, selected = False):
268 res = [ (absolute, isDir, selected, name) ]
269 res.append((eListboxPythonMultiContent.TYPE_TEXT, 55, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
271 png = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "extensions/directory.png"))
273 extension = name.split('.')
274 extension = extension[-1].lower()
275 if EXTENSIONS.has_key(extension):
276 png = LoadPixmap(resolveFilename(SCOPE_CURRENT_SKIN, "extensions/" + EXTENSIONS[extension] + ".png"))
280 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 30, 2, 20, 20, png))
282 if not name.startswith('<'):
283 if selected is False:
284 icon = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "skin_default/icons/lock_off.png"))
285 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
287 icon = LoadPixmap(cached=True, path=resolveFilename(SCOPE_CURRENT_SKIN, "skin_default/icons/lock_on.png"))
288 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
293 class MultiFileSelectList(FileList):
294 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):
295 self.selectedFiles = preselectedFiles
296 if self.selectedFiles is None:
297 self.selectedFiles = []
298 FileList.__init__(self, directory, showMountpoints = showMountpoints, matchingPattern = matchingPattern, showDirectories = showDirectories, showFiles = showFiles, useServiceRef = useServiceRef, inhibitDirs = inhibitDirs, inhibitMounts = inhibitMounts, isTop = isTop, enableWrapAround = enableWrapAround, additionalExtensions = additionalExtensions)
299 self.changeDir(directory)
300 self.l.setItemHeight(25)
301 self.l.setFont(0, gFont("Regular", 20))
302 self.onSelectionChanged = [ ]
304 def selectionChanged(self):
305 for f in self.onSelectionChanged:
308 def changeSelectionState(self):
309 idx = self.l.getCurrentSelectionIndex()
314 if x[0][3].startswith('<'):
318 realPathname = x[0][0]
320 realPathname = self.current_directory + x[0][0]
323 for entry in self.selectedFiles:
324 if entry == realPathname:
325 self.selectedFiles.remove(entry)
329 alreadyinList = False
330 for entry in self.selectedFiles:
331 if entry == realPathname:
333 if not alreadyinList:
334 self.selectedFiles.append(realPathname)
335 newList.append(MultiFileSelectEntryComponent(name = x[0][3], absolute = x[0][0], isDir = x[0][1], selected = SelectState ))
342 self.l.setList(self.list)
344 def getSelectedList(self):
345 return self.selectedFiles
347 def changeDir(self, directory, select = None):
350 # if we are just entering from the list of mount points:
351 if self.current_directory is None:
352 if directory and self.showMountpoints:
353 self.current_mountpoint = self.getMountpointLink(directory)
355 self.current_mountpoint = None
356 self.current_directory = directory
360 if directory is None and self.showMountpoints: # present available mountpoints
361 for p in harddiskmanager.getMountedPartitions():
362 path = os_path.join(p.mountpoint, "")
363 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
364 self.list.append(MultiFileSelectEntryComponent(name = p.description, absolute = path, isDir = True))
367 elif directory is None:
370 elif self.useServiceRef:
371 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
372 if self.additional_extensions:
373 root.setName(self.additional_extensions)
374 serviceHandler = eServiceCenter.getInstance()
375 list = serviceHandler.list(root)
382 if s.flags & s.mustDescent:
383 directories.append(s.getPath())
389 if fileExists(directory):
391 files = listdir(directory)
397 if os_path.isdir(directory + x):
398 directories.append(directory + x + "/")
401 if directory is not None and self.showDirectories and not self.isTop:
402 if directory == self.current_mountpoint and self.showMountpoints:
403 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
404 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
405 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
407 if self.showDirectories:
408 for x in directories:
409 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
410 name = x.split('/')[-2]
411 alreadySelected = False
412 for entry in self.selectedFiles:
414 alreadySelected = True
416 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = True))
418 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = False))
422 if self.useServiceRef:
424 name = path.split('/')[-1]
429 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
430 alreadySelected = False
431 for entry in self.selectedFiles:
432 if os_path.basename(entry) == x:
433 alreadySelected = True
435 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = True))
437 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = False))
439 self.l.setList(self.list)
441 if select is not None:
447 if isinstance(p, eServiceReference):