def file_upload(request):
context = {}
form = FileForm()
if request.method == "POST":
form = FileForm(request.POST or None, files=request.FILES)
if form.is_valid():
file = form.save(commit=False)
file.save()
messages.success(request, "Successfully uploaded file.")
return redirect('files:file_upload')
context['form'] = form
return render(request, 'files/upload.html', context)
def test_form_can_be_validated(self):
form = FileForm(data={
'name': 'Demo Name',
'desc': 'This is a demo description',
},
files={'file': get_test_file()})
self.assertTrue(form.is_valid())
def NewFile(request):
if request.user.is_authenticated:
if request.method == "POST":
fform = FileForm(request.POST , request.FILES)
if fform.is_valid():
ff = fform.save(commit=False)
ff.owner = request.user
ff.name = request.FILES['filedata'].name
ff.size = request.FILES['filedata'].size
with open(UPLOAD_DIR+ff.name.encode("hex") , "wb+") as f:
for chunk in request.FILES['filedata'].chunks():
f.write(chunk)
ff.save()
return HttpResponseRedirect('/files')
else:
return render(request , 'new_file.html' , {'form' : fform})
else:
fform = FileForm()
return render(request , 'new_file.html' , {'form':fform})
return HttpResponseRedirect('/files')
def create_file(request):
'''
:param request:
:return:
'''
form = FileForm(request.POST or None, request.FILES or None)
if form.is_valid():
collection_name = create_collection(request, form)
instance = form.save(commit=False)
instance.collection = collection_name
instance.save()
return redirect(reverse('files:list'))
else:
return render(request, 'files/create.html', {'form': form})
def test_form_can_save_data_in_db(self):
user_obj = get_user_obj_by_name('Zlatan Ibrahimovic')
form = FileForm(data={
'name': 'Demo Name',
'desc': 'This is a demo description',
},
files={'file': get_test_file()})
if form.is_valid():
instance = form.save(user_obj)
file_obj = File.objects.first()
self.assertEqual(file_obj.name, instance.name)
self.assertEqual(file_obj.desc, instance.desc)
self.assertEqual(file_obj.user.username, instance.user.username)
self.assertEqual(file_obj.file, instance.file)
def update_file(request, id):
'''
:param request:
:return:
'''
instance = get_object_or_404(File, id=id)
form = FileForm(request.POST or None,
request.FILES or None,
instance=instance)
if form.is_valid():
collection_name = update_collection(request,
form,
pre_instance=instance)
instance = form.save(commit=False)
instance.collection = collection_name
instance.save()
return redirect(reverse('files:list'))
else:
return render(request, 'files/update.html', {'form': form})
def list_file(request):
'''
:param request:
:return:a response
'''
report_form = ReportForm(request.POST or None)
file_form = FileForm(request.POST or None, request.FILES or None)
if file_form.is_valid():
collection_name = create_collection(request, file_form)
instance = file_form.save(commit=False)
instance.collection = collection_name
instance.save()
collection_obj = get_collection(request, collection_name)
field_list, _ = get_fields_and_data_from_collection(collection_obj)
field_instance_list = []
for turn in field_list:
field_instance_list.append(
Field.objects.create(name=turn[0], type=turn[1],
file=instance))
print(Field.objects.all())
return redirect(reverse('files:list'))
elif report_form.is_valid():
instance = report_form.save()
narrative_instance = Narrative.objects.create(report=instance)
narrative_instance.segments.create(index=0)
print(instance.name, instance.file)
return redirect(reverse('reports:editor', kwargs={'id': instance.id}))
else:
file_qs = File.objects.all()
collection_list = []
for turn in file_qs:
collection_list.append(get_summary_of_collection(request, turn))
context = {
'collection_list': collection_list,
'file_form': file_form,
'report_form': report_form,
}
return render(request, 'files/list.html', context)
def test_form_can_update_file_info(self):
user_obj = get_user_obj_by_name('Zlatan Ibrahimovic')
file_obj = File.objects.create(
name='test_bank_revenue.csv',
desc='Data of test banks revenue',
file=get_test_file(),
user=user_obj,
)
form = FileForm(instance=file_obj,
data={
'name': 'Demo Name',
'desc': 'This is a demo description',
},
files={'file': get_test_file()})
if form.is_valid():
instance = form.save(user_obj)
self.assertEqual(File.objects.count(), 1)
file_obj = File.objects.first()
self.assertEqual(file_obj.name, instance.name)
self.assertEqual(file_obj.desc, instance.desc)
self.assertEqual(file_obj.file, instance.file)
