Added payee transaction handling to TransactionToFinishModel

barion · Sep 21, 2016 · 8fde837 · 8fde837
1 parent a9c4cd5
commit 8fde837
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/library/models/TransactionToFinishModel.php b/library/models/TransactionToFinishModel.php
@@ -50,4 +50,23 @@ public function AddItems($items)
             }
         }
     }
+
+    public function AddPayeeTransaction(PayeeTransactionModel $model)
+    {
+        if ($this->PayeeTransactions == null) {
+            $this->PayeeTransactions = array();
+        }
+        array_push($this->PayeeTransactions, $model);
+    }
+
+    public function AddPayeeTransactions($transactions)
+    {
+        if (!empty($transactions)) {
+            foreach ($transactions as $transaction) {
+                if ($transaction instanceof PayeeTransactionModel) {
+                    $this->AddPayeeTransaction($transaction);
+                }
+            }
+        }
+    }
 }