From 872fff004be1df4f81fb1ae2c82d5ffb1a8a79ca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Michael=20H=C3=A4rtl?= <haertl.mike@gmail.com>
Date: Fri, 3 Aug 2018 09:12:22 +0200
Subject: [PATCH] Issue #26 Fix error for non-DB attributes

---
 src/ActiveExcelSheet.php | 48 ++++++++++++++++++++++++----------------
 1 file changed, 29 insertions(+), 19 deletions(-)

diff --git a/src/ActiveExcelSheet.php b/src/ActiveExcelSheet.php
index df71c79..911b012 100644
--- a/src/ActiveExcelSheet.php
+++ b/src/ActiveExcelSheet.php
@@ -27,7 +27,7 @@ class ActiveExcelSheet extends ExcelSheet
 
     protected $_query;
     protected $_attributes;
-    protected $_columnTypes;
+    protected $_columnSchemas;
     protected $_modelInstance;
 
     /**
@@ -131,9 +131,12 @@ public function getFormats()
         if ($this->_formats === null) {
             $this->_formats = [];
             $attrs = $this->normalizeIndex($this->getAttributes());
-            $types = $this->normalizeIndex($this->getColumnTypes());
+            $schemas = $this->normalizeIndex($this->getColumnSchemas());
             foreach ($attrs as $c => $attr) {
-                switch ($types[$c]->type) {
+                if (!isset($schemas[$c])) {
+                    continue;
+                }
+                switch ($schemas[$c]->type) {
                     case 'date':
                         $this->_formats[$c] = $this->dateFormat;
                         break;
@@ -141,8 +144,8 @@ public function getFormats()
                         $this->_formats[$c] = $this->dateTimeFormat;
                         break;
                     case 'decimal':
-                        $decimals = str_pad('#,', $types[$c]->scale, '#');
-                        $zeroPad = str_pad('0.', $types[$c]->scale, '0');
+                        $decimals = str_pad('#,', $schemas[$c]->scale, '#');
+                        $zeroPad = str_pad('0.', $schemas[$c]->scale, '0');
                         $this->_formats[$c] = $decimals.$zeroPad;
                         break;
                 }
@@ -181,9 +184,12 @@ public function getFormatters()
         if ($this->_formatters === null) {
             $this->_formatters = [];
             $attrs = $this->normalizeIndex($this->getAttributes());
-            $types = $this->normalizeIndex($this->getColumnTypes());
+            $schemas = $this->normalizeIndex($this->getColumnSchemas());
             foreach ($attrs as $c => $attr) {
-                switch ($types[$c]->type) {
+                if (!isset($schemas[$c])) {
+                    continue;
+                }
+                switch ($schemas[$c]->type) {
                     case 'date':
                         $this->_formatters[$c] = function ($v) {
                             if (empty($v)) {
@@ -271,18 +277,20 @@ protected static function getRelatedModelInstance($model, $name)
     }
 
     /**
-     * @return yii\db\ColumnSchema[] the DB column types `ColumnSchema::$type`
-     * indexed by 0-based column index
+     * @return yii\db\ColumnSchema[] the DB column schemas indexed by 0-based
+     * column index. This only includes columns for which a DB schema exists.
      */
-    protected function getColumnTypes()
+    protected function getColumnSchemas()
     {
-        if ($this->_columnTypes === null) {
+        if ($this->_columnSchemas === null) {
             $model = $this->getModelInstance();
-            $this->_columnTypes = array_map(function ($attr) use ($model) {
-                return self::getType($model, $attr);
+            $schemas = array_map(function ($attr) use ($model) {
+                return self::getSchema($model, $attr);
             }, $this->getAttributes());
+            // Filter out null values
+            $this->_columnSchemas = array_filter($schemas);
         }
-        return $this->_columnTypes;
+        return $this->_columnSchemas;
     }
 
     /**
@@ -350,19 +358,21 @@ protected function toExcelTime($value)
      * @param mixed $isRelation whether the name specifies a relation, in which
      * case an `ActiveRecord` is returned. Default is `false`, which returns a
      * `ColumnSchema`.
-     * @return yii\db\ColumnSchema|yii\db\ActiveRecord the type instance of the
-     * attribute
+     * @return yii\db\ColumnSchema|yii\db\ActiveRecord|null the type instance
+     * of the attribute or `null` if the attribute is not a DB column (e.g.
+     * public property or defined by getter)
      */
-    public static function getType($model, $attribute, $isRelation = false)
+    public static function getSchema($model, $attribute, $isRelation = false)
     {
         if (($pos = strrpos($attribute, '.')) !== false) {
-            $model = self::getType($model, substr($attribute, 0, $pos), true);
+            $model = self::getSchema($model, substr($attribute, 0, $pos), true);
             $attribute = substr($attribute, $pos + 1);
         }
         if ($isRelation) {
             return self::getRelatedModelInstance($model, $attribute);
         } else {
-            return $model->getTableSchema()->columns[$attribute];
+            $columnSchemas = $model->getTableSchema()->columns;
+            return isset($columnSchemas[$attribute]) ? $columnSchemas[$attribute] : null;
         }
     }
 }