trunk/basic_image_finder.py
author Sam Hart <criswellious@gmail.com>
Sun Apr 27 21:16:21 2008 -0400 (4 years ago)
branchpersonal-scripts
changeset 47 852be106a4bd
child 4819d9647d0639
permissions -rwxr-xr-x
Add basic image finder
     1 #! /usr/bin/env python
     2 
     3 pattern_list = ['*.jpg',
     4                 '*.jpeg',
     5                 '*.gif',
     6                 '*.png',
     7                 '*.tiff',
     8                ]
     9 
    10 class Bunch:
    11     def __init__(self, **kwds): self.__dict__.update(kwds)
    12 arg = Bunch(recurse=True, pattern_list=pattern_list, return_folders=0, results={})
    13 
    14 def visit(arg, dirname, files):
    15     # Append to arg.results all relevant files
    16     for name in files:
    17         fullname = os.path.normpath(os.path.join(dirname, name))
    18         fullname = fullname.rstrip()
    19         if arg.return_folders or os.path.isfile(fullname):
    20             normalized_filename = name.lower()
    21             for pattern in arg.pattern_list:
    22                 if fnmatch.fnmatch(normalized_filename, pattern):
    23                     arg.results[name] = fullname
    24                     break
    25     # Block recursion if disallowed
    26     if not arg.recurse: files[:]=[]
    27 
    28 root = "/home/sam/images"
    29 
    30 os.path.walk(root, visit, arg)
    31 print arg.results