From b35463e594cfa8dfcc069963adda8af4f3b2fd97 Mon Sep 17 00:00:00 2001
From: Jarrod Johnson <jjohnson2@lenovo.com>
Date: Wed, 16 Aug 2017 17:33:23 -0400
Subject: [PATCH] Implement argument to nodefirmware to backup

---
 confluent_client/bin/nodefirmware | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/confluent_client/bin/nodefirmware b/confluent_client/bin/nodefirmware
index 243f3271..8cee84f4 100755
--- a/confluent_client/bin/nodefirmware
+++ b/confluent_client/bin/nodefirmware
@@ -66,7 +66,9 @@ def printfirm(node, prefix, data):
 
 
 argparser = optparse.OptionParser(
-    usage="Usage: %prog <noderange> [update <file>]")
+    usage="Usage: %prog <noderange> [update [--backup <file>]")
+argparser.add_option('-b', '--backup', action='store_true',
+                     help='Target a backup bank rather than primary')
 (options, args) = argparser.parse_args()
 upfile = None
 try:
@@ -98,7 +100,10 @@ def update_firmware(session, filename):
     filename = os.path.abspath(filename)
     resource = '/noderange/{0}/inventory/firmware/updates/active'.format(
         noderange)
-    for res in session.create(resource, {'filename': filename}):
+    upargs = {'filename': filename}
+    if options.backup:
+        upargs['bank'] = 'bank'
+    for res in session.create(resource, upargs):
         watchurl = res['created']
         currnode = watchurl.split('/')[1]
         nodeurls[currnode] = '/' + watchurl