First Commit

gallery/views.py

@@ -0,0 +1,52 @@
+from django.shortcuts import render
+from django.contrib import messages
+from .forms import AddImgGallery
+from django.contrib.auth.decorators import login_required
+import os
+from django.conf import settings
+
+@login_required()
+def home_gallery(request):
+    user_id = request.user.id
+    user_directory = os.path.join(settings.MEDIA_ROOT, 'galleries', str(user_id))
+    img_urls = []
+
+    if os.path.exists(user_directory):
+        for filename in os.listdir(user_directory):
+            if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')):
+                img_urls.append(f"{settings.MEDIA_URL}galleries/{user_id}/{filename}")
+
+    return render(request, 'gallery/home.html', {'images': img_urls})
+
+@login_required()
+def import_img(request):
+    if request.method == "POST":
+        form = AddImgGallery(request.POST, request.FILES)
+        if form.is_valid():
+            image = form.clean_img()
+            user_id = request.user.id 
+            saved_image_url = save_image_to_gallery(image, user_id)
+
+            messages.success(request, 'Votre image a bien été ajoutée à la galerie')
+    form = AddImgGallery()
+    return render(request, 'gallery/form.html', {'form': form})
+
+def save_image_to_gallery(image, user_id):
+    user_directory = os.path.join(settings.MEDIA_ROOT, 'galleries', str(user_id))
+    os.makedirs(user_directory, exist_ok=True) 
+    
+    # Créer un fichier index.html vide pour sécuriser le répertoire
+    index_file_path = os.path.join(user_directory, 'index.html')
+    if not os.path.exists(index_file_path):  # Vérifie si le fichier n'existe pas déjà
+        with open(index_file_path, 'w') as index_file:
+            index_file.write('')  # Écrit un fichier vide
+    
+    # Définir le chemin complet du fichier
+    file_path = os.path.join(user_directory, image.name)
+
+    # Enregistrer le fichier
+    with open(file_path, 'wb+') as destination:
+        for chunk in image.chunks():
+            destination.write(chunk)
+
+    return f"{settings.MEDIA_URL}galleries/{user_id}/{image.name}"