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
28 def FileEntryComponent(name, absolute = None, isDir = False):
29 res = [ (absolute, isDir) ]
30 res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
32 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
34 extension = name.split('.')
35 extension = extension[-1].lower()
36 if EXTENSIONS.has_key(extension):
37 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
41 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
45 class FileList(MenuList):
46 def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
47 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
48 self.additional_extensions = additionalExtensions
50 self.current_directory = None
51 self.current_mountpoint = None
52 self.useServiceRef = useServiceRef
53 self.showDirectories = showDirectories
54 self.showMountpoints = showMountpoints
55 self.showFiles = showFiles
57 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
58 self.matchingPattern = matchingPattern
59 self.inhibitDirs = inhibitDirs or []
60 self.inhibitMounts = inhibitMounts or []
62 self.refreshMountpoints()
63 self.changeDir(directory)
64 self.l.setFont(0, gFont("Regular", 18))
65 self.l.setItemHeight(23)
66 self.serviceHandler = eServiceCenter.getInstance()
68 def refreshMountpoints(self):
69 self.mountpoints = [os_path.join(p.mountpoint, "") for p in harddiskmanager.getMountedPartitions()]
70 self.mountpoints.sort(reverse = True)
72 def getMountpoint(self, file):
73 file = os_path.join(os_path.realpath(file), "")
74 for m in self.mountpoints:
75 if file.startswith(m):
79 def getMountpointLink(self, file):
80 if os_path.realpath(file) == file:
81 return self.getMountpoint(file)
85 mp = self.getMountpoint(file)
87 file = os_path.dirname(file)
88 while last != "/" and mp == self.getMountpoint(file):
90 file = os_path.dirname(file)
91 return os_path.join(last, "")
93 def getSelection(self):
94 if self.l.getCurrentSelection() is None:
96 return self.l.getCurrentSelection()[0]
98 def getCurrentEvent(self):
99 l = self.l.getCurrentSelection()
100 if not l or l[0][1] == True:
103 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
105 def getFileList(self):
108 def inParentDirs(self, dir, parents):
109 dir = os_path.realpath(dir)
111 if dir.startswith(p):
115 def changeDir(self, directory, select = None):
118 if directory and not os_path.isdir(directory):
120 # if we are just entering from the list of mount points:
121 if self.current_directory is None:
122 if directory and self.showMountpoints:
123 self.current_mountpoint = self.getMountpointLink(directory)
125 self.current_mountpoint = None
126 self.current_directory = directory
130 if directory is None and self.showMountpoints: # present available mountpoints
131 for p in harddiskmanager.getMountedPartitions():
132 path = os_path.join(p.mountpoint, "")
133 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
134 self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
137 elif self.useServiceRef:
138 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
139 if self.additional_extensions:
140 root.setName(self.additional_extensions)
141 serviceHandler = eServiceCenter.getInstance()
142 list = serviceHandler.list(root)
149 if s.flags & s.mustDescent:
150 directories.append(s.getPath())
156 if os_path.exists(directory):
157 files = listdir(directory)
161 if os_path.isdir(directory + x):
162 directories.append(directory + x + "/")
165 if directory is not None and self.showDirectories and not self.isTop:
166 if directory == self.current_mountpoint and self.showMountpoints:
167 self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
168 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
169 self.list.append(FileEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
171 if self.showDirectories:
172 for x in directories:
173 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
174 name = x.split('/')[-2]
175 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
179 if self.useServiceRef:
181 name = path.split('/')[-1]
186 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
187 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
189 self.l.setList(self.list)
191 if select is not None:
197 if isinstance(p, eServiceReference):
204 def getCurrentDirectory(self):
205 return self.current_directory
207 def canDescent(self):
208 if self.getSelection() is None:
210 return self.getSelection()[1]
213 if self.getSelection() is None:
215 self.changeDir(self.getSelection()[0], select = self.current_directory)
217 def getFilename(self):
218 if self.getSelection() is None:
220 x = self.getSelection()[0]
221 if isinstance(x, eServiceReference):
225 def getServiceRef(self):
226 if self.getSelection() is None:
228 x = self.getSelection()[0]
229 if isinstance(x, eServiceReference):
234 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
237 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
240 self.changeDir(self.current_directory, self.getFilename())
242 def partitionListChanged(self, action, device):
243 self.refreshMountpoints()
244 if self.current_directory is None: