# -*- coding: UTF-8 -*-

import json

import logging

import os

import pprint

import re

import shutil

import imghdr

from django.conf import settings

from django.core.management.base import BaseCommand, CommandError

from PIL import Image as ImagePIL

from sorl.thumbnail import get_thumbnail

from iconolab.management.commands.importimages import BaseImportImagesCommand

from iconolab.models import (Collection, Folder, Image, ImageStats, Item,

                             ItemMetadata, MetaCategory)

if settings.IMPORT_LOGGER_NAME and settings.LOGGING['loggers'].get(settings.IMPORT_LOGGER_NAME, ''):

    logger = logging.getLogger(settings.IMPORT_LOGGER_NAME)

else:

    logger = logging.getLogger(__name__)

class Command(BaseImportImagesCommand):

    help = 'import images from a directory into the media folder and creates item and image objects'

    def add_arguments(self, parser):

        parser.add_argument('source_dir')

        parser.add_argument(

            '--collection-id',

            dest='collection_id',

            default=False,

            help='insert extracted data into the specified collection instead of trying to load a collection fixture',

        )

        parser.add_argument(

            '--no-jpg-conversion',

            dest='no-jpg-conversion',

            default=False,

            help='use this option if you only want the image copied and not converted'

        )

    def handle(self, *args, **options):

        #Set no image size limit to PIL to be able to process big images.

        ImagePIL.MAX_IMAGE_PIXELS = None

        print('# Logging with logger '+logger.name)

        logger.debug('# Initializing command with args: %r', options)

        self.source_dir = options.get('source_dir')

        collection_id = options.get('collection_id')

        if not collection_id:

            raise CommandError("No collection id, aborting")

        print('## Finding collection with id %s' % collection_id) 

        try:

            try:

                collection = Collection.objects.get(pk=int(collection_id))

            except ValueError:

                collection = Collection.objects.get(name=collection_id)

        except Collection.DoesNotExist:

            raise CommandError('!!! Collection with id ' + collection_id

                                +' was not found, aborting !!!')

        print(

            '## Converting image and moving it to static dir, creating Image and Item objects')

        print('### Images will be stored in ' + os.path.join(settings.MEDIA_ROOT,'uploads'))

        for dirname, dirs, files in os.walk(self.source_dir):

            for filename in files:

                print("::Examining %s" % filename)

                filename_without_extension, extension = os.path.splitext(filename)

                if imghdr.what(os.path.join(dirname, filename)) is None:

                    print("-> This is not an image: continue")

                    continue

                json_path = os.path.join(dirname, filename_without_extension + ".json")

                if not os.path.isfile(json_path):

                    print("-> has not a matching json: continue")

                    continue

                print("-> Processing %s" %json_path)

                with open(json_path) as json_data:

                    eso_data = json.load(json_data)

                    eso_object = eso_data['object']

                    eso_image = eso_data['image']

                    path_images = os.path.join(filename_without_extension, filename)

                    image_list = [path_images]

                    image_dir = filename_without_extension

                    natural_key = ItemMetadata.get_natural_key(collection, eso_image['id'])

                    if ItemMetadata.objects.filter(item__collection=collection, natural_key=natural_key).exists():

                        print('#### An item with ' +

                            natural_key +' for natural key, already exists in database in the import collection')

                    else:

                        try:

                            os.mkdir(os.path.join(settings.MEDIA_ROOT,'uploads', image_dir))

                            print(image_dir, "directory created")

                        except FileExistsError:

                            print(image_dir, "directory already exists")

                        try:

                            self.create_item_and_metadata(

                                natural_key, collection, eso_data, image_list, options, self.source_dir)

                        except Exception as e:

                            print("!!! Exception processing %s : %s" % (json_path, e))

                            continue

        print('# All done!')