tools/findleaves.py

   1 #!/usr/bin/env python
   2 #
   3 # Copyright (C) 2009 The Android Open Source Project
   4 #
   5 # Licensed under the Apache License, Version 2.0 (the "License");
   6 # you may not use this file except in compliance with the License.
   7 # You may obtain a copy of the License at
   8 #
   9 #      http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing, software
  12 # distributed under the License is distributed on an "AS IS" BASIS,
  13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 # See the License for the specific language governing permissions and
  15 # limitations under the License.
  16 #
  17
  18 #
  19 # Finds files with the specified name under a particular directory, stopping
  20 # the search in a given subdirectory when the file is found.
  21 #
  22
  23 import os
  24 import sys
  25
  26 def perform_find(mindepth, prune, dirlist, filename):
  27   result = []
  28   pruneleaves = set(map(lambda x: os.path.split(x)[1], prune))
  29   for rootdir in dirlist:
  30     rootdepth = rootdir.count("/")
  31     for root, dirs, files in os.walk(rootdir, followlinks=True):
  32       # prune
  33       check_prune = False
  34       for d in dirs:
  35         if d in pruneleaves:
  36           check_prune = True
  37           break
  38       if check_prune:
  39         i = 0
  40         while i < len(dirs):
  41           if dirs[i] in prune:
  42             del dirs[i]
  43           else:
  44             i += 1
  45       # mindepth
  46       if mindepth > 0:
  47         depth = 1 + root.count("/") - rootdepth
  48         if depth < mindepth:
  49           continue
  50       # match
  51       if filename in files:
  52         result.append(os.path.join(root, filename))
  53         del dirs[:]
  54   return result
  55
  56 def usage():
  57   sys.stderr.write("""Usage: %(progName)s [<options>] <dirlist> <filename>
  58 Options:
  59    --mindepth=<mindepth>
  60        Both behave in the same way as their find(1) equivalents.
  61    --prune=<dirname>
  62        Avoids returning results from inside any directory called <dirname>
  63        (e.g., "*/out/*"). May be used multiple times.
  64 """ % {
  65       "progName": os.path.split(sys.argv[0])[1],
  66     })
  67   sys.exit(1)
  68
  69 def main(argv):
  70   mindepth = -1
  71   prune = []
  72   i=1
  73   while i<len(argv) and len(argv[i])>2 and argv[i][0:2] == "--":
  74     arg = argv[i]
  75     if arg.startswith("--mindepth="):
  76       try:
  77         mindepth = int(arg[len("--mindepth="):])
  78       except ValueError:
  79         usage()
  80     elif arg.startswith("--prune="):
  81       p = arg[len("--prune="):]
  82       if len(p) == 0:
  83         usage()
  84       prune.append(p)
  85     else:
  86       usage()
  87     i += 1
  88   if len(argv)-i < 2: # need both <dirlist> and <filename>
  89     usage()
  90   dirlist = argv[i:-1]
  91   filename = argv[-1]
  92   results = list(set(perform_find(mindepth, prune, dirlist, filename)))
  93   results.sort()
  94   for r in results:
  95     print r
  96
  97 if __name__ == "__main__":
  98   main(sys.argv)