forked from 0x2620/pandora
47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
from django.http import HttpResponse,Http404
|
|
from django.core.servers.basehttp import FileWrapper
|
|
from django.conf import settings
|
|
|
|
import mimetypes
|
|
import os
|
|
|
|
def basic_sendfile(fname,download_name=None):
|
|
if not os.path.exists(fname):
|
|
raise Http404
|
|
|
|
wrapper = FileWrapper(open(fname,"r"))
|
|
|
|
content_type = mimetypes.guess_type(fname)[0]
|
|
response = HttpResponse(wrapper, content_type=content_type)
|
|
response['Content-Length'] = os.path.getsize(fname)
|
|
|
|
if download_name:
|
|
response['Content-Disposition'] = "attachment; filename=%s"%download_name
|
|
|
|
return response
|
|
|
|
def x_sendfile(fname,download_name=None):
|
|
if not os.path.exists(fname):
|
|
raise Http404
|
|
|
|
content_type = mimetypes.guess_type(fname)[0]
|
|
response = HttpResponse('', content_type=content_type)
|
|
response['Content-Length'] = os.path.getsize(fname)
|
|
response['X-Sendfile'] = fname
|
|
|
|
if download_name:
|
|
response['Content-Disposition'] = "attachment; filename=%s"%download_name
|
|
|
|
return response
|
|
|
|
try:
|
|
__sendfile = getattr(settings,'SENDFILE',False) == 'x_sendfile'
|
|
except:
|
|
__sendfile = False
|
|
if __sendfile == 'x_sendfile':
|
|
sendfile = x_sendfile
|
|
else:
|
|
sendfile = basic_sendfile
|
|
|