create file API v2 and X-Accel-Redirect endpoint for images

2024-01-07 21:06:06 +01:00 · 2024-01-07 21:06:06 +01:00 · 77828295f8
commit 77828295f8
parent ab5e8f36d1
10 changed files with 164 additions and 1 deletions
--- a/core/files/media_v2.py
+++ b/core/files/media_v2.py
@ -0,0 +1,80 @@
+from datetime import datetime, timedelta
+
+import os
+from django.http import HttpResponse
+from django.urls import path
+from drf_yasg.utils import swagger_auto_schema
+from rest_framework import status
+from rest_framework.decorators import api_view, permission_classes
+from rest_framework.permissions import IsAuthenticated
+from rest_framework.response import Response
+
+from core.settings import MEDIA_ROOT
+from files.models import File
+
+
+@swagger_auto_schema(method='GET', auto_schema=None)
+@api_view(['GET'])
+@permission_classes([IsAuthenticated])
+def media_urls(request, hash):
+    try:
+        if request.META.get('HTTP_IF_NONE_MATCH') and request.META.get('HTTP_IF_NONE_MATCH') == hash:
+            return HttpResponse(status=status.HTTP_304_NOT_MODIFIED)
+
+        file = File.objects.get(hash=hash)
+        hash_path = file.file
+        return HttpResponse(status=status.HTTP_200_OK,
+                            content_type=file.mime_type,
+                            headers={
+                                'X-Accel-Redirect': f'/redirect_media/{hash_path}',
+                                'Access-Control-Allow-Origin': '*',
+                                'Cache-Control': 'max-age=31536000, private',
+                                'Expires': datetime.utcnow() + timedelta(days=365),
+                                'Age': 0,
+                                'ETag': file.hash,
+                            })
+
+    except File.DoesNotExist:
+        return Response(status=status.HTTP_404_NOT_FOUND)
+
+
+@swagger_auto_schema(method='GET', auto_schema=None)
+@api_view(['GET'])
+@permission_classes([IsAuthenticated])
+def thumbnail_urls(request, size, hash):
+    if size not in [32, 64, 256]:
+        return Response(status=status.HTTP_404_NOT_FOUND)
+    if request.META.get('HTTP_IF_NONE_MATCH') and request.META.get('HTTP_IF_NONE_MATCH') == hash + "_" + str(size):
+        return HttpResponse(status=status.HTTP_304_NOT_MODIFIED)
+    try:
+        file = File.objects.get(hash=hash)
+        hash_path = file.file
+        if not os.path.exists(MEDIA_ROOT + f'/thumbnails/{size}/{hash_path}'):
+            from PIL import Image
+            image = Image.open(file.file)
+            image.thumbnail((size, size))
+            rgb_image = image.convert('RGB')
+            thumb_dir = os.path.dirname(MEDIA_ROOT + f'/thumbnails/{size}/{hash_path}')
+            if not os.path.exists(thumb_dir):
+                os.makedirs(thumb_dir)
+            rgb_image.save(MEDIA_ROOT + f'/thumbnails/{size}/{hash_path}', 'jpeg', quality=90)
+
+        return HttpResponse(status=status.HTTP_200_OK,
+                            content_type="image/jpeg",
+                            headers={
+                                'X-Accel-Redirect': f'/redirect_thumbnail/{size}/{hash_path}',
+                                'Access-Control-Allow-Origin': '*',
+                                'Cache-Control': 'max-age=31536000, private',
+                                'Expires': datetime.utcnow() + timedelta(days=365),
+                                'Age': 0,
+                                'ETag': file.hash + "_" + str(size),
+                            })
+
+    except File.DoesNotExist:
+        return Response(status=status.HTTP_404_NOT_FOUND)
+
+
+urlpatterns = [
+    path('<int:size>/<path:hash>/', thumbnail_urls),
+    path('<path:hash>/', media_urls),
+]