loic/clusterman
1
0
Fork 0
mirror of https://gitlab.com/manzerbredes/clusterman.git synced 2025-04-06 03:56:27 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Minor changes

Browse source
This commit is contained in:
Loïc Guégan 2023-10-27 14:18:57 +02:00
parent eb41b0b49a
commit 28c1833601
2 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
clusterman/__main__.py
View file

@ -20,6 +20,7 @@ def main():
node_cmd_list.add_argument("-g", "--group" ,help="Group to list") node_cmd_list.add_argument("-g", "--group" ,help="Group to list")
# Exec # Exec
node_cmd_list=node_subparsers.add_parser("exec") node_cmd_list=node_subparsers.add_parser("exec")
node_cmd_list.add_argument("-g", "--group" ,help="Group to run the command on")
node_cmd_list.add_argument("cmd",help="Command to run",nargs=argparse.REMAINDER) node_cmd_list.add_argument("cmd",help="Command to run",nargs=argparse.REMAINDER)
##### Frontend commands ##### ##### Frontend commands #####
@ -58,7 +59,10 @@ def main():
else: else:
node.ls() node.ls()
elif args.command == "exec": elif args.command == "exec":
node.exec(args.cmd) if args.group:
node.exec(args.cmd,args.group)
else:
node.exec(args.cmd)
else: else:
target_node.print_help(sys.stderr) target_node.print_help(sys.stderr)
sys.exit(1) sys.exit(1)

5
clusterman/commands/node.py
View file

@ -83,10 +83,11 @@ def check(timeout):
else: else:
print("Error: Some of your nodes are not reachable") print("Error: Some of your nodes are not reachable")
def exec(command): def exec(command, group=None):
user="root" if len(CONF["ssh"]["user"]) <= 0 else CONF["ssh"]["user"] user="root" if len(CONF["ssh"]["user"]) <= 0 else CONF["ssh"]["user"]
key_path=CONF["ssh"]["key_path"] key_path=CONF["ssh"]["key_path"]
for ip in get_node_list(): nodes=get_node_list() if group is None else get_node_in_group(group)
for ip in nodes:
print("----- Node {} -----".format(ip)) print("----- Node {} -----".format(ip))
if len(key_path)>0: if len(key_path)>0:
output=subprocess.check_output(["ssh","-o", "StrictHostKeyChecking=no", "-i", CONF["ssh"]["key_path"],"{}@{}".format(user,ip), " ".join(command)]) output=subprocess.check_output(["ssh","-o", "StrictHostKeyChecking=no", "-i", CONF["ssh"]["key_path"],"{}@{}".format(user,ip), " ".join(command)])