# -*- coding: utf-8 -*-

# Filename      main.py
# First edited  05-05-2012 21:04
# Last edited   05-05-2012 21:23

import os, sys
import convenience
from PIL import Image
import argparse

def run(s):
    cur_dir = os.getcwd()
    os.chdir(s)
    try:
        os.mkdir("OutputTesting")
    except WindowsError:
        pass

    files = os.listdir(os.getcwd())
    files = [x for x in files if os.path.isfile(x)]
    files = [x for x in files if os.path.splitext(x)[1].lower() in [".jpg", ".png"]]

    for i in files:
        convenience.print_r("%05d out of %05d processed." % (files.index(i)+1, len(files)))
        try:
            fn, ext = os.path.splitext(i)
            im = Image.open(i)
            im.thumbnail((128,128), Image.ANTIALIAS)
            im.save("OutputTesting\\" + fn + "-thumbnail" + ".jpg", "JPEG")
        except IOError:
            pass
    print ""

    os.chdir(cur_dir)

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('directory', metavar='D', type=str, nargs='+', default=None,
            help='Set urls to get')
    args = parser.parse_args()

    for i in args.directory:
        if os.path.isdir(i):
            run(i)
        else:
            print "Not a valid directory."

if __name__ == '__main__':
    main()