vassili/seitime-frappe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

fix test case

Browse source
This commit is contained in:
Achilles Rasquinha 2018-03-23 15:34:46 +05:30
parent 5debf79e5a
commit c6be3cbdd7
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
frappe/core/doctype/data_export/exporter.py
View file

@ -29,7 +29,7 @@ def export_data(doctype=None, parent_doctype=None, all_doctypes=True, with_data=
select_columns=select_columns, file_type=file_type, template=template, filters=filters)
exporter.build_response()
class DataExporter():
class DataExporter:
def __init__(self, doctype=None, parent_doctype=None, all_doctypes=True, with_data=False,
select_columns=None, file_type='CSV', template=False, filters=None):
self.doctype = doctype
@ -98,10 +98,9 @@ class DataExporter():
self.add_data()
if self.with_data and not self.data:
frappe.respond_as_web_page(_('No Data'), _('There is no data to be exported'), indicator_color='orange')
return
if self.file_type == 'Excel':
return self.build_response_as_excel()
self.build_response_as_excel()
else:
# write out response as a type csv
frappe.response['result'] = cstr(self.writer.getvalue())
@ -323,7 +322,7 @@ class DataExporter():
def build_response_as_excel(self):
filename = frappe.generate_hash("", 10)
with open(filename, 'wb') as f:
f.write(cstr(self.writer.getvalue()).encode("utf-8"))
f.write(cstr(self.writer.getvalue()))
f = open(filename)
reader = csv.reader(f)