setup.command

#!/usr/bin/env python

import StringIO
import struct
import os
import sys
import json

PATH = os.path.dirname(sys.argv[0]) + '/'
RELATIVE_PATH = 'photos'
PHOTO_PATH = PATH + RELATIVE_PATH


def get_directories():
    items = os.listdir(PHOTO_PATH)
    return filter(lambda x: os.path.isdir(PHOTO_PATH + '/' + x), items)


def is_image_path(path):
    return path.endswith('jpg') or \
           path.endswith('jpeg') or \
           path.endswith('png')


def get_images(path):
    items = os.listdir(PHOTO_PATH + '/' + path)
    filtered_items = filter(is_image_path, items)

    result = []
    for img in filtered_items:
        width, height = 0, 0
        with open(PHOTO_PATH + '/' + path + '/' + img) as f:
            _, width, height = getImageInfo(f.read())
        result.append({
            'width': width,
            'height': height,
            'path': './' + RELATIVE_PATH + '/' + path + '/' + img
        })
    return result


def write_config(config):
    with open(PATH + 'config.json', 'w') as f:
        f.write(json.dumps(config, indent=2, separators=(',', ': ')))


def run():
    print 'Starting to collect all albums within the /photos directory...'
    config = {}
    dirs = get_directories()
    print 'Found {length} directories'.format(length=len(dirs))
    for i, path in enumerate(dirs):
        print str(i+1) + ': Processing photos for the album "{album}"'.format(
            album=path)
        config[path] = get_images(path)

        print '   Done processing {l} photos for "{album}"\n'.format(
            l=len(config[path]),
            album=path)

    print 'Done processing all {length} albums'.format(length=len(dirs))
    print 'Writing files to {path} now...'.format(path=PATH + 'config.json')
    write_config(config)
    print '''Done writing! You may now safely close this window :)

Thank you for using gallery! Share your gallery on Github!
https://github.com/andyzg/gallery/issues/1'''
    return 0


############################################################
# Thanks StackOverflow: http://stackoverflow.com/a/3175473 #
############################################################
def getImageInfo(data):
    data = str(data)
    size = len(data)
    height = -1
    width = -1
    content_type = ''

    # handle GIFs
    if (size >= 10) and data[:6] in ('GIF87a', 'GIF89a'):
        # Check to see if content_type is correct
        content_type = 'image/gif'
        w, h = struct.unpack("<HH", data[6:10])
        width = int(w)
        height = int(h)

    # See PNG 2. Edition spec (http://www.w3.org/TR/PNG/)
    # Bytes 0-7 are below, 4-byte chunk length, then 'IHDR'
    # and finally the 4-byte width, height
    elif ((size >= 24) and data.startswith('\211PNG\r\n\032\n') and
          (data[12:16] == 'IHDR')):
        content_type = 'image/png'
        w, h = struct.unpack(">LL", data[16:24])
        width = int(w)
        height = int(h)

    # Maybe this is for an older PNG version.
    elif (size >= 16) and data.startswith('\211PNG\r\n\032\n'):
        # Check to see if we have the right content type
        content_type = 'image/png'
        w, h = struct.unpack(">LL", data[8:16])
        width = int(w)
        height = int(h)

    # handle JPEGs
    elif (size >= 2) and data.startswith('\377\330'):
        content_type = 'image/jpeg'
        jpeg = StringIO.StringIO(data)
        jpeg.read(2)
        b = jpeg.read(1)
        try:
            while (b and ord(b) != 0xDA):
                while (ord(b) != 0xFF):
                    b = jpeg.read(1)
                while (ord(b) == 0xFF):
                    b = jpeg.read(1)
                if (ord(b) >= 0xC0 and ord(b) <= 0xC3):
                    jpeg.read(3)
                    h, w = struct.unpack(">HH", jpeg.read(4))
                    break
                else:
                    jpeg.read(int(struct.unpack(">H", jpeg.read(2))[0])-2)
                b = jpeg.read(1)
            width = int(w)
            height = int(h)
        except struct.error:
            pass
        except ValueError:
            pass

    return content_type, width, height

if __name__ == '__main__':
    sys.exit(run())