lib/python/Components/FileList.py

   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
   5
   6 from Tools.Directories import SCOPE_SKIN_IMAGE, resolveFilename, fileExists
   7
   8 from enigma import RT_HALIGN_LEFT, eListboxPythonMultiContent, \
   9         eServiceReference, eServiceCenter, gFont
  10 from Tools.LoadPixmap import LoadPixmap
  11
  12 EXTENSIONS = {
  13                 "m4a": "music",
  14                 "mp2": "music",
  15                 "mp3": "music",
  16                 "wav": "music",
  17                 "ogg": "music",
  18                 "flac": "music",
  19                 "jpg": "picture",
  20                 "jpeg": "picture",
  21                 "png": "picture",
  22                 "bmp": "picture",
  23                 "ts": "movie",
  24                 "avi": "movie",
  25                 "divx": "movie",
  26                 "m4v": "movie",
  27                 "mpg": "movie",
  28                 "mpeg": "movie",
  29                 "mkv": "movie",
  30                 "mp4": "movie",
  31                 "mov": "movie"
  32         }
  33
  34 def FileEntryComponent(name, absolute = None, isDir = False):
  35         res = [ (absolute, isDir) ]
  36         res.append((eListboxPythonMultiContent.TYPE_TEXT, 35, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
  37         if isDir:
  38                 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
  39         else:
  40                 extension = name.split('.')
  41                 extension = extension[-1].lower()
  42                 if EXTENSIONS.has_key(extension):
  43                         png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
  44                 else:
  45                         png = None
  46         if png is not None:
  47                 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 10, 2, 20, 20, png))
  48
  49         return res
  50
  51 class FileList(MenuList):
  52         def __init__(self, directory, showDirectories = True, showFiles = True, showMountpoints = True, matchingPattern = None, useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
  53                 MenuList.__init__(self, list, enableWrapAround, eListboxPythonMultiContent)
  54                 self.additional_extensions = additionalExtensions
  55                 self.mountpoints = []
  56                 self.current_directory = None
  57                 self.current_mountpoint = None
  58                 self.useServiceRef = useServiceRef
  59                 self.showDirectories = showDirectories
  60                 self.showMountpoints = showMountpoints
  61                 self.showFiles = showFiles
  62                 self.isTop = isTop
  63                 # example: matching .nfi and .ts files: "^.*\.(nfi|ts)"
  64                 self.matchingPattern = matchingPattern
  65                 self.inhibitDirs = inhibitDirs or []
  66                 self.inhibitMounts = inhibitMounts or []
  67
  68                 self.refreshMountpoints()
  69                 self.changeDir(directory)
  70                 self.l.setFont(0, gFont("Regular", 18))
  71                 self.l.setItemHeight(23)
  72                 self.serviceHandler = eServiceCenter.getInstance()
  73
  74         def refreshMountpoints(self):
  75                 self.mountpoints = [os_path.join(p.mountpoint, "") for p in harddiskmanager.getMountedPartitions()]
  76                 self.mountpoints.sort(reverse = True)
  77
  78         def getMountpoint(self, file):
  79                 file = os_path.join(os_path.realpath(file), "")
  80                 for m in self.mountpoints:
  81                         if file.startswith(m):
  82                                 return m
  83                 return False
  84
  85         def getMountpointLink(self, file):
  86                 if os_path.realpath(file) == file:
  87                         return self.getMountpoint(file)
  88                 else:
  89                         if file[-1] == "/":
  90                                 file = file[:-1]
  91                         mp = self.getMountpoint(file)
  92                         last = file
  93                         file = os_path.dirname(file)
  94                         while last != "/" and mp == self.getMountpoint(file):
  95                                 last = file
  96                                 file = os_path.dirname(file)
  97                         return os_path.join(last, "")
  98
  99         def getSelection(self):
 100                 if self.l.getCurrentSelection() is None:
 101                         return None
 102                 return self.l.getCurrentSelection()[0]
 103
 104         def getCurrentEvent(self):
 105                 l = self.l.getCurrentSelection()
 106                 if not l or l[0][1] == True:
 107                         return None
 108                 else:
 109                         return self.serviceHandler.info(l[0][0]).getEvent(l[0][0])
 110
 111         def getFileList(self):
 112                 return self.list
 113
 114         def inParentDirs(self, dir, parents):
 115                 dir = os_path.realpath(dir)
 116                 for p in parents:
 117                         if dir.startswith(p):
 118                                 return True
 119                 return False
 120
 121         def changeDir(self, directory, select = None):
 122                 self.list = []
 123
 124                 # if we are just entering from the list of mount points:
 125                 if self.current_directory is None:
 126                         if directory and self.showMountpoints:
 127                                 self.current_mountpoint = self.getMountpointLink(directory)
 128                         else:
 129                                 self.current_mountpoint = None
 130                 self.current_directory = directory
 131                 directories = []
 132                 files = []
 133
 134                 if directory is None and self.showMountpoints: # present available mountpoints
 135                         for p in harddiskmanager.getMountedPartitions():
 136                                 path = os_path.join(p.mountpoint, "")
 137                                 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
 138                                         self.list.append(FileEntryComponent(name = p.description, absolute = path, isDir = True))
 139                         files = [ ]
 140                         directories = [ ]
 141                 elif directory is None:
 142                         files = [ ]
 143                         directories = [ ]
 144                 elif self.useServiceRef:
 145                         root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
 146                         if self.additional_extensions:
 147                                 root.setName(self.additional_extensions)
 148                         serviceHandler = eServiceCenter.getInstance()
 149                         list = serviceHandler.list(root)
 150
 151                         while 1:
 152                                 s = list.getNext()
 153                                 if not s.valid():
 154                                         del list
 155                                         break
 156                                 if s.flags & s.mustDescent:
 157                                         directories.append(s.getPath())
 158                                 else:
 159                                         files.append(s)
 160                         directories.sort()
 161                         files.sort()
 162                 else:
 163                         if fileExists(directory):
 164                                 try:
 165                                         files = listdir(directory)
 166                                 except:
 167                                         files = []
 168                                 files.sort()
 169                                 tmpfiles = files[:]
 170                                 for x in tmpfiles:
 171                                         if os_path.isdir(directory + x):
 172                                                 directories.append(directory + x + "/")
 173                                                 files.remove(x)
 174
 175                 if directory is not None and self.showDirectories and not self.isTop:
 176                         if directory == self.current_mountpoint and self.showMountpoints:
 177                                 self.list.append(FileEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
 178                         elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
 179                                 self.list.append(FileEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
 180
 181                 if self.showDirectories:
 182                         for x in directories:
 183                                 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
 184                                         name = x.split('/')[-2]
 185                                         self.list.append(FileEntryComponent(name = name, absolute = x, isDir = True))
 186
 187                 if self.showFiles:
 188                         for x in files:
 189                                 if self.useServiceRef:
 190                                         path = x.getPath()
 191                                         name = path.split('/')[-1]
 192                                 else:
 193                                         path = directory + x
 194                                         name = x
 195
 196                                 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
 197                                         self.list.append(FileEntryComponent(name = name, absolute = x , isDir = False))
 198
 199                 self.l.setList(self.list)
 200
 201                 if select is not None:
 202                         i = 0
 203                         self.moveToIndex(0)
 204                         for x in self.list:
 205                                 p = x[0][0]
 206
 207                                 if isinstance(p, eServiceReference):
 208                                         p = p.getPath()
 209
 210                                 if p == select:
 211                                         self.moveToIndex(i)
 212                                 i += 1
 213
 214         def getCurrentDirectory(self):
 215                 return self.current_directory
 216
 217         def canDescent(self):
 218                 if self.getSelection() is None:
 219                         return False
 220                 return self.getSelection()[1]
 221
 222         def descent(self):
 223                 if self.getSelection() is None:
 224                         return
 225                 self.changeDir(self.getSelection()[0], select = self.current_directory)
 226
 227         def getFilename(self):
 228                 if self.getSelection() is None:
 229                         return None
 230                 x = self.getSelection()[0]
 231                 if isinstance(x, eServiceReference):
 232                         x = x.getPath()
 233                 return x
 234
 235         def getServiceRef(self):
 236                 if self.getSelection() is None:
 237                         return None
 238                 x = self.getSelection()[0]
 239                 if isinstance(x, eServiceReference):
 240                         return x
 241                 return None
 242
 243         def execBegin(self):
 244                 harddiskmanager.on_partition_list_change.append(self.partitionListChanged)
 245
 246         def execEnd(self):
 247                 harddiskmanager.on_partition_list_change.remove(self.partitionListChanged)
 248
 249         def refresh(self):
 250                 self.changeDir(self.current_directory, self.getFilename())
 251
 252         def partitionListChanged(self, action, device):
 253                 self.refreshMountpoints()
 254                 if self.current_directory is None:
 255                         self.refresh()
 256
 257
 258 def MultiFileSelectEntryComponent(name, absolute = None, isDir = False, selected = False):
 259         res = [ (absolute, isDir, selected, name) ]
 260         res.append((eListboxPythonMultiContent.TYPE_TEXT, 55, 1, 470, 20, 0, RT_HALIGN_LEFT, name))
 261         if isDir:
 262                 png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/directory.png"))
 263         else:
 264                 extension = name.split('.')
 265                 extension = extension[-1].lower()
 266                 if EXTENSIONS.has_key(extension):
 267                         png = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "extensions/" + EXTENSIONS[extension] + ".png"))
 268                 else:
 269                         png = None
 270         if png is not None:
 271                 res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 30, 2, 20, 20, png))
 272
 273         if not name.startswith('<'):
 274                 if selected is False:
 275                         icon = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "skin_default/icons/lock_off.png"))
 276                         res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
 277                 else:
 278                         icon = LoadPixmap(resolveFilename(SCOPE_SKIN_IMAGE, "skin_default/icons/lock_on.png"))
 279                         res.append((eListboxPythonMultiContent.TYPE_PIXMAP_ALPHATEST, 2, 0, 25, 25, icon))
 280
 281         return res
 282
 283
 284 class MultiFileSelectList(FileList):
 285         def __init__(self, preselectedFiles, directory, showMountpoints = False, matchingPattern = None, showDirectories = True, showFiles = True,  useServiceRef = False, inhibitDirs = False, inhibitMounts = False, isTop = False, enableWrapAround = False, additionalExtensions = None):
 286                 self.selectedFiles = preselectedFiles
 287                 if self.selectedFiles is None:
 288                         self.selectedFiles = []
 289                 FileList.__init__(self, directory, showMountpoints = showMountpoints, matchingPattern = matchingPattern, showDirectories = showDirectories, showFiles = showFiles,  useServiceRef = useServiceRef, inhibitDirs = inhibitDirs, inhibitMounts = inhibitMounts, isTop = isTop, enableWrapAround = enableWrapAround, additionalExtensions = additionalExtensions)
 290                 self.changeDir(directory)
 291                 self.l.setItemHeight(25)
 292                 self.l.setFont(0, gFont("Regular", 20))
 293                 self.onSelectionChanged = [ ]
 294
 295         def selectionChanged(self):
 296                 for f in self.onSelectionChanged:
 297                         f()
 298
 299         def changeSelectionState(self):
 300                 idx = self.l.getCurrentSelectionIndex()
 301                 count = 0
 302                 newList = []
 303                 for x in self.list:
 304                         if idx == count:
 305                                 if x[0][3].startswith('<'):
 306                                         newList.append(x)
 307                                 else:
 308                                         if x[0][1] is True:
 309                                                 realPathname = x[0][0]
 310                                         else:
 311                                                 realPathname = self.current_directory + x[0][0]
 312                                         if x[0][2] == True:
 313                                                 SelectState = False
 314                                                 for entry in self.selectedFiles:
 315                                                         if entry == realPathname:
 316                                                                 self.selectedFiles.remove(entry)
 317
 318                                         else:
 319                                                 SelectState = True
 320                                                 alreadyinList = False
 321                                                 for entry in self.selectedFiles:
 322                                                         if entry == realPathname:
 323                                                                 alreadyinList = True
 324                                                 if not alreadyinList:
 325                                                         self.selectedFiles.append(realPathname)
 326                                         newList.append(MultiFileSelectEntryComponent(name = x[0][3], absolute = x[0][0], isDir = x[0][1], selected = SelectState ))
 327                         else:
 328                                 newList.append(x)
 329
 330                         count += 1
 331
 332                 self.list = newList
 333                 self.l.setList(self.list)
 334
 335         def getSelectedList(self):
 336                 return self.selectedFiles
 337
 338         def changeDir(self, directory, select = None):
 339                 self.list = []
 340
 341                 # if we are just entering from the list of mount points:
 342                 if self.current_directory is None:
 343                         if directory and self.showMountpoints:
 344                                 self.current_mountpoint = self.getMountpointLink(directory)
 345                         else:
 346                                 self.current_mountpoint = None
 347                 self.current_directory = directory
 348                 directories = []
 349                 files = []
 350
 351                 if directory is None and self.showMountpoints: # present available mountpoints
 352                         for p in harddiskmanager.getMountedPartitions():
 353                                 path = os_path.join(p.mountpoint, "")
 354                                 if path not in self.inhibitMounts and not self.inParentDirs(path, self.inhibitDirs):
 355                                         self.list.append(MultiFileSelectEntryComponent(name = p.description, absolute = path, isDir = True))
 356                         files = [ ]
 357                         directories = [ ]
 358                 elif directory is None:
 359                         files = [ ]
 360                         directories = [ ]
 361                 elif self.useServiceRef:
 362                         root = eServiceReference("2:0:1:0:0:0:0:0:0:0:" + directory)
 363                         if self.additional_extensions:
 364                                 root.setName(self.additional_extensions)
 365                         serviceHandler = eServiceCenter.getInstance()
 366                         list = serviceHandler.list(root)
 367
 368                         while 1:
 369                                 s = list.getNext()
 370                                 if not s.valid():
 371                                         del list
 372                                         break
 373                                 if s.flags & s.mustDescent:
 374                                         directories.append(s.getPath())
 375                                 else:
 376                                         files.append(s)
 377                         directories.sort()
 378                         files.sort()
 379                 else:
 380                         if fileExists(directory):
 381                                 try:
 382                                         files = listdir(directory)
 383                                 except:
 384                                         files = []
 385                                 files.sort()
 386                                 tmpfiles = files[:]
 387                                 for x in tmpfiles:
 388                                         if os_path.isdir(directory + x):
 389                                                 directories.append(directory + x + "/")
 390                                                 files.remove(x)
 391
 392                 if directory is not None and self.showDirectories and not self.isTop:
 393                         if directory == self.current_mountpoint and self.showMountpoints:
 394                                 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("List of Storage Devices") + ">", absolute = None, isDir = True))
 395                         elif (directory != "/") and not (self.inhibitMounts and self.getMountpoint(directory) in self.inhibitMounts):
 396                                 self.list.append(MultiFileSelectEntryComponent(name = "<" +_("Parent Directory") + ">", absolute = '/'.join(directory.split('/')[:-2]) + '/', isDir = True))
 397
 398                 if self.showDirectories:
 399                         for x in directories:
 400                                 if not (self.inhibitMounts and self.getMountpoint(x) in self.inhibitMounts) and not self.inParentDirs(x, self.inhibitDirs):
 401                                         name = x.split('/')[-2]
 402                                         alreadySelected = False
 403                                         for entry in self.selectedFiles:
 404                                                 if entry  == x:
 405                                                         alreadySelected = True
 406                                         if alreadySelected:
 407                                                 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = True))
 408                                         else:
 409                                                 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x, isDir = True, selected = False))
 410
 411                 if self.showFiles:
 412                         for x in files:
 413                                 if self.useServiceRef:
 414                                         path = x.getPath()
 415                                         name = path.split('/')[-1]
 416                                 else:
 417                                         path = directory + x
 418                                         name = x
 419
 420                                 if (self.matchingPattern is None) or re_compile(self.matchingPattern).search(path):
 421                                         alreadySelected = False
 422                                         for entry in self.selectedFiles:
 423                                                 if os_path.basename(entry)  == x:
 424                                                         alreadySelected = True
 425                                         if alreadySelected:
 426                                                 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = True))
 427                                         else:
 428                                                 self.list.append(MultiFileSelectEntryComponent(name = name, absolute = x , isDir = False, selected = False))
 429
 430                 self.l.setList(self.list)
 431
 432                 if select is not None:
 433                         i = 0
 434                         self.moveToIndex(0)
 435                         for x in self.list:
 436                                 p = x[0][0]
 437
 438                                 if isinstance(p, eServiceReference):
 439                                         p = p.getPath()
 440
 441                                 if p == select:
 442                                         self.moveToIndex(i)
 443                                 i += 1
 444