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:
Loic Guegan 2023-10-29 09:37:06 +01:00
parent 38019cabe8
commit e34fc73d55
2 changed files with 15 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
clusterman/commands/node.py
View file

@ -1,5 +1,6 @@
import os, json, time, re, sys, subprocess import os, json, time, re, sys, subprocess
from clusterman.config import CONF from clusterman.config import CONF
from clusterman.utils import ssh_exec
def get_node_list(): def get_node_list():
nodes_path=CONF.NODE_FILE nodes_path=CONF.NODE_FILE
@ -84,15 +85,8 @@ def check(timeout):
print("Error: Some of your nodes are not reachable") print("Error: Some of your nodes are not reachable")
def exec(command, group=None): def exec(command, group=None):
user="root" if len(CONF["ssh"]["user"]) <= 0 else CONF["ssh"]["user"]
key_path=CONF["ssh"]["key_path"]
nodes=get_node_list() if group is None else get_node_in_group(group) nodes=get_node_list() if group is None else get_node_in_group(group)
for ip in nodes: for ip in nodes:
print("----- Node {} -----".format(ip)) print("----- Node {} -----".format(ip))
if len(key_path)>0: print(ssh_exec(ip," ".join(command)))
output=subprocess.check_output(["ssh","-o", "StrictHostKeyChecking=no", "-o", "PasswordAuthentication=no", "-i", CONF["ssh"]["key_path"],"{}@{}".format(user,ip), " ".join(command)])
print(output.decode("utf-8"),end="")
else:
output=subprocess.check_output(["ssh","-o", "StrictHostKeyChecking=no", "-o", "PasswordAuthentication=no", "{}@{}".format(user,ip), " ".join(command)])
print(output.decode("utf-8"),end="")

13
clusterman/utils.py Normal file
View file

@ -0,0 +1,13 @@
import subprocess
from clusterman.config import CONF
def ssh_exec(host,command,use_key=True):
user="root" if len(CONF["ssh"]["user"]) <= 0 else CONF["ssh"]["user"]
key_path=CONF["ssh"]["key_path"]
if use_key:
output=subprocess.check_output(["ssh","-o", "StrictHostKeyChecking=no", "-o", "PasswordAuthentication=no", "-i", CONF["ssh"]["key_path"],"{}@{}".format(user,ip), command])
return output.decode("utf-8")
else:
output=subprocess.check_output(["ssh","-o", "StrictHostKeyChecking=no", "{}@{}".format(user,ip), command])
return output.decode("utf-8")