import os
from glob import glob

from pymongo import MongoClient

import settings
from utils.image import resize


def generate_image_filename(image, size_name, ext):
    """
    Example: 2015-history-competition-1-small.jpg
    :param image:
    :param size_name:
    :param ext:
    :return:
    """
    return '{image}-{size}{ext}'.format(
            image=image,
            size=size_name,
            ext=ext
    )


def process_file(file):
    sizes = settings.image_sizes
    image = resize.read_image(file)
    images_dir = '/Users/tmshv/Desktop/Hudozka Awards'
    public_base_url = 'https://static.shlisselburg.org/art/awards/'

    basename = os.path.basename(file)
    uri, ext = os.path.splitext(basename)

    award = {
        'uri': uri,
        'thumbs': {}
    }

    for size in sizes:
        size_name, width, height = size

        image_filename = generate_image_filename(uri, size_name, ext)
        d = os.path.join(images_dir, image_filename)
        image_url = os.path.join(public_base_url, image_filename)

        thumb = resize.optimize(image, d, quality=90) if size_name == 'original' else resize.thumbnail(image, d, (width, height))

        width, height = thumb.size
        award['thumbs'][size_name] = {
            'url': image_url,
            'size': size_name,
            'width': width,
            'height': height
        }

    return award


def get_db():
    client = MongoClient(settings.database_uri)
    return client.hudozka


def sync(awards):
    clear_db()
    db = get_db()
    for a in awards:
        db.awards.insert(a)


def clear_db():
    db = get_db()
    db.awards.drop()


if __name__ == '__main__':
    dir = '/Users/tmshv/Dropbox/Dev/Hud school/Awards'
    files = glob(dir + '/*')

    awards = list(map(
            process_file,
            files
            # map(
            #     lambda f : os.path.join(dir, f),
            #     files
            # )
    ))

    print(awards)
    sync(awards)