ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-03 17:23:42 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix Notebook data export to excel/csv not including headers (#11491)

Browse Source 
* added includeHeaders param when getting rows

* fix spacing

* fix syntax and use of map
This commit is contained in:
Lucy Zhang
2020-07-27 09:08:55 -07:00
committed by GitHub
parent 4cf7f9ed66
commit 807d03725f
2 changed files with 21 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/sql/workbench/contrib/notebook/browser/outputs/gridOutput.component.ts
View File

@@ -325,7 +325,7 @@ class DataResourceDataProvider implements IGridDataProvider {
maxRow = singleSelection.toRow + 1;
columns = columns.slice(singleSelection.fromCell, singleSelection.toCell + 1);
}
let getRows: ((index: number, rowCount: number) => ICellValue[][]) = (index, rowCount) => {
let getRows: ((index: number, includeHeaders: boolean, rowCount: number) => ICellValue[][]) = (index, includeHeaders, rowCount) => {
// Offset for selections by adding the selection startRow to the index
index = index + minRow;
if (rowLength === 0 || index < 0 || index >= maxRow) {
@@ -335,12 +335,25 @@ class DataResourceDataProvider implements IGridDataProvider {
if (endIndex > maxRow) {
endIndex = maxRow;
}
let result = this.rows.slice(index, endIndex).map(row => {
let result: ICellValue[][] = [];
if (includeHeaders) {
result.push(columns.map(col => {
let headerData: azdata.DbCellValue;
headerData = {
displayValue: col.columnName,
isNull: false,
invariantCultureDisplayValue: col.columnName
};
return headerData;
}));
}
result = result.concat(this.rows.slice(index, endIndex).map(row => {
if (this.isSelected(singleSelection)) {
return row.slice(singleSelection.fromCell, singleSelection.toCell + 1);
} else {
return row;
}
return row;
});
}));
return result;
};
@@ -348,7 +361,7 @@ class DataResourceDataProvider implements IGridDataProvider {
saveFormat: format,
columns: columns,
filePath: filePath.fsPath,
getRowRange: (rowStart, numberOfRows) => getRows(rowStart, numberOfRows),
getRowRange: (rowStart, includeHeaders, numberOfRows) => getRows(rowStart, includeHeaders, numberOfRows),
rowCount: rowLength
});
return this._serializationService.serializeResults(serializeRequestParams);