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, matchingPattern = None, useServiceRef = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
46 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
47 self.additional_extensions = additionalExtensions
48 self.mount_point = None
49 self.current_directory = None
50 self.useServiceRef = useServiceRef
51 self.showDirectories = showDirectories
52 self.showFiles = showFiles
54 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
55 self.matchingPattern = matchingPattern
56 self.changeDir(directory)
58 self.l.setFont(0, gFont("Regular", 18))
59 self.l.setItemHeight(23)
60 self.serviceHandler = eServiceCenter.getInstance()
62 def getSelection(self):
63 if self.l.getCurrentSelection() is None:
65 return self.l.getCurrentSelection()[0]
67 def getCurrentEvent(self):
68 l = self.l.getCurrentSelection()
69 if not l or l[0][1] == True:
72 return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
74 def getFileList(self):
77 def changeDir(self, directory, select = None):
80 # if we are just entering from the list of mount points:
81 if self.current_directory is None:
83 self.mount_point = None
85 # Sort Mountpoints by length (longest first)
86 sortedp = harddiskmanager.getMountedPartitions()
87 sortedp.sort(key=lambda p: 0 - len(p.mountpoint))
89 # Search for the longest matching mp (should at least match /)
91 if directory.startswith(p.mountpoint):
92 self.mount_point = p.mountpoint
93 if p.mountpoint != "/":
94 self.mount_point += "/"
96 self.current_directory = directory
100 if directory is None: # present available mountpoints
101 print "listing partitions:"
102 for p in harddiskmanager.getMountedPartitions():
103 if p.mountpoint == "/":
104 self.list.append(FileEntryComponent(name = p.description, absolute = p.mountpoint, isDir = True))
106 self.list.append(FileEntryComponent(name = p.description, absolute = p.mountpoint + "/", isDir = True))
109 elif self.useServiceRef:
110 root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
111 if self.additional_extensions:
112 root.setName(self.additional_extensions)
113 serviceHandler = eServiceCenter.getInstance()
114 list = serviceHandler.list(root)
121 if s.flags & s.mustDescent:
122 directories.append(s.getPath())
125 print s.getName(), s.flags
129 files = listdir(directory)
133 if os_path.isdir(directory + x):
134 directories.append(directory + x + "/")
137 if directory is not None and self.showDirectories and not self.isTop:
138 if directory == self.mount_point:
139 self.list.append(FileEntryComponent(name = ".. (" +_("List of Storage Devices") + ")", absolute = None, isDir = True))
141 self.list.append(FileEntryComponent(name = "..", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
143 if self.showDirectories:
144 for x in directories:
145 name = x.split('/')[-2]
146 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
150 if self.useServiceRef:
152 name = path.split('/')[-1]
157 if self.matchingPattern is not None:
158 if re.compile(self.matchingPattern).search(path):
159 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
161 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
163 self.l.setList(self.list)
165 if select is not None:
171 if isinstance(p, eServiceReference):
178 def getCurrentDirectory(self):
179 return self.current_directory
181 def canDescent(self):
182 if self.getSelection() is None:
184 return self.getSelection()[1]
187 if self.getSelection() is None:
189 self.changeDir(self.getSelection()[0], select = self.current_directory)
191 def getFilename(self):
192 if self.getSelection() is None:
194 x = self.getSelection()[0]
195 if isinstance(x, eServiceReference):
199 def getServiceRef(self):
200 if self.getSelection() is None:
202 x = self.getSelection()[0]
203 if isinstance(x, eServiceReference):
208 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
211 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
214 self.changeDir(self.current_directory, self.getFilename())
216 def partitionListChanged(self, action, device):
217 if self.current_directory is None: