87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
import os
|
|
from datetime import datetime
|
|
|
|
from PIL import Image, ExifTags
|
|
|
|
# Variables
|
|
INPUT_DIR = "pictures_inbox"
|
|
DONE_DIR = "pictures_done"
|
|
OUTPUT_DIR = "output/pictures"
|
|
YEAR = datetime.now().strftime("%Y")
|
|
SMALL_SUFFIX = "_small"
|
|
IMAGE_FORMAT = "jpg"
|
|
max_width_resized=600 # pixels max de largeur
|
|
url_folder_pics="https://www.tykayn.fr/wp-uploads/content/i"
|
|
|
|
# Créer le dossier de sortie s'il n'existe pas
|
|
os.makedirs(os.path.join(OUTPUT_DIR, YEAR), exist_ok=True)
|
|
|
|
|
|
def get_exif_orientation(image):
|
|
"""Obtenir l'orientation de l'image à partir des métadonnées EXIF."""
|
|
try:
|
|
exif = image._getexif()
|
|
if exif is not None:
|
|
for orientation in ExifTags.TAGS.keys():
|
|
if ExifTags.TAGS[orientation] == 'Orientation':
|
|
return exif[orientation]
|
|
except AttributeError:
|
|
pass
|
|
return None
|
|
|
|
|
|
def apply_orientation(image, orientation):
|
|
"""Appliquer l'orientation à l'image."""
|
|
if orientation == 3:
|
|
return image.rotate(180, expand=True)
|
|
elif orientation == 6:
|
|
return image.rotate(270, expand=True)
|
|
elif orientation == 8:
|
|
return image.rotate(90, expand=True)
|
|
return image
|
|
|
|
|
|
if len(os.listdir(INPUT_DIR)):
|
|
print('traitement des images dans pictures inbox', len(os.listdir(INPUT_DIR)))
|
|
# Parcourir toutes les images dans le dossier d'entrée
|
|
for filename in os.listdir(INPUT_DIR):
|
|
file_path = os.path.join(INPUT_DIR, filename)
|
|
# Vérifier si c'est bien un fichier
|
|
if os.path.isfile(file_path) and (file_path.lower().endswith('.jpg') or file_path.lower().endswith('.png')):
|
|
# Récupérer le nom de base de l'image et son extension
|
|
base_name = os.path.splitext(filename)
|
|
extension = os.path.splitext(filename)[1].lower()
|
|
|
|
# Créer le nom pour la version réduite
|
|
small_image_name = f"{base_name[0]}{SMALL_SUFFIX}.{IMAGE_FORMAT}"
|
|
|
|
# Chemins des images
|
|
input_image_path = os.path.join(INPUT_DIR, filename)
|
|
done_image_path = os.path.join(DONE_DIR, filename)
|
|
small_image_path = os.path.join(OUTPUT_DIR, YEAR, small_image_name)
|
|
original_image_path = os.path.join(OUTPUT_DIR, YEAR, filename)
|
|
|
|
# Ouvrir l'image et appliquer l'orientation
|
|
with Image.open(input_image_path) as img:
|
|
orientation = get_exif_orientation(img)
|
|
if orientation is not None:
|
|
img = apply_orientation(img, orientation)
|
|
|
|
# Redimensionner l'image
|
|
img = img.resize((max_width_resized, int(img.height * max_width_resized / img.width)), Image.Resampling.LANCZOS)
|
|
img.save(small_image_path, 'JPEG') # Utiliser 'JPEG' au lieu de 'JPG'
|
|
|
|
# Copier l'image originale dans le dossier de sortie
|
|
with open(input_image_path, 'rb') as f:
|
|
with open(original_image_path, 'wb') as f_out:
|
|
f_out.write(f.read())
|
|
# déplacer l'image originale dans le dossier des images traitées
|
|
os.rename(input_image_path, done_image_path)
|
|
# Écrire la ligne pour le document .org
|
|
|
|
org_line = f"\n\n[[{url_folder_pics}/{YEAR}/{small_image_name}][{filename}]]\n"
|
|
with open(os.path.join("output", f"images_{YEAR}.org"), "a") as org_file:
|
|
org_file.write(org_line)
|
|
|
|
print("Traitement terminé.")
|