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: