Geometrical-oriented image search (Python recipe) by Keisuke URAGO

ActiveState Code (http://code.activestate.com/recipes/181424/)

#!/usr/bin/env python
# -*- python -*-

"""
Image finder. Usage is run this script with no arguments. :)
"""

import os
import sys
import re
from getopt import getopt
from PIL import Image

class ImageScanner:
    def __init__(self):
        self.maxsize = (1000, 1000)
        self.minsize = (0, 0)
        self.printsize = 1
        self.verbose = 0

    def __parsesize(self, geom):
        matched = re.match('(?P<W>\d+)x(?P<H>\d+)', geom)
        if not matched:
            raise ValueError, 'Geometory format is not "XxY"'
        return (int(matched.group('W')), int(matched.group('H')))

    def setmaxsize(self, geom):
        self.maxsize = self.__parsesize(geom)

    def setminsize(self, geom):
        self.minsize = self.__parsesize(geom)

    def setnoprint(self):
        self.printsize = 0

    def setverbose(self):
        self.verbose = 1

    def scan(self, fn):
        try:
            size = Image.open(fn).size
            if self.maxsize >= size and self.minsize <= size:
                if self.printsize or self.verbose:
                    return fn + ' %dx%d' % size
                return fn
        except: pass
        if self.verbose:
            return fn + ' nil'
        return None

    def __recursivescan(self, arg, path, dirlist):
        for d in dirlist:
            self.__tmpbuf.append(self.scan(os.path.join(path, d)))
    def recursivescan(self, args):
        self.__tmpbuf = []
        for arg in args: os.path.walk(arg, self.__recursivescan, None)
        return self.__tmpbuf

def __list(arg, path, dirlist):
    return

def __usage():
    print >>sys.stderr, 'usage: %s [options] imagefile ...' % sys.argv[0]
    print >>sys.stderr, '''options:
    -h     print this usage
    -M XxY filter maximum image size
    -m XxY filter minimum image size
    -n     do not print size
    -r     recursive
    -v     verbose'''

def __main():
    optdic = {}
    arglist = []
    optlist = []
    tmp = sys.argv[1:]
    while 1:
        opts, tmp = getopt(tmp, 'hm:M:nrv')
        if opts: optlist += opts
        if not tmp: break
        arglist.append(tmp[0])
        tmp = tmp[1:]

    # Examine options and filenames.
    if optlist:
        for key, val in optlist:
            if key == '-h':
                __usage()
                return
            if optdic.has_key(key):
                raise ValueError, '%s option duplicated' % key
            optdic[key] = val
    if not arglist:
        raise ValueError, 'requires filename'

    scanner = ImageScanner()
    if optdic.has_key('-M'): scanner.setmaxsize(optdic['-M'])
    if optdic.has_key('-m'): scanner.setminsize(optdic['-m'])
    if optdic.has_key('-n'): scanner.setnoprint()
    if optdic.has_key('-v'): scanner.setverbose()

    if optdic.has_key('-r'):
        for data in filter(lambda a:a, scanner.recursivescan(arglist)):
            print data
    else:
        for data in filter(lambda a:a, map(scanner.scan, arglist)): print data

if __name__ == '__main__':
    try:
        __main()
    except Exception, e:
        print >>sys.stderr, e
        __usage()
        sys.exit(1)