From 846a966a47f153664b7255e432828cad73624c7f Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Fri, 3 Sep 2021 21:20:52 +1000
Subject: [PATCH] Minor fixes for creating sample data

---
 app/Console/Commands/CreateSingleAccount.php | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/app/Console/Commands/CreateSingleAccount.php b/app/Console/Commands/CreateSingleAccount.php
index ca72019603..2d3c38cc5a 100644
--- a/app/Console/Commands/CreateSingleAccount.php
+++ b/app/Console/Commands/CreateSingleAccount.php
@@ -22,6 +22,7 @@ use App\Factory\RecurringInvoiceFactory;
 use App\Factory\SubscriptionFactory;
 use App\Helpers\Invoice\InvoiceSum;
 use App\Jobs\Company\CreateCompanyTaskStatuses;
+use App\Libraries\MultiDB;
 use App\Models\Account;
 use App\Models\Client;
 use App\Models\ClientContact;
@@ -62,7 +63,7 @@ class CreateSingleAccount extends Command
     /**
      * @var string
      */
-    protected $signature = 'ninja:create-single-account {gateway=all}';
+    protected $signature = 'ninja:create-single-account {gateway=all} {--database=db-ninja-01}';
 
     protected $invoice_repo;
 
@@ -89,6 +90,8 @@ class CreateSingleAccount extends Command
      */
     public function handle()
     {
+        MultiDB::setDb($this->option('database'));
+
         $this->info(date('r').' Create Single Sample Account...');
         $this->count = 1;
         $this->gateway = $this->argument('gateway');