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
31 def FileEntryComponent(name, absolute = None, isDir = False):
32 res = [ (absolute, isDir) ]
33 res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
35 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
37 extension = name.split('.')
38 extension = extension[-1].lower()
39 if EXTENSIONS.has_key(extension):
40 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
44 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
48 class FileList(MenuList):
49 def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
50 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
51 self.additional_extensions = additionalExtensions
53 self.current_directory = None
54 self.current_mountpoint = None
55 self.useServiceRef = useServiceRef
56 self.showDirectories = showDirectories
57 self.showMountpoints = showMountpoints
58 self.showFiles = showFiles
60 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
61 self.matchingPattern = matchingPattern
62 self.inhibitDirs = inhibitDirs or []
63 self.inhibitMounts = inhibitMounts or []
65 self.refreshMountpoints()
66 self.changeDir(directory)
67 self.l.setFont(0, gFont("Regular", 18))
68 self.l.setItemHeight(23)
69 self.serviceHandler = eServiceCenter.getInstance()
71 def refreshMountpoints(self):
72 self.mountpoints = [os_path.join(p.mountpoint, "") for p in harddiskmanager.getMountedPartitions()]
73 self.mountpoints.sort(reverse = True)
75 def getMountpoint(self, file):
76 file = os_path.join(os_path.realpath(file), "")
77 for m in self.mountpoints:
78 if file.startswith(m):
82 def getMountpointLink(self, file):
83 if os_path.realpath(file) == file:
84 return self.getMountpoint(file)
88 mp = self.getMountpoint(file)
90 file = os_path.dirname(file)
91 while last != "/" and mp == self.getMountpoint(file):
93 file = os_path.dirname(file)
94 return os_path.join(last, "")
96 def getSelection(self):
97 if self.l.getCurrentSelection() is None:
99 return self.l.getCurrentSelection()[0]
101 def getCurrentEvent(self):
102 l = self.l.getCurrentSelection()
103 if not l or l[0][1] == True:
106 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
108 def getFileList(self):
111 def inParentDirs(self, dir, parents):
112 dir = os_path.realpath(dir)
114 if dir.startswith(p):
118 def changeDir(self, directory, select = None):
121 if directory and not os_path.isdir(directory):
123 # if we are just entering from the list of mount points:
124 if self.current_directory is None:
125 if directory and self.showMountpoints:
126 self.current_mountpoint = self.getMountpointLink(directory)
128 self.current_mountpoint = None
129 self.current_directory = directory
133 if directory is None and self.showMountpoints: # present available mountpoints
134 for p in harddiskmanager.getMountedPartitions():
135 path = os_path.join(p.mountpoint, "")
136 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
137 self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
140 elif self.useServiceRef:
141 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
142 if self.additional_extensions:
143 root.setName(self.additional_extensions)
144 serviceHandler = eServiceCenter.getInstance()
145 list = serviceHandler.list(root)
152 if s.flags & s.mustDescent:
153 directories.append(s.getPath())
159 if os_path.exists(directory):
160 files = listdir(directory)
164 if os_path.isdir(directory + x):
165 directories.append(directory + x + "/")
168 if directory is not None and self.showDirectories and not self.isTop:
169 if directory == self.current_mountpoint and self.showMountpoints:
170 self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
171 elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
172 self.list.append(FileEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
174 if self.showDirectories:
175 for x in directories:
176 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
177 name = x.split('/')[-2]
178 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
182 if self.useServiceRef:
184 name = path.split('/')[-1]
189 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
190 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
192 self.l.setList(self.list)
194 if select is not None:
200 if isinstance(p, eServiceReference):
207 def getCurrentDirectory(self):
208 return self.current_directory
210 def canDescent(self):
211 if self.getSelection() is None:
213 return self.getSelection()[1]
216 if self.getSelection() is None:
218 self.changeDir(self.getSelection()[0], select = self.current_directory)
220 def getFilename(self):
221 if self.getSelection() is None:
223 x = self.getSelection()[0]
224 if isinstance(x, eServiceReference):
228 def getServiceRef(self):
229 if self.getSelection() is None:
231 x = self.getSelection()[0]
232 if isinstance(x, eServiceReference):
237 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
240 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
243 self.changeDir(self.current_directory, self.getFilename())
245 def partitionListChanged(self, action, device):
246 self.refreshMountpoints()
247 if self.current_directory is None: