tsndr/sync-ssh-keys
1
1
Fork 0
Code Pull Requests Activity

updated for new config syntax

Browse Source
This commit is contained in:
Toby
2023-09-19 22:14:09 +02:00
parent 27fa711778
commit e964f74f58
1 changed files with 44 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
monitor.py
View File

@@ -1,17 +1,19 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import time
import paramiko import paramiko
import threading import threading
import yaml import yaml
class task_thread(threading.Thread): class task_thread(threading.Thread):
def __init__(self, host, user, host_length): def __init__(self, host, port, user, keys):
threading.Thread.__init__(self) threading.Thread.__init__(self)
self.host = host self.host = host
self.user = user self.port = port
self.host_length = host_length self.user = user
def run(self): self.keys = keys
load_metrics(self.host, self.user, self.host_length) def run(self):
load_metrics(self.host, self.port, self.user, self.keys)
def read_config(): def read_config():
with open('config.yaml', 'r') as stream: with open('config.yaml', 'r') as stream:
@@ -44,11 +46,11 @@ def parse_top_string(data):
return load, cpu_percent, ram_total, ram_free return load, cpu_percent, ram_total, ram_free
def load_metrics(host, user, host_length): def load_metrics(host, port, user, host_length):
try: try:
client = paramiko.SSHClient() client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy()) client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
client.connect(host, username = user, timeout = 1) client.connect(host, port = port, username = user, timeout = 1)
stdin, stdout, stderr = client.exec_command('top -bn1 | grep "^top\\|^%Cpu\\|^.iB Mem"') stdin, stdout, stderr = client.exec_command('top -bn1 | grep "^top\\|^%Cpu\\|^.iB Mem"')
stdin.close() stdin.close()
@@ -59,10 +61,11 @@ def load_metrics(host, user, host_length):
load, cpu_percent, ram_total, ram_free = parse_top_string(data) load, cpu_percent, ram_total, ram_free = parse_top_string(data)
ram_used = ram_total - ram_free ram_used = ram_total - ram_free
print(('✅ ' + user + '@' + host).ljust(host_length + 5) + load + ' (' + '{:3.1f}'.format(cpu_percent).rjust(5) + '%) ' + '{:3.1f}'.format(ram_used).rjust(5) + ' / ' + '{:3.1f}'.format(ram_total).rjust(5) + ' GiB (' + '{:3.1f}'.format(ram_used / ram_total * 100).rjust(5) + '%)') print(('✅ ' + host).ljust(host_length + 5) + load + ' (' + '{:3.1f}'.format(cpu_percent).rjust(5) + '%) ' + '{:3.1f}'.format(ram_used).rjust(5) + ' / ' + '{:3.1f}'.format(ram_total).rjust(5) + ' GiB (' + '{:3.1f}'.format(ram_used / ram_total * 100).rjust(5) + '%)')
except: except:
print('❌ ' + user + '@' + host) time.sleep(1)
print('❌ ' + host)
def main(): def main():
config = read_config() config = read_config()
@@ -71,18 +74,39 @@ def main():
for host in config['hosts']: for host in config['hosts']:
for user in host['users'].keys(): for user in host['users'].keys():
if len(user) + len(host['host']) > host_length: if len(user) + len(host['host']) > host_length:
host_length = len(user) + len(host['host']) host_length = len(host['host'])
print('Host'.center(host_length + 3) + ' ' + 'Load'.center(25) + ' ' + 'Ram Usage'.center(26)) print('Host'.center(host_length + 3) + ' ' + 'Load'.center(25) + ' ' + 'Ram Usage'.center(26))
for host in config['hosts']: for host in config['hosts']:
if 'root' not in host['users'].keys(): for user_name, user_data in host['users'].items():
continue if user_name != 'root':
try: continue
thread = task_thread(host['host'], 'root', host_length) host_keys = []
thread.start() if 'groups' in user_data.keys():
except: for group in user_data['groups']:
print('❌ ' + user + '@' + host['host']) if group not in config['groups'].keys():
print('WARNING: Key-group "' + group + '" not found!')
continue
for key_name in config['groups'][group]:
host_keys.append(config['keys'][key_name])
if 'keys' in user_data.keys():
for key_name in user_data['keys']:
if key_name not in config['keys'].keys():
print('WARNING: Key "' + key_name + '" not found!')
continue
host_keys.append(config['keys'][key_name])
host_keys = list(set(host_keys)) # Filter duplicates
if not host_keys:
continue
if not 'port' in host:
host['port'] = 22
try:
thread = task_thread(host['host'], host['port'], user_name, host_length)
thread.start()
except:
time.sleep(1)
print('❌ ' + host['host'])
if __name__ == '__main__': if __name__ == '__main__':
main() main()