from os import listdir
from os.path import abspath, isdir, join, isfile, splitext
from PIL import Image

def main():
    paths('.')

def compress_img(img_path):
    new_name, ext = splitext(img_path)
    if ext in ['.jpg', '.jpeg', '.png']:
        extent = ''
        img = Image.open(img_path)
        if ext[1:] in ['.jpg', '.jpeg']:
            extent = "JPEG"
        if ext[1:] == '.png':
            extent = "PNG"
        img.save(new_name+ext, extent, optimize=True,quality=90)


def paths(path):
    for d in listdir(path):
        p = join(path, d)
        if isdir(p) and d != 'env':
            paths(p)
        # if d == 'env':
        #     continue
        _, ext = splitext(p)
        if ext in ['.jpg', '.jpeg', '.png']:
            # print abspath(p)
            try:
                compress_img(abspath(p))
            except Exception as e:
                print e