From e4434babcbfbdf92ea5db86bd1c3b58dae54835b Mon Sep 17 00:00:00 2001
From: Anna Schumaker <Anna.Schumaker@Netapp.com>
Date: Wed, 19 Jul 2023 12:01:58 -0400
Subject: [PATCH] rdma.zsh: Various Updates

* Use $COLORS from common.zsh
* Fix colored output when run from another script

Signed-off-by: Anna Schumaker <Anna.Schumaker@Netapp.com>
---
 rdma.zsh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/rdma.zsh b/rdma.zsh
index 486703b..2eab549 100755
--- a/rdma.zsh
+++ b/rdma.zsh
@@ -1,4 +1,5 @@
 #!/bin/zsh
+source common.zsh
 
 if [ "$#" -lt 2 ]; then
 	echo "Usage: $0 {rxe,siw,info,start,status,stop,off} {remote host,...}"
@@ -55,7 +56,7 @@ rdma_stop() {
 }
 
 IFS=$'\n'
-tags=($(python /home/anna/bin/colors/vm.py ${@:2}))
+tags=($(FORCE_COLOR=1 python $COLORS/vm.py ${@:2}))
 for vm in "${@:2}"; do
 	index=${@[(ie)$vm]}
 	tag=${tags[index-1]}