image.py

import os.path as path
import string
import argparse
import glob
import re

def basename(filename):
    base = filename
    if filename.find('@2x') > 0:
        base = filename[:filename.find('@2x')]
    elif filename.find('~') > 0:
        base = filename[:filename.find('~')]
    elif filename.find('.') > 0:
        base = filename[:filename.find('.')]
    return base

def file_type(filename):
    isRetina = False
    isIpad = False
    if filename.find('@2x') > 0:
        isRetina = True
    if filename.find('~ipad') > 0:
        isIpad = True
    
    if isRetina:
        if isIpad:
            return 'ipad2x'
        else:
            return 'iphone2x'
    else:
        if isIpad:
            return 'ipad'
    return 'iphone'
    
def mungedName(basename):
    parts = re.split('[-_]', basename)
    capitalized = [word.capitalize() for word in parts]
    return string.join(capitalized, '')
    
class ImageGroup:
    iphone = None
    iphone2x = None
    ipad = None
    ipad2x = None
    extras = []
    
    def __init__(self, file_name):
        setattr(self, file_type(file_name), file_name)
                
    def add_file(self, file_name):
        type = file_type(file_name)
        
        if getattr(self, type) is not None:
            self.extras.append(file_name)
        else:
            setattr(self, type, file_name)
    
    def warnings(self, iPhone=True, iPad=True, retina=True, duplicates=True):
        definition = ''
        if iPhone and self.iphone is None:
            definition += '#warning image formatted for iPhone %s not found\n' % filename
        if iPhone and retina and self.iphone2x is None:
            definition += '#warning image formatted for retina iPhone %s not found\n' % filename
        if iPad and self.ipad is None:
            definition += '#warning image formatted for iPad %s not found\n' % filename
        if iPad and retina and self.ipad2x is None:
            definition += '#warning image formatted for retina iPad %s not found\n' % filename
        if duplicates:
            for file in self.extras:
                definition += '#warning duplicate image %s found in project. Verify proper capitalization.\n' % file
        return definition
        
    def output(self, filename, prefix):
        return args.format % {'prefix':prefix, 'identifier':mungedName(filename), 'filename':filename}
        
        
DEFAULT_FORMAT = '#define %(prefix)s%(identifier)s %(filename)s\n'

### cmd folder outputFile
parser = argparse.ArgumentParser(description='Create a header file with contants for each image file in the given folder.')
parser.add_argument('-s', '--source', type=str, default='./', help='A folder which contains images.')
parser.add_argument('-d', '--destination', type=str, default='./images.h', help='The filename to write to.')
parser.add_argument('--prefix', type=str, default='img', help='The prefix added at the begining of each image\'s filename.')
parser.add_argument('--format', type=str, default=DEFAULT_FORMAT, help='The format string specifying how the file should be written')
parser.add_argument('--warn-retina', dest='retina', type=bool, default=True, help='Warn for missing retina images.')
parser.add_argument('--warn-ipad', dest='ipad', type=bool, default=False, help='Warn for missing iPad (~ipad) images.')
parser.add_argument('--warn-iphone', dest='iphone', type=bool, default=False, help='Warn for missing iPhone (~iphone) images')
parser.add_argument('--warn-duplicates', dest='duplicates', type=bool, default=True, help='Warn for duplicate images.')

args = parser.parse_args()
source = path.join(path.expanduser(args.source), '*.png')
iterator = glob.iglob(source)
all_files = {}

for fullpath in iterator:
    (leading, filename) = path.split(fullpath)
    key = basename(filename)

    if key in all_files:
        current_file = all_files[key]
        current_file.add_file(filename)
    else:
        current_file = ImageGroup(filename)
        all_files[key] = current_file

original_output = ''   
file_path = path.join(path.expanduser('.'), args.destination)
if path.exists(file_path):
    f = open(file_path, 'r')
    original_output += f.read()
    f.close()

output = '// Created using the image.py script written by Patrick Hughes. He\'s a pretty cool guy.\n'
output +='//\n// DO NOT EDIT THIS FILE. \n//\n'
output +='// This file is automatically generated. Any changes will be overwritten the next time images.py is invoked.\n\n'

for key in all_files:
    image_group = all_files[key]
    warnings = image_group.warnings(iPhone=args.iphone, iPad=args.ipad, retina=args.retina, duplicates=args.duplicates)
    output += warnings
    output += image_group.output(key, args.prefix)


output += '\n\n'

if original_output == output:
    print "no changes"
else:
    file = open(file_path, 'w')
    print "writing changes"
    file.write(output)
    file.close()