website/import_mess/import_1.py

45 lines
1.2 KiB
Python
Raw Normal View History

2022-09-28 20:49:59 +00:00
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()