3 from os import path as os_path, listdir
4 from MenuList import MenuList
5 from Components.Harddisk import harddiskmanager
7 from Tools.Directories import SCOPE_SKIN_IMAGE, resolveFilename
9 from enigma import RT_HALIGN_LEFT, eListbox, eListboxPythonMultiContent, \
10 eServiceReference, eServiceCenter, gFont
11 from Tools.LoadPixmap import LoadPixmap
29 def FileEntryComponent(name, absolute = None, isDir = False):
30 res = [ (absolute, isDir) ]
31 res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
33 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
35 extension = name.split('.')
36 extension = extension[-1].lower()
37 if EXTENSIONS.has_key(extension):
38 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
42 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
46 class FileList(MenuList):
47 def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
48 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
49 self.additional_extensions = additionalExtensions
51 self.current_directory = None
52 self.current_mountpoint = None
53 self.useServiceRef = useServiceRef
54 self.showDirectories = showDirectories
55 self.showMountpoints = showMountpoints
56 self.showFiles = showFiles
58 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
59 self.matchingPattern = matchingPattern
60 self.inhibitDirs = inhibitDirs or []
61 self.inhibitMounts = inhibitMounts or []
63 self.refreshMountpoints()
64 self.changeDir(directory)
65 self.l.setFont(0, gFont("Regular", 18))
66 self.l.setItemHeight(23)
67 self.serviceHandler = eServiceCenter.getInstance()
69 def refreshMountpoints(self):
70 self.mountpoints = [os_path.join(p.mountpoint, "") for p in harddiskmanager.getMountedPartitions()]
71 self.mountpoints.sort(reverse = True)
73 def getMountpoint(self, file):
74 file = os_path.join(os_path.realpath(file), "")
75 for m in self.mountpoints:
76 if file.startswith(m):
80 def getMountpointLink(self, file):
81 if os_path.realpath(file) == file:
82 return self.getMountpoint(file)
86 mp = self.getMountpoint(file)
88 file = os_path.dirname(file)
89 while last != "/" and mp == self.getMountpoint(file):
91 file = os_path.dirname(file)
92 return os_path.join(last, "")
94 def getSelection(self):
95 if self.l.getCurrentSelection() is None:
97 return self.l.getCurrentSelection()[0]
99 def getCurrentEvent(self):
100 l = self.l.getCurrentSelection()
101 if not l or l[0][1] == True:
104 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
106 def getFileList(self):
109 def inParentDirs(self, dir, parents):
110 dir = os_path.realpath(dir)
112 if dir.startswith(p):
116 def changeDir(self, directory, select = None):
119 if directory and not os_path.isdir(directory):
121 # if we are just entering from the list of mount points:
122 if self.current_directory is None:
123 if directory and self.showMountpoints:
124 self.current_mountpoint = self.getMountpointLink(directory)
126 self.current_mountpoint = None
127 self.current_directory = directory
131 if directory is None and self.showMountpoints: # present available mountpoints
132 for p in harddiskmanager.getMountedPartitions():
133 path = os_path.join(p.mountpoint, "")
134 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
135 self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
138 elif self.useServiceRef:
139 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
140 if self.additional_extensions:
141 root.setName(self.additional_extensions)
142 serviceHandler = eServiceCenter.getInstance()
143 list = serviceHandler.list(root)
150 if s.flags & s.mustDescent:
151 directories.append(s.getPath())
157 if os_path.exists(directory):
158 files = listdir(directory)
162 if os_path.isdir(directory + x):
163 directories.append(directory + x + "/")
166 if directory is not None and self.showDirectories and not self.isTop:
167 if directory == self.current_mountpoint and self.showMountpoints:
168 self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
169 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
170 self.list.append(FileEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
172 if self.showDirectories:
173 for x in directories:
174 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
175 name = x.split('/')[-2]
176 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
180 if self.useServiceRef:
182 name = path.split('/')[-1]
187 if (self.matchingPattern is None) or re.compile(self.matchingPattern).search(path):
188 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
190 self.l.setList(self.list)
192 if select is not None:
198 if isinstance(p, eServiceReference):
205 def getCurrentDirectory(self):
206 return self.current_directory
208 def canDescent(self):
209 if self.getSelection() is None:
211 return self.getSelection()[1]
214 if self.getSelection() is None:
216 self.changeDir(self.getSelection()[0], select = self.current_directory)
218 def getFilename(self):
219 if self.getSelection() is None:
221 x = self.getSelection()[0]
222 if isinstance(x, eServiceReference):
226 def getServiceRef(self):
227 if self.getSelection() is None:
229 x = self.getSelection()[0]
230 if isinstance(x, eServiceReference):
235 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
238 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
241 self.changeDir(self.current_directory, self.getFilename())
243 def partitionListChanged(self, action, device):
244 self.refreshMountpoints()
245 if self.current_directory is None: