Merge pull request #14 from brettt89/patch-2

Userforms Bulk Export Button - Fix default submission fields
silverstripe · Jun 13, 2016 · 96c3df7 · 96c3df7
2 parents 52a670b + 92443bc
commit 96c3df7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/code/extensions/UserFormUseQueuedExportExtension.php b/code/extensions/UserFormUseQueuedExportExtension.php
@@ -8,7 +8,13 @@ function updateCMSFields(FieldList $fields) {
         $gridField = $fields->fieldByName('Root.Submissions.Submissions');
 
         $config = $gridField->getConfig();
+        $oldExportButton = $config->getComponentByType('GridFieldExportButton');
+        $config->addComponent($newExportButton = new GridFieldQueuedExportButton('buttons-after-left'));
+
+        // Set Header and Export columns on new Export Button
+        $newExportButton->setCsvHasHeader($oldExportButton->getCsvHasHeader()); 
+        $newExportButton->setExportColumns($oldExportButton->getExportColumns());
+
         $config->removeComponentsByType('GridFieldExportButton');
-        $config->addComponent(new GridFieldQueuedExportButton('buttons-after-left'));
     }
 }