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