lib/python/Components/FileList.py

   1 import re
   2
   3 from os import path as os_path, listdir
   4 from MenuList import MenuList
   5 from Components.Harddisk import harddiskmanager
   6
   7 from Tools.Directories import SCOPE_SKIN_IMAGE, resolveFilename
   8
   9 from enigma import RT_HALIGN_LEFT, eListbox, eListboxPythonMultiContent, \
  10         eServiceReference, eServiceCenter, gFont
  11 from Tools.LoadPixmap import LoadPixmap
  12
  13 EXTENSIONS = {
  14                 "mp3": "music",
  15                 "wav": "music",
  16                 "ogg": "music",
  17                 "jpg": "picture",
  18                 "jpeg": "picture",
  19                 "png": "picture",
  20                 "bmp": "picture",
  21                 "ts": "movie",
  22                 "avi": "movie",
  23                 "mpg": "movie",
  24                 "mpeg": "movie",
  25         }
  26
  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))
  30         if isDir:
  31                 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
  32         else:
  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"))
  37                 else:
  38                         png = None
  39         if png is not None:
  40                 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
  41
  42         return res
  43
  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
  53                 self.isTop = isTop
  54                 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
  55                 self.matchingPattern = matchingPattern
  56                 self.changeDir(directory)
  57
  58                 self.l.setFont(0, gFont("Regular", 18))
  59                 self.l.setItemHeight(23)
  60                 self.serviceHandler = eServiceCenter.getInstance()
  61
  62         def getSelection(self):
  63                 if self.l.getCurrentSelection() is None:
  64                         return None
  65                 return self.l.getCurrentSelection()[0]
  66
  67         def getCurrentEvent(self):
  68                 l = self.l.getCurrentSelection()
  69                 if not l or l[0][1] == True:
  70                         return None
  71                 else:
  72                         return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
  73
  74         def getFileList(self):
  75                 return self.list
  76
  77         def changeDir(self, directory, select = None):
  78                 self.list = []
  79
  80                 # if we are just entering from the list of mount points:
  81                 if self.current_directory is None:
  82                         if directory is None:
  83                                 self.mount_point = None
  84                         else:
  85                                 # Sort Mountpoints by length (longest first)
  86                                 sortedp = harddiskmanager.getMountedPartitions()
  87                                 sortedp.sort(key=lambda p: 0 - len(p.mountpoint))
  88
  89                                 # Search for the longest matching mp (should at least match /)
  90                                 for p in sortedp:
  91                                         if directory.startswith(p.mountpoint):
  92                                                 self.mount_point = p.mountpoint
  93                                                 if p.mountpoint != "/":
  94                                                         self.mount_point += "/"
  95                                                 break
  96                 self.current_directory = directory
  97                 directories = []
  98                 files = []
  99
 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))
 105                                 else:
 106                                         self.list.append(FileEntryComponent(name = p.description, absolute = p.mountpoint + "/", isDir = True))
 107                         files = [ ]
 108                         directories = [ ]
 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)
 115
 116                         while 1:
 117                                 s = list.getNext()
 118                                 if not s.valid():
 119                                         del list
 120                                         break
 121                                 if s.flags & s.mustDescent:
 122                                         directories.append(s.getPath())
 123                                 else:
 124                                         files.append(s)
 125                                 print s.getName(), s.flags
 126                         directories.sort()
 127                         files.sort()
 128                 else:
 129                         if os_path.exists(directory):
 130                                 files = listdir(directory)
 131                                 files.sort()
 132                                 tmpfiles = files[:]
 133                                 for x in tmpfiles:
 134                                         if os_path.isdir(directory + x):
 135                                                 directories.append(directory + x + "/")
 136                                                 files.remove(x)
 137
 138                 if directory is not None and self.showDirectories and not self.isTop:
 139                         if directory == self.mount_point:
 140                                 self.list.append(FileEntryComponent(name = ".. (" +_("List of Storage Devices") + ")", absolute = None, isDir = True))
 141                         else:
 142                                 self.list.append(FileEntryComponent(name = "..", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
 143
 144                 if self.showDirectories:
 145                         for x in directories:
 146                                 name = x.split('/')[-2]
 147                                 self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
 148
 149                 if self.showFiles:
 150                         for x in files:
 151                                 if self.useServiceRef:
 152                                         path = x.getPath()
 153                                         name = path.split('/')[-1]
 154                                 else:
 155                                         path = directory + x
 156                                         name = x
 157
 158                                 if self.matchingPattern is not None:
 159                                         if re.compile(self.matchingPattern).search(path):
 160                                                 self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
 161                                 else:
 162                                         self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
 163
 164                 self.l.setList(self.list)
 165
 166                 if select is not None:
 167                         i = 0
 168                         self.moveToIndex(0)
 169                         for x in self.list:
 170                                 p = x[0][0]
 171
 172                                 if isinstance(p, eServiceReference):
 173                                         p = p.getPath()
 174
 175                                 if p == select:
 176                                         self.moveToIndex(i)
 177                                 i += 1
 178
 179         def getCurrentDirectory(self):
 180                 return self.current_directory
 181
 182         def canDescent(self):
 183                 if self.getSelection() is None:
 184                         return False
 185                 return self.getSelection()[1]
 186
 187         def descent(self):
 188                 if self.getSelection() is None:
 189                         return
 190                 self.changeDir(self.getSelection()[0], select = self.current_directory)
 191
 192         def getFilename(self):
 193                 if self.getSelection() is None:
 194                         return None
 195                 x = self.getSelection()[0]
 196                 if isinstance(x, eServiceReference):
 197                         x = x.getPath()
 198                 return x
 199
 200         def getServiceRef(self):
 201                 if self.getSelection() is None:
 202                         return None
 203                 x = self.getSelection()[0]
 204                 if isinstance(x, eServiceReference):
 205                         return x
 206                 return None
 207
 208         def execBegin(self):
 209                 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
 210
 211         def execEnd(self):
 212                 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
 213
 214         def refresh(self):
 215                 self.changeDir(self.current_directory, self.getFilename())
 216
 217         def partitionListChanged(self, action, device):
 218                 if self.current_directory is None:
 219                         self.refresh()