tonyair/silverstripe-framework
1
0
Fork 0
mirror of https://github.com/silverstripe/silverstripe-framework synced 2024-10-22 14:05:37 +02:00
Code Issues Projects Releases Wiki Activity

Use columns from gridfield for export

Browse Source
This commit is contained in:
Anton Smith 2016-07-19 08:26:07 +12:00
parent a73f08bd88
commit 4c856c4377
1 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
forms/gridfield/GridFieldExportButton.php
View File

@ -92,6 +92,25 @@ class GridFieldExportButton implements GridField_HTMLProvider, GridField_ActionP
} }
} }
/**
* Return the columns to export
*
* @param GridField $gridField
*
* @return array
*/
protected function getExportColumnsForGridField(GridField $gridField) {
if($this->exportColumns) {
$exportColumns = $this->exportColumns;
} else if($dataCols = $gridField->getConfig()->getComponentByType('GridFieldDataColumns')) {
$exportColumns = $dataCols->getDisplayFields($gridField);
} else {
$exportColumns = singleton($gridField->getModelClass())->summaryFields();
}
return $exportColumns;
}
/** /**
* Generate export fields for CSV. * Generate export fields for CSV.
* *
@ -100,9 +119,7 @@ class GridFieldExportButton implements GridField_HTMLProvider, GridField_ActionP
*/ */
public function generateExportFileData($gridField) { public function generateExportFileData($gridField) {
$separator = $this->csvSeparator; $separator = $this->csvSeparator;
$csvColumns = ($this->exportColumns) $csvColumns = $this->getExportColumnsForGridField($gridField);
? $this->exportColumns
: singleton($gridField->getModelClass())->summaryFields();
$fileData = ''; $fileData = '';
if($this->csvHasHeader) { if($this->csvHasHeader) {