From 38228ebc9b5730069325aa8a2e0eaa13be33ce0a Mon Sep 17 00:00:00 2001 From: Jarrod Johnson Date: Tue, 4 Sep 2018 11:09:26 -0400 Subject: [PATCH] Fix the prompting code changes --- confluent_client/bin/nodegroupattrib | 1 + confluent_client/confluent/client.py | 10 ++++------ 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/confluent_client/bin/nodegroupattrib b/confluent_client/bin/nodegroupattrib index b066b284..bba961ed 100755 --- a/confluent_client/bin/nodegroupattrib +++ b/confluent_client/bin/nodegroupattrib @@ -17,6 +17,7 @@ __author__ = 'alin37' +from getpass import getpass import optparse import os import signal diff --git a/confluent_client/confluent/client.py b/confluent_client/confluent/client.py index b8c6c6f9..bc5cd4fe 100644 --- a/confluent_client/confluent/client.py +++ b/confluent_client/confluent/client.py @@ -530,13 +530,11 @@ def updateattrib(session, updateargs, nodetype, noderange, options, dictassign=N elif dictassign: for key in dictassign: if nodetype == 'nodegroups': - exitcode = session.simp_nodegroups_command(noderange, - 'attributes/all', - dictassign[key], key) + exitcode = session.simple_nodegroups_command( + noderange, 'attributes/all', dictassign[key], key) else: - exitcode = session.simp_nodegroups_command(noderange, - 'attributes/all', - dictassign[key], key) + exitcode = session.simple_noderange_command( + noderange, 'attributes/all', dictassign[key], key) else: if "=" in updateargs[1]: try: