c3lf-system-3/core/files/media_v1.py

66 lines
2.3 KiB
Python
Raw Normal View History

2024-01-07 15:32:58 +00:00
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
2023-11-20 20:01:03 +00:00
from rest_framework.decorators import api_view, permission_classes, authentication_classes
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'])
2023-11-20 20:01:03 +00:00
@permission_classes([])
@authentication_classes([])
def media_urls(request, hash):
try:
2023-11-20 20:01:03 +00:00
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': '*',
}) # TODO Expires and Cache-Control
except File.DoesNotExist:
return Response(status=status.HTTP_404_NOT_FOUND)
@swagger_auto_schema(method='GET', auto_schema=None)
@api_view(['GET'])
2023-11-20 20:01:03 +00:00
@permission_classes([])
@authentication_classes([])
def thumbnail_urls(request, hash):
size = 200
try:
2023-11-20 20:01:03 +00:00
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
2024-01-07 15:32:58 +00:00
image = Image.open(file.file)
image.thumbnail((size, size))
rgb_image = image.convert('RGB')
2023-11-20 20:17:27 +00:00
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,
2023-11-20 20:17:27 +00:00
content_type="image/jpeg",
headers={
'X-Accel-Redirect': f'/redirect_thumbnail/{size}/{hash_path}',
'Access-Control-Allow-Origin': '*',
}) # TODO Expires and Cache-Control
except File.DoesNotExist:
return Response(status=status.HTTP_404_NOT_FOUND)
urlpatterns = [
2023-11-20 20:01:03 +00:00
path('thumbs/<path:hash>', thumbnail_urls),
path('images/<path:hash>', media_urls),
2023-11-20 20:17:27 +00:00
]