57 lines
2 KiB
Python
57 lines
2 KiB
Python
# -*- coding: utf-8 -*-
|
|
from datetime import datetime, timedelta
|
|
from urllib.parse import quote
|
|
import mimetypes
|
|
import os
|
|
|
|
from django.http import HttpResponse, Http404
|
|
from django.conf import settings
|
|
|
|
|
|
def HttpFileResponse(path, content_type=None, filename=None):
|
|
if not os.path.exists(path):
|
|
raise Http404
|
|
if not content_type:
|
|
content_type = mimetypes.guess_type(path)[0]
|
|
if not content_type:
|
|
content_type = 'application/octet-stream'
|
|
|
|
if getattr(settings, 'XACCELREDIRECT', False):
|
|
response = HttpResponse()
|
|
response['Content-Length'] = os.stat(path).st_size
|
|
|
|
for PREFIX in ('STATIC', 'MEDIA'):
|
|
root = getattr(settings, PREFIX+'_ROOT', '')
|
|
url = getattr(settings, PREFIX+'_URL', '')
|
|
if root and path.startswith(root):
|
|
path = url + path[len(root)+1:]
|
|
if not isinstance(path, bytes):
|
|
path = path.encode('utf-8')
|
|
response['X-Accel-Redirect'] = path
|
|
if content_type:
|
|
response['Content-Type'] = content_type
|
|
elif getattr(settings, 'XSENDFILE', False):
|
|
response = HttpResponse()
|
|
if not isinstance(path, bytes):
|
|
path = path.encode('utf-8')
|
|
response['X-Sendfile'] = path
|
|
if content_type:
|
|
response['Content-Type'] = content_type
|
|
response['Content-Length'] = os.stat(path).st_size
|
|
else:
|
|
response = HttpResponse(open(path, 'rb'), content_type=content_type)
|
|
if filename:
|
|
if not isinstance(filename, bytes):
|
|
filename = filename.encode('utf-8')
|
|
response['Content-Disposition'] = "attachment; filename*=UTF=8''%s" % quote(filename)
|
|
|
|
response['Expires'] = datetime.strftime(datetime.utcnow() + timedelta(days=1), "%a, %d-%b-%Y %H:%M:%S GMT")
|
|
|
|
def allow_access():
|
|
for key in ('X-Accel-Redirect', 'X-Sendfile'):
|
|
if key in response:
|
|
del response[key]
|
|
response['Access-Control-Allow-Origin'] = '*'
|
|
response.allow_access = allow_access
|
|
return response
|
|
|