Respect displayFields in GridFieldPrintButton

Provides more coherent and expected default behaviour
silverstripe · Jan 14, 2013 · 80feb8c · 80feb8c
1 parent 703c10a
commit 80feb8c
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/forms/gridfield/GridFieldPrintButton.php b/forms/gridfield/GridFieldPrintButton.php
@@ -91,9 +91,14 @@ public function handlePrint($gridField, $request = null) {
 	 * Export core.
  	 */
 	public function generatePrintData($gridField) {
-		$printColumns = ($this->printColumns)
-			? $this->printColumns
-			: singleton($gridField->getModelClass())->summaryFields();
+		if($this->printColumns) {
+			$printColumns = $this->printColumns;
+		} else if($dataCols = $gridField->getConfig()->getComponentByType('GridFieldDataColumns')) {
+			$printColumns = $dataCols->getDisplayFields($gridField);
+		} else {
+			$printColumns = singleton($gridField->getModelClass())->summaryFields();
+		}
+
 		$header = null;
 		if($this->printHasHeader){
 			$header = new ArrayList();