31 lines
1.8 KiB
Python
31 lines
1.8 KiB
Python
import mimetypes
|
|
from django.conf import settings
|
|
from django.core.exceptions import PermissionDenied
|
|
from django.http import FileResponse
|
|
from django.shortcuts import get_object_or_404
|
|
from apps.tickets.policies import TicketPolicy
|
|
from apps.accounts.magic_access import get_ticket_magic_access
|
|
from .delivery import signed_media_redirect
|
|
from .models import Attachment
|
|
def _authorized_attachment(request,public_id):
|
|
attachment=get_object_or_404(Attachment.objects.select_related("message__ticket"),public_id=public_id); ticket=attachment.message.ticket
|
|
magic_link = None if request.user.is_authenticated else get_ticket_magic_access(request,ticket)
|
|
if not TicketPolicy.can_view(request.user,ticket) and not magic_link: raise PermissionDenied
|
|
if attachment.message.is_internal and (magic_link or getattr(request.user,"account_type",None)=="CUSTOMER"): raise PermissionDenied
|
|
if attachment.message.deleted_at and not request.user.has_perm("messaging.view_deleted_message_content"): raise PermissionDenied
|
|
return attachment
|
|
def view(request,public_id):
|
|
attachment=_authorized_attachment(request,public_id)
|
|
if settings.USE_NGINX_MEDIA:
|
|
return signed_media_redirect(attachment.relative_path)
|
|
content_type=attachment.mime_type or mimetypes.guess_type(attachment.original_name)[0] or "application/octet-stream"
|
|
response=FileResponse(attachment.file.open("rb"),content_type=content_type)
|
|
response["Content-Disposition"]=f'inline; filename="{attachment.original_name}"'
|
|
response["X-Content-Type-Options"]="nosniff"
|
|
return response
|
|
def download(request,public_id):
|
|
attachment=_authorized_attachment(request,public_id)
|
|
if settings.USE_NGINX_MEDIA:
|
|
return signed_media_redirect(attachment.relative_path, download=True)
|
|
return FileResponse(attachment.file.open("rb"),as_attachment=True,filename=attachment.original_name)
|