在Django管理员上传后提示文件下载



我有一个django admin页面,其中用户正在上传文件,读取数据 in-memory ,而无需实际存储文件。

基于内容,在磁盘上生成了一个新文件。这在 clean方法中发生在 save_model被调用之前。

def clean(self):
    obj = Parser()
    cleaned_data = super(MyMethodAdminForm, self).clean()
    input_data = self.request.FILES['file_name'].read().split('n')
    base_file = settings.MEDIA_ROOT + "/uploads/" + cleaned_data.get('file_name').name
    # This is the output file, which needs to be downloaded
    output_filename = base_file.replace('.csv', '_output.csv')
    output_list = []
    # Read the file and prepare the output data
    # on the basis of the input data
    for line in input_data:
        data = line.split(",")
        if len(data) > 1:
            dt = obj.get_data()
            current_list = [data[0], data[1]]
        else:
            dt = obj.get_data()
            current_list = [data[0]]
        current_list.append(str(dt))
        output_list.append(current_list)
    # Prepare the file on the basis of the output data
    with open(output_filename, 'a') as outcsv:
        # configure writer to write standard csv file
        writer = csv.writer(outcsv, delimiter=',', quotechar=' ', quoting=csv.QUOTE_MINIMAL, lineterminator='n')
        for item in output_list:
            if len(item) > 2:
                # Write item to outcsv
                writer.writerow([item[0], item[1], item[2]])
            else:
                writer.writerow([item[0], item[1]])
    return cleaned_data

现在,我的问题是我必须允许此文件下载,但是由于上传域是通过负载平衡器处理的,所以我无法真正提供链接(老实说,也不是首选的方式)。有什么办法可以强制提示文件下载?

我从下面的两个问题中得到了一些提示,但后来我不了解这些change header thing

  • 如何在Django管理中的FileField中显示下载链接?

  • 生成使用Django下载的文件

编辑

进行了更多调试后,我尝试更新admin Redirect中建议的response_change

    class MyMethodAdmin(admin.ModelAdmin):
        form = MyMethodAdminForm
        def response_change(self, request, obj):
            import ipdb;
            ipdb.set_trace()
            # response = HttpResponse(content_type='text/csv')
            # response['Content-Disposition'] = 'attachment; filename="%s"' % form.cleaned_data['output_file_path'] (somehow access the file-path, its incorrect ryt now, i know
            super(MyMethodAdmin, self).response_change(request, obj)

,但甚至没有被称为PDB ....所以还没有帮助!:(

完成课程:

class MyMethodAdminForm(forms.ModelForm):
    class Meta(object):
        model = MyMethod
        fields = ["file_name"]
    def clean(self):
        obj = Parser()
        cleaned_data = super(MyMethodAdminForm, self).clean()
        input_data = self.request.FILES['file_name'].read().split('n')
        base_file = settings.MEDIA_ROOT + "/uploads/" + cleaned_data.get('file_name').name
        output_filename = base_file.replace('.csv', '_output.csv')
        output_list = []
        # Read the file and prepare the output data
        # on the basis of the input data
        for line in input_data:
            data = line.split(",")
            if len(data) > 1:
                dt = obj.get_data()
                current_list = [data[0], data[1]]
            else:
                dt = obj.get_data()
                current_list = [data[0]]
            current_list.append(str(dt))
            output_list.append(current_list)
        # Prepare the file on the basis of the output data
        with open(output_filename, 'a') as outcsv:
            # configure writer to write standard csv file
            writer = csv.writer(outcsv, delimiter=',', quotechar=' ', quoting=csv.QUOTE_MINIMAL, lineterminator='n')
            for item in output_list:
                if len(item) > 2:
                    # Write item to outcsv
                    writer.writerow([item[0], item[1], item[2]])
                else:
                    writer.writerow([item[0], item[1]])
        return cleaned_data

无论您准备将答复发送给用户:

response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = 'attachment; filename="%s"' % filename
writer = csv.writer(response)
# now write to writer however you want to output your CSV

根据您从何处呼唤您的视图,您可能需要覆盖多种方法。原因是Django支持不同类型的管理员操作,包括changeadddelete,您可以在此处阅读。我不确定您正在使用哪一个,因此您可能必须同时尝试两者。作为第一步,我将尝试覆盖response_changeresponse_add方法:

class MyMethodAdmin(admin.ModelAdmin):
    form = MyMethodAdminForm
    def return_csv(self):
        response = HttpResponse(content_type='text/csv')
        response['Content-Disposition'] = 'attachment; filename="%s"' % filename
        writer = csv.writer(response)
        writer.writerow(['Django','Admin'])
        return response
    def response_change(self, request, obj):
        return self.return_csv()
    def response_add(self, request, obj):
        return self.return_csv()

最新更新