81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
import os
|
|
|
|
from django.conf import settings
|
|
from django.shortcuts import get_object_or_404
|
|
from django.http import HttpResponse, FileResponse
|
|
|
|
from sorl.thumbnail import get_thumbnail
|
|
|
|
from .models import PostFileModel
|
|
|
|
|
|
def serve_content_file(request, file_hash):
|
|
"""
|
|
View function to serve content files for download or inline viewing
|
|
"""
|
|
|
|
print(f"Requesting file: {file_hash}")
|
|
|
|
download = request.GET.get("d")
|
|
|
|
thumbnail = request.GET.get("t")
|
|
|
|
try:
|
|
obj_file = get_object_or_404(PostFileModel, hash_blake3=file_hash)
|
|
file_name = obj_file.name.first().filename
|
|
file_type = obj_file.file_type
|
|
source_file = obj_file.file
|
|
|
|
if thumbnail:
|
|
if file_type != "image":
|
|
source_file = obj_file.thumbnail
|
|
|
|
if thumbnail == "sx":
|
|
file_name += ".thumb_64.jpg"
|
|
thumbnail_file = get_thumbnail(source_file.path, "64", upscale=False)
|
|
file = open(
|
|
os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
|
|
"rb",
|
|
)
|
|
elif thumbnail == "sm":
|
|
file_name += ".thumb_256.jpg"
|
|
thumbnail_file = get_thumbnail(source_file.path, "256", upscale=False)
|
|
file = open(
|
|
os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
|
|
"rb",
|
|
)
|
|
elif thumbnail == "md":
|
|
file_name += ".thumb_748.jpg"
|
|
thumbnail_file = get_thumbnail(source_file.path, "748", upscale=False)
|
|
file = open(
|
|
os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
|
|
"rb",
|
|
)
|
|
elif thumbnail == "lg":
|
|
file_name += ".thumb_1024.jpg"
|
|
thumbnail_file = get_thumbnail(source_file.path, "1024", upscale=False)
|
|
file = open(
|
|
os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
|
|
"rb",
|
|
)
|
|
elif thumbnail == "xl":
|
|
file_name += ".thumb_2048.jpg"
|
|
thumbnail_file = get_thumbnail(source_file.path, "2048", upscale=False)
|
|
file = open(
|
|
os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
|
|
"rb",
|
|
)
|
|
else:
|
|
file = source_file.file
|
|
|
|
response = FileResponse(file)
|
|
if download == "1":
|
|
response["Content-Disposition"] = f'attachment; filename="{file_name}"'
|
|
else:
|
|
response["Content-Disposition"] = f'inline; filename="{file_name}"'
|
|
|
|
return response
|
|
|
|
except Exception as e:
|
|
print(f"Error serving file: {e}")
|
|
return HttpResponse("File not found", status=404)
|