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
27 def FileEntryComponent(name, absolute = None, isDir = False):
28 res = [ (absolute, isDir) ]
29 res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
31 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
33 extension = name.split('.')
34 extension = extension[-1].lower()
35 if EXTENSIONS.has_key(extension):
36 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
40 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
44 class FileList(MenuList):
45 def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, enableWrapAround = False, additionalExtensions = None):
46 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
47 self.additional_extensions = additionalExtensions
49 self.current_directory = None
50 self.current_mountpoint = None
51 self.useServiceRef = useServiceRef
52 self.showDirectories = showDirectories
53 self.showMountpoints = showMountpoints
54 self.showFiles = showFiles
55 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
56 self.matchingPattern = matchingPattern
57 self.inhibitDirs = inhibitDirs or []
58 self.inhibitMounts = inhibitMounts or []
60 self.refreshMountpoints()
61 self.changeDir(directory)
62 self.l.setFont(0, gFont("Regular", 18))
63 self.l.setItemHeight(23)
64 self.serviceHandler = eServiceCenter.getInstance()
66 def refreshMountpoints(self):
67 self.mountpoints = [os_path.join(p.mountpoint, "") for p in harddiskmanager.getMountedPartitions()]
68 self.mountpoints.sort(reverse = True)
70 def getMountpoint(self, file):
71 file = os_path.join(os_path.realpath(file), "")
72 for m in self.mountpoints:
73 if file.startswith(m):
77 def getMountpointLink(self, file):
78 if os_path.realpath(file) == file:
79 return self.getMountpoint(file)
83 mp = self.getMountpoint(file)
85 file = os_path.dirname(file)
86 while last != "/" and mp == self.getMountpoint(file):
88 file = os_path.dirname(file)
89 return os_path.join(last, "")
91 def getSelection(self):
92 if self.l.getCurrentSelection() is None:
94 return self.l.getCurrentSelection()[0]
96 def getCurrentEvent(self):
97 l = self.l.getCurrentSelection()
98 if not l or l[0][1] == True:
101 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
103 def getFileList(self):
106 def inParentDirs(self, dir, parents):
107 dir = os_path.realpath(dir)
109 if dir.startswith(p):
113 def changeDir(self, directory, select = None):
116 if directory and not os_path.isdir(directory):
118 # if we are just entering from the list of mount points:
119 if self.current_directory is None:
120 if directory and self.showMountpoints:
121 self.current_mountpoint = self.getMountpointLink(directory)
123 self.current_mountpoint = None
124 self.current_directory = directory
128 if directory is None and self.showMountpoints: # present available mountpoints
129 for p in harddiskmanager.getMountedPartitions():
130 path = os_path.join(p.mountpoint,"")
131 if not self.inhibitMounts or ((not path in self.inhibitMounts) and (not self.inParentDirs(path, self.inhibitDirs))):
132 self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
135 elif self.useServiceRef:
136 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
137 if self.additional_extensions:
138 root.setName(self.additional_extensions)
139 serviceHandler = eServiceCenter.getInstance()
140 list = serviceHandler.list(root)
147 if s.flags & s.mustDescent:
148 directories.append(s.getPath())
154 if os_path.exists(directory):
155 files = listdir(directory)
159 if os_path.isdir(directory + x):
160 directories.append(directory + x + "/")
163 if directory is not None and self.showDirectories:
164 if directory == self.current_mountpoint and self.showMountpoints:
165 self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
166 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
167 self.list.append(FileEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
169 if self.showDirectories:
170 for x in directories:
171 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
172 name = x.split('/')[-2]
173 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
177 if self.useServiceRef:
179 name = path.split('/')[-1]
184 if (self.matchingPattern is None) or re.compile(self.matchingPattern).search(path):
185 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
187 self.l.setList(self.list)
189 if select is not None:
195 if isinstance(p, eServiceReference):
202 def getCurrentDirectory(self):
203 return self.current_directory
205 def canDescent(self):
206 if self.getSelection() is None:
208 return self.getSelection()[1]
211 if self.getSelection() is None:
213 self.changeDir(self.getSelection()[0], select = self.current_directory)
215 def getFilename(self):
216 if self.getSelection() is None:
218 x = self.getSelection()[0]
219 if isinstance(x, eServiceReference):
223 def getServiceRef(self):
224 if self.getSelection() is None:
226 x = self.getSelection()[0]
227 if isinstance(x, eServiceReference):
232 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
235 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
238 self.changeDir(self.current_directory, self.getFilename())
240 def partitionListChanged(self, action, device):
241 self.refreshMountpoints()
242 if self.current_directory is None: