root/trunk/RBFoundation/RBFoundation/_InstallerHookUtilities.py

#!/usr/bin/env python
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
##~ License 
##~ 
##- The RuneBlade Foundation library is intended to ease some 
##- aspects of writing intricate Jabber, XML, and User Interface (wxPython, etc.) 
##- applications, while providing the flexibility to modularly change the 
##- architecture. Enjoy.
##~ 
##~ Copyright (C) 2002  TechGame Networks, LLC.
##~ 
##~ This library is free software; you can redistribute it and/or
##~ modify it under the terms of the BSD style License as found in the 
##~ LICENSE file included with this distribution.
##~ 
##~ TechGame Networks, LLC can be reached at:
##~ 3578 E. Hartsel Drive #211
##~ Colorado Springs, Colorado, USA, 80920
##~
##~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

"""
Requires path.py from:
    Jason Orendorff <jason@jorendorff.com> (and others - see the url!)
    http://www.jorendorff.com/articles/python/path
"""

#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#~ Imports 
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

import os
try:
    from path import path
except ImportError, e:
    print """Requires path.py from:
        Jason Orendorff <jason@jorendorff.com> (and others - see the url!)
        http://www.jorendorff.com/articles/python/path
    """
    raise

#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
#~ Definitions 
#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

def PackageFilter(filename):
    return filename not in ('CVS',)

def ModuleFilter(filename):
    return (filename[0]!='_' or filename[1]=='_')

def MakePackageHook(prefix, file, recurse=True, includefiletypes=('*.py','*.pyw','*.pyc','*.pyo','*.pyd'), packagefilter=PackageFilter, modulefilter=ModuleFilter):
    def GetPyModules(packagedir):
        modules = {}
        for filetype in includefiletypes:
            for filename in filter(modulefilter, packagedir.files(filetype)):
                modules[filename.splitext()[0].name] = True

        if recurse:
            for dirname in filter(packagefilter, packagedir.dirs()):
                for result in GetPyModules(dirname):
                    modules['%s.%s'%(dirname.name, result)] = True
        return modules.keys()

    packagedir = path(file)
    if packagedir.isfile():
        packagedir = packagedir.splitpath()[0]
    modules = {}
    for result in GetPyModules(packagedir):
        modules['%s.%s'%(prefix, result)] = True
    return map(str, modules.keys())

def MakePackageHookVerbose(*args, **kw):
    print "Hidden imports:"
    hiddenimports = MakePackageHook(*args, **kw)
    for each in hiddenimports:
        print "    \"%s\"" % each
    return hiddenimports