pysize/core/browsing.py

   1 # This program is free software; you can redistribute it and/or modify
   2 # it under the terms of the GNU General Public License as published by
   3 # the Free Software Foundation; either version 2 of the License, or
   4 # (at your option) any later version.
   5 #
   6 # This program is distributed in the hope that it will be useful,
   7 # but WITHOUT ANY WARRANTY; without even the implied warranty of
   8 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   9 # GNU Library General Public License for more details.
  10 #
  11 # You should have received a copy of the GNU General Public License
  12 # along with this program; if not, write to the Free Software
  13 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  14 #
  15 # See the COPYING file for license information.
  16 #
  17 # Copyright (c) 2006, 2007, 2008 Guillaume Chazarain <guichaz@gmail.com>
  18
  19 import os
  20 import stat
  21
  22 from pysize.core.deletion import filter_deleted
  23
  24 def browsedir(fullpath, cross_device=True, account_deletion=True):
  25     try:
  26         basenames = os.listdir(fullpath)
  27     except OSError, e:
  28         print e
  29         return []
  30     if account_deletion:
  31         basenames = filter_deleted(fullpath, basenames)
  32     if not cross_device:
  33         def same_dev(basename):
  34             test_fullpath = fullpath + '/' + basename
  35             try:
  36                 st = os.lstat(test_fullpath)
  37             except OSError:
  38                 return False
  39             if st.st_dev != dev:
  40                 return False
  41             if stat.S_ISDIR(st.st_mode):
  42                 # Some automount-like directories (~/.snapshot) reveal their
  43                 # true st_dev after being opened.
  44                 try:
  45                     fd = os.open(test_fullpath, os.O_RDONLY)
  46                 except OSError:
  47                     return False
  48                 try:
  49                     st = os.fstat(fd)
  50                 except OSError:
  51                     st = None
  52                 try:
  53                     os.close(fd)
  54                 except OSError:
  55                     return False
  56                 return st and st.st_dev == dev
  57
  58         try:
  59             dev = os.lstat(fullpath).st_dev
  60         except OSError, e:
  61             print e
  62             dev = None
  63         basenames = filter(same_dev, basenames)
  64     # We want the order to be always the same so that hard links
  65     # detection will be consistent across filesystems
  66     basenames.sort()
  67     fullpath = fullpath.rstrip('/')
  68     return [fullpath + '/' + basename for basename in basenames]