49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
from django.views.generic import TemplateView
|
|
from django.shortcuts import get_object_or_404
|
|
from django.http import FileResponse as BaseFileResponse
|
|
from .models import SharedFile, FileToken
|
|
import mimetypes
|
|
from django.utils import timezone
|
|
from django.contrib.auth.decorators import login_required
|
|
|
|
|
|
mimetypes.init()
|
|
|
|
|
|
class SharedFileDetails(TemplateView):
|
|
template_name = "file.html"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context['file'] = get_object_or_404(SharedFile, short_id=kwargs['id'])
|
|
context['token'] = FileToken.objects.create(file=context['file'])
|
|
return context
|
|
|
|
|
|
def FileResponse(file):
|
|
response = BaseFileResponse(file.file)
|
|
response['Content-Type'] = mimetypes.guess_type(file.get_original_filename())[0]
|
|
response['Content-Disposition'] = 'attachment; filename="{0}"'.format(file.get_original_filename())
|
|
return response
|
|
|
|
|
|
def file_download(request, id, token):
|
|
time_threshold = timezone.now() - FileToken.valid_time
|
|
FileToken.objects.filter(created__lt=time_threshold)
|
|
file = get_object_or_404(SharedFile, short_id=id, published=True)
|
|
token = get_object_or_404(FileToken, token=token, file=file)
|
|
token.delete() # delete after used
|
|
return FileResponse(file)
|
|
|
|
|
|
def hotlink_file_download(request, id):
|
|
file = get_object_or_404(SharedFile, short_id=id, hotlink=True, published=True)
|
|
return FileResponse(file)
|
|
|
|
|
|
@login_required
|
|
def uploaded_file(request, filename):
|
|
file = get_object_or_404(SharedFile, file=request.get_full_path()[1:]) # strip preceding slash
|
|
response = FileResponse(file)
|
|
return response
|