import os
import json
import sys
import shutil

import magic
import mimetypes

HUGO_IMAGES = '../static/files/images/'
TO_IMPORT = './to_import_interieur_nef/'
INITIAL_TAGS = ['interieur_nef']

current_index = 51

res = os.listdir(TO_IMPORT)

processed_images = []

for original_file_name in res:
    print(f'{current_index} loading {original_file_name}')
    folder_name = str(current_index).zfill(5)
    os.mkdir(HUGO_IMAGES + folder_name)
    old_path = TO_IMPORT + original_file_name
    mime_type = magic.from_file(old_path, mime=True)
    print(f'found mime {mime_type}')

    new_ext = list(filter(lambda x: x[1] == mime_type, mimetypes.types_map.items()))[0][0]

    new_path = HUGO_IMAGES + folder_name + '/original' + new_ext

    shutil.copyfile(old_path, new_path)

    processed_images.append({
                                'id': current_index,
                                'original_name': original_file_name,
                                'mime_type': mime_type,
                                'tags': INITIAL_TAGS
                                # name, description
                            })
    current_index += 1

of = open('./output.json', 'w')
of.write(json.dumps(processed_images))
of.close()