Added monitor script as a bonus
This commit is contained in:
47
monitor.py
Executable file
47
monitor.py
Executable file
@@ -0,0 +1,47 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import paramiko
|
||||
import threading
|
||||
import yaml
|
||||
|
||||
class taskThread(threading.Thread):
|
||||
def __init__(self, host, keys):
|
||||
threading.Thread.__init__(self)
|
||||
self.host = host
|
||||
self.keys = keys
|
||||
def run(self):
|
||||
updateKeys(self.host, self.keys)
|
||||
|
||||
def readConfig():
|
||||
with open('config.yaml', 'r') as stream:
|
||||
return yaml.safe_load(stream)
|
||||
|
||||
def updateKeys(host, keys):
|
||||
try:
|
||||
client = paramiko.SSHClient()
|
||||
client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
|
||||
client.connect(host, username = 'root')
|
||||
stdin, stdout, stderr = client.exec_command('uptime')
|
||||
stdin.close()
|
||||
print('✅ ' + host + ': ' + stdout.read().decode('utf-8').strip().split('load average: ', 2)[1])
|
||||
client.close()
|
||||
except Exception:
|
||||
print('❌ ' + host)
|
||||
|
||||
def main():
|
||||
config = readConfig()
|
||||
|
||||
keys = []
|
||||
|
||||
for key in config['keys']:
|
||||
keys.append(key['key'])
|
||||
|
||||
for host in config['hosts']:
|
||||
try:
|
||||
thread = taskThread(host, keys)
|
||||
thread.start()
|
||||
except:
|
||||
print('❌ ' + host)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user