def export(self, object_list, name, filename_prefix, data=None, filter_description=None): xls = XLSExporter(name) xls.add_from_object_list(object_list, data, filter_description=filter_description) temporary = xls.save(filename_prefix) self.export_temporary(temporary)
def test_export_from_object_list(self): fruits = ObjectList([Column('name', data_type=str), Column('price', data_type=int)]) for name, price in [('Apple', 4), ('Pineapple', 2), ('Kiwi', 8), ('Banana', 3), ('Melon', 5)]: fruits.append(Fruit(name, price)) ofx = XLSExporter() ofx.add_from_object_list(fruits) try: temp_file = ofx.save() data = open(temp_file.name).read() # We should use xlrd to 're-open' the spreadsheet and parse its content. self.assertTrue(len(data) > 0) finally: temp_file.close() os.unlink(temp_file.name)
def test_export_from_object_list(self): from kiwi.ui.objectlist import ObjectList, Column fruits = ObjectList( [Column('name', data_type=str), Column('price', data_type=int)]) for name, price in [('Apple', 4), ('Pineapple', 2), ('Kiwi', 8), ('Banana', 3), ('Melon', 5)]: fruits.append(Fruit(name, price)) ofx = XLSExporter() ofx.add_from_object_list(fruits) try: temp_file = ofx.save() with open(temp_file.name, 'rb') as f: data = f.read() # We should use xlrd to 're-open' the spreadsheet and parse its content. self.assertTrue(len(data) > 0) finally: temp_file.close() os.unlink(temp_file.name)
def test_xls_exporter_write_one(): exporter = XLSExporter() exporter._ws = mock.Mock() style = xlwt.XFStyle() i = 10 exporter._write_one(i, 'foo', style) exporter._ws.write.assert_called_once_with(1, i, 'foo', style)
def testExportFromObjectList(self): fruits = ObjectList( [Column('name', data_type=str), Column('price', data_type=int)]) for name, price in [('Apple', 4), ('Pineapple', 2), ('Kiwi', 8), ('Banana', 3), ('Melon', 5)]: fruits.append(Fruit(name, price)) ofx = XLSExporter() ofx.add_from_object_list(fruits) temp_file = ofx.save() data = open(temp_file.name).read() # We should use xlrd to 're-open' the spreadsheet and parse its content. self.assertTrue(len(data) > 0)
def export(self, object_list, name, filename_prefix): xls = XLSExporter(name) xls.add_from_object_list(object_list) temporary = xls.save(filename_prefix) self.export_temporary(temporary)
def test_xls_exporter_convert_one(data, expected_value): exporter = XLSExporter() assert exporter._convert_one(data) == expected_value