def _is_post_data_valid(ret, tables_info, post_data):
for table in tables_info:
table_rows = _get_table_rows_from_post(table.get('name'), post_data)
if len(table_rows) == 0:
return False
schema = Schema(ret.return_type.get_schema_by_name(table.get('name')))
if not schema.is_all_valid(table_rows):
return False
return True
def _is_post_data_valid(ret, tables_info, post_data):
for table in tables_info:
table_rows = _get_table_rows_from_post(table.get("name"), post_data)
if len(table_rows) == 0:
return False
schema = Schema(ret.return_type.get_schema_by_name(table.get("name")))
if not schema.is_all_valid(table_rows):
return False
return True
def post(self, request, *args, **kwargs):
context = self.get_context_data()
ret = context['return']
if 'upload' in request.POST:
form = UploadSpreadsheetForm(request.POST, request.FILES)
if form.is_valid():
temp_file_dir = tempfile.mkdtemp(dir=settings.MEDIA_ROOT)
try:
data = form.cleaned_data.get('spreadsheet_file')
path = default_storage.save(os.path.join(temp_file_dir, str(data)), ContentFile(data.read()))
workbook = excel.load_workbook_content(path)
for table in context['tables']:
worksheet = excel.get_sheet(workbook, table.get('title')) \
or excel.get_sheet(workbook, table.get('name'))
if worksheet is not None:
table_data = excel.TableData(worksheet)
schema = Schema(ret.return_type.get_schema_by_name(table.get('name')))
excel_rows = list(table_data.rows_by_col_header_it())
has_errors = not schema.is_all_valid(excel_rows)
if has_errors:
messages.error(request, "Your return contains some errors. See below.")
validated_rows = list(schema.rows_validator(excel_rows))
# We want to stringify the datetime/date that might have been created by the excel parser
for vr in validated_rows:
for col, validation in vr.items():
value = validation.get('value')
if isinstance(value, datetime.datetime) or isinstance(value, datetime.date):
validation['value'] = value.strftime(DATE_FORMAT)
table['data'] = validated_rows
else:
messages.warning(request, 'Missing worksheet ' + table.get('name'))
finally:
shutil.rmtree(temp_file_dir)
else:
context['upload_spreadsheet_form'] = form
elif 'draft' in request.POST or 'draft_continue' in request.POST:
_create_return_data_from_post_data(ret, context['tables'], request.POST)
if is_officer(request.user):
ret.proxy_customer = request.user
ret.status = 'draft'
ret.save()
messages.warning(request, 'Return saved as draft.')
# redirect or reshow page depending on whether save or save/continue was clicked
if 'draft' in request.POST:
return redirect('home')
else:
for table in context['tables']:
table['data'] = _get_validated_rows_from_post(ret, table.get('name'), request.POST)
elif 'lodge' in request.POST:
if _is_post_data_valid(ret, context['tables'], request.POST):
_create_return_data_from_post_data(ret, context['tables'], request.POST)
self._set_submitted(ret)
return redirect('home')
else:
for table in context['tables']:
table['data'] = _get_validated_rows_from_post(ret, table.get('name'), request.POST)
if len(table['data']) == 0:
messages.warning(request,
"You must enter data for {} or submit a Nil Return".format(table.get('name')))
else:
messages.error(request,
"Your return contains some errors. See below.")
elif 'nil' in request.POST:
form = NilReturnForm(request.POST)
if form.is_valid():
ret.nil_return = True
ret.comments = form.cleaned_data['comments']
self._set_submitted(ret)
return redirect('home')
return render(request, self.template_name, context)
def post(self, request, *args, **kwargs):
context = self.get_context_data()
ret = context['return']
if 'upload' in request.POST:
form = UploadSpreadsheetForm(request.POST, request.FILES)
if form.is_valid():
temp_file_dir = tempfile.mkdtemp(dir=settings.MEDIA_ROOT)
try:
data = form.cleaned_data.get('spreadsheet_file')
path = default_storage.save(os.path.join(temp_file_dir, str(data)), ContentFile(data.read()))
workbook = excel.load_workbook_content(path)
for table in context['tables']:
worksheet = excel.get_sheet(workbook, table.get('title')) \
or excel.get_sheet(workbook, table.get('name'))
if worksheet is not None:
table_data = excel.TableData(worksheet)
schema = Schema(ret.return_type.get_schema_by_name(table.get('name')))
excel_rows = list(table_data.rows_by_col_header_it())
has_errors = not schema.is_all_valid(excel_rows)
if has_errors:
messages.error(request, "Your return contains some errors. See below.")
validated_rows = list(schema.rows_validator(excel_rows))
# We want to stringify the datetime/date that might have been created by the excel parser
for vr in validated_rows:
for col, validation in vr.items():
value = validation.get('value')
if isinstance(value, datetime.datetime) or isinstance(value, datetime.date):
validation['value'] = value.strftime(DATE_FORMAT)
table['data'] = validated_rows
else:
messages.warning(request, 'Missing worksheet ' + table.get('name'))
finally:
shutil.rmtree(temp_file_dir)
else:
context['upload_spreadsheet_form'] = form
elif 'draft' in request.POST or 'draft_continue' in request.POST:
_create_return_data_from_post_data(ret, context['tables'], request.POST)
if is_officer(request.user):
ret.proxy_customer = request.user
ret.status = 'draft'
ret.save()
messages.warning(request, 'Return saved as draft.')
# redirect or reshow page depending on whether save or save/continue was clicked
if 'draft' in request.POST:
return redirect('home')
else:
for table in context['tables']:
table['data'] = _get_validated_rows_from_post(ret, table.get('name'), request.POST)
elif 'lodge' in request.POST:
if _is_post_data_valid(ret, context['tables'], request.POST):
_create_return_data_from_post_data(ret, context['tables'], request.POST)
self._set_submitted(ret)
return redirect('home')
else:
for table in context['tables']:
table['data'] = _get_validated_rows_from_post(ret, table.get('name'), request.POST)
if len(table['data']) == 0:
messages.warning(request,
"You must enter data for {} or submit a Nil Return".format(table.get('name')))
else:
messages.error(request,
"Your return contains some errors. See below.")
elif 'nil' in request.POST:
form = NilReturnForm(request.POST)
if form.is_valid():
ret.nil_return = True
ret.comments = form.cleaned_data['comments']
self._set_submitted(ret)
return redirect('home')
return render(request, self.template_name, context)
