From 6ba6dc8f74fd0bc6b73dca5e3fc6b174c4d6e1ca Mon Sep 17 00:00:00 2001
From: Toby <mail@tobias-schneider.io>
Date: Wed, 20 Sep 2023 00:47:14 +0200
Subject: [PATCH] clean up

---
 monitor2.py | 26 +++++++++++---------------
 1 file changed, 11 insertions(+), 15 deletions(-)

diff --git a/monitor2.py b/monitor2.py
index 94e30fe..3d0797f 100755
--- a/monitor2.py
+++ b/monitor2.py
@@ -5,8 +5,6 @@ import paramiko
 import threading
 import yaml
 
-out_buf = []
-
 def string_to_float(data):
     return float(data.strip().split(' ', 2)[0].strip())
 
@@ -42,7 +40,7 @@ class Host(threading.Thread):
         self.host_length = host_length
         self.idx = idx
         self.status = 'none'
-        self.done = False
+        self.row = ''
 
     def render_status(self):
         if self.status == 'none':
@@ -68,22 +66,20 @@ class Host(threading.Thread):
         return 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) + '%)'
 
     def render_row(self):
-        return self.render_status() + self.host.ljust(self.host_length + 3) + self.render_top()
+        self.row = self.render_status() + self.host.ljust(self.host_length + 3) + self.render_top()
 
     def run(self):
         try:
             self.status = 'none'
-            out_buf[self.idx] = self.render_row()
+            self.render_row()
             self.client = paramiko.SSHClient()
             self.client.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
             self.client.connect(self.host, port = self.port, username = 'root', timeout = 1)
             self.status = 'ok'
-            out_buf[self.idx] = self.render_row()
         except Exception as e:
             #print(e)
             self.status = 'fail'
-            out_buf[self.idx] = self.render_row()
-        self.done = True
+        self.render_row()
 
 def read_config():
     with open('config.yaml', 'r') as stream:
@@ -124,7 +120,6 @@ def main():
                 continue
             if not 'port' in host:
                 host['port'] = 22
-            out_buf.append('')
             hosts.append(Host(host['host'], host['port'] or 22, host_length, i))
 
     for host in hosts:
@@ -135,13 +130,8 @@ def main():
 
     while True:
         screen = []
-        for row in out_buf:
-            screen.append(row)
-
         for host in hosts:
-            if not host.done:
-              break
-            end = True
+            screen.append(host.row)
 
         if first:
             first = False
@@ -153,6 +143,12 @@ def main():
         if end:
             break
 
+        for host in hosts:
+            if host.is_alive():
+                break
+            end = True
+            break
+
         time.sleep(0.1)
 
 if __name__ == '__main__':