我有一个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支持不同类型的管理员操作,包括change
,add
,delete
,您可以在此处阅读。我不确定您正在使用哪一个,因此您可能必须同时尝试两者。作为第一步,我将尝试覆盖response_change
和response_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()