Скачивание CSV-файла в Django - PullRequest
7 голосов
/ 17 марта 2010

Я пытаюсь загрузить CSV-файл, используя HttpResponse, чтобы браузер считал его вложением Я следую инструкциям здесь , но мой браузер не выводит диалоговое окно "Сохранить как" Я не могу понять, что не так с моей функцией. Вся помощь приветствуется.

  dev savefile(request):
        try:
            myfile = request.GET['filename']
            filepath = settings.MEDIA_ROOT + 'results/'
            destpath = os.path.join(filepath, myfile)
            response = HttpResponse(FileWrapper(file(destpath)), mimetype='text/csv' ) 
            response['Content-Disposition'] = 'attachment; filename="%s"' %(myfile)
            return response
        except Exception, err:
            errmsg = "%s"%(err)
            return HttpResponse(errmsg)

С Днем Пэт!

Ответы [ 5 ]

9 голосов
/ 18 марта 2010

Если файл статический (т. Е. Не сгенерирован специально для этого запроса), вы все равно не должны передавать его через django. Вы должны настроить некоторый путь (например, / static /), который будет обслуживаться вашим веб-сервером, и сохранить все служебные данные django.

Если файл динамический , есть 2 варианта:

  1. Создайте это в памяти и подайте из Джанго.
  2. Создайте его на диске и верните ему HttpResponseRedirect, чтобы ваш веб-сервер занимался самой загрузкой (если файл очень большой, вам следует использовать эту опцию).

Что касается динамического обслуживания, я использовал следующий код (упрощенная версия ExcelResponse )

import StringIO
from django.db.models.query import ValuesQuerySet, QuerySet

class CSVResponse(HttpResponse):

  def __init__(self, data, output_name='data', headers=None, encoding='utf8'):

    # Make sure we've got the right type of data to work with
    valid_data = False
    if isinstance(data, ValuesQuerySet):
        data = list(data)
    elif isinstance(data, QuerySet):
        data = list(data.values())
    if hasattr(data, '__getitem__'):
        if isinstance(data[0], dict):
            if headers is None:
                headers = data[0].keys()
            data = [[row[col] for col in headers] for row in data]
            data.insert(0, headers)
        if hasattr(data[0], '__getitem__'):
            valid_data = True
    assert valid_data is True, "CSVResponse requires a sequence of sequences"

    output = StringIO.StringIO()
    for row in data:
        out_row = []
        for value in row:
            if not isinstance(value, basestring):
                value = unicode(value)
            value = value.encode(encoding)
            out_row.append(value.replace('"', '""'))
        output.write('"%s"\n' %
                     '","'.join(out_row))            
    mimetype = 'text/csv'
    file_ext = 'csv'
    output.seek(0)
    super(CSVResponse, self).__init__(content=output.getvalue(),
                                        mimetype=mimetype)
    self['Content-Disposition'] = 'attachment;filename="%s.%s"' % \
        (output_name.replace('"', '\"'), file_ext)

Чтобы использовать его, просто используйте return CSVResponse (...), передавая список списков, список диктовок (с теми же ключами), QuerySet, ValuesQuerySet

5 голосов
/ 17 марта 2010

Вы пытались указать тип контента? например,

response['Content-Type'] = 'application/x-download';

Edit:

Обратите внимание, этот код успешно запускает диалоговое окно "Сохранить как" для меня. Примечание. Я указываю «application / x-download» непосредственно в аргументе mimetype. Вы также можете перепроверить свой код и убедиться, что путь к файлу указан правильно, а FileWrapper () не делает ничего странного.

def save_file(request):
    data = open(os.path.join(settings.PROJECT_PATH,'data/table.csv'),'r').read()
    resp = django.http.HttpResponse(data, mimetype='application/x-download')
    resp['Content-Disposition'] = 'attachment;filename=table.csv'
    return resp
3 голосов
/ 18 марта 2010

Спасибо всем за ваши предложения. Я выбрал несколько новых трюков :) Однако я думаю, что нашел ответ на мою проблему здесь: Загрузка CSV через AJAX Моя функция «savefile» вызывается через Ajax-запрос, и кажется, что у ajax есть ограничение, при котором «сохранить как диалоговое окно» не отображается независимо от заголовков HTTP.

Я должен был упомянуть, что использую Ajax для вызова этой функции, но мне никогда не приходило в голову, что это может быть проблемой. :) Спасибо, StackOverflow!

1 голос
/ 02 июля 2010

Томас, я использовал функцию Ajax для сохранения и загрузки этого файла.Похоже, что в этом случае поле «Сохранить как» не появится независимо от заголовков.Я просто использовал javascript для загрузки этого файла.window.open ( "путь / к / файлу");и это делает трюк.Я проверил на IE6 и Firefox, и появляется диалоговое окно.

0 голосов
/ 18 марта 2010

Имеет ли значение, если вы не заключаете имя файла в двойные кавычки? В примере кода не указывается имя файла:

response['Content-Disposition'] = 'attachment; filename=foo.xls'

но ваш код делает:

response['Content-Disposition'] = 'attachment; filename="foo.xls"'
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...