qrcgen.py

#!/usr/bin/env python
"""
Autogenerating a qrc file from the full contents
of a directory tree
copyright 2011 by Flavio Codeco Coelho
licese: GPL v3
"""
import os
import argparse
import sys
import fnmatch
import re

def scan(direc,excludes):
    """
    Scan tree starting from direc
    """
    excludes = excludes or []
    excludes = r'|'.join([fnmatch.translate(x) for x in excludes]) or r'$.'

    resources = []
    for path, dirs, files in os.walk(direc):

        # exclude files
        files = [os.path.join(path, f) for f in files]
        files = [f for f in files if not re.match(excludes, f)]

        resources += files

    write_to_qrc(resources)

def write_to_qrc(resources):
    """
    Write to the qrc file under the prefix specified
    """
    with open('%s.qrc'%resname.strip('/'),'w') as f:
        f.write('<RCC>\n  <qresource prefix="%s">\n'%prefix)
        for r in resources:
            f.write('    <file>%s</file>\n'%r)
        f.write('  </qresource>\n</RCC>\n')

def valid_path(string):
    """
    check if the path entered is a valid one
    """
    if not os.path.isdir(string):
        msg = "%s is not a valid directory"%string
        raise argparse.ArgumentTypeError(msg)
    return string

if __name__=="__main__":
    parser = argparse.ArgumentParser(description='Generates a qrc (Qt resource file) from all files on a directory tree.',
        epilog='A directory.qrc file will be generated in the current directory')
    parser.add_argument('directory',metavar='directory',
        type=valid_path,
        help='A valid path, full or local.')
    parser.add_argument('prefix',metavar='prefix',
        type=str,
        help='The prefix in the qrc file under which the resources will be available.')
    parser.add_argument('-e','--exclude', action='append',metavar='exclude',
        type=str,
        help='Pattern(s) to exclude' )

    #~ parser.print_help()
    args = parser.parse_args()
    prefix = args.prefix
    resname = os.path.split(args.directory)[-1]
    scan(args.directory,args.exclude)