feb git sync

files/oop_code/HostRedis.py

@@ -0,0 +1,94 @@
+
+import redis
+import subprocess
+import re
+import json
+
+####################################################
+### Redis Functions
+####################################################
+debug_output = False
+r = redis.Redis(host='172.17.0.1', port=6379)
+
+def update_stats_redis():
+    # store the data in vm_list
+    data = get_host_stats(as_json=False)
+    # push data to redis
+    # Publish to the Redis channel that the WS server is listening on
+    r.publish('host_stats', json.dumps(data))
+    if debug_output:
+        print("=== Stats Redis Update ===")
+        print(json.dumps(data, indent=2))
+    return True
+
+####################################################
+### Host Stats Function
+####################################################
+def get_host_stats(as_json=False):
+    total_memory_command = "free -h | grep 'Mem:' | awk '{print $2}'"
+    total_memory = run_shell(total_memory_command, zero_only=True)
+    used_memory_command = "free -h | grep 'Mem:' | awk '{print $3}'"
+    used_memory = run_shell(used_memory_command, zero_only=True)
+    free_memory_command = "free -h | grep 'Mem:' | awk '{print $4}'"
+    free_memory = run_shell(free_memory_command, zero_only=True)
+    cpu_load_command = "uptime | grep -oP '(?<=age: ).*'"
+    cpu_load = run_shell(cpu_load_command, zero_only=True)
+    # nano pi command
+    #cpu_temp_command = "sensors | grep 'temp1:' | cut -d+ -f 2 | awk '{print $1}'"
+    cpu_temp_command = "sensors | grep -e Sensor -e Package | cut -d+ -f 2 | awk '{print $1}'"
+    cpu_temp = run_shell(cpu_temp_command, zero_only=True)
+    cpu_temp_stripped = re.sub(r'\u00b0C', '', cpu_temp)
+    cpu_temp_fixed = f"{cpu_temp_stripped} C"
+    ip_address_command = "ip -o -4 ad | grep -v -e docker -e 127.0.0.1 | awk '{print $2\": \" $4}'"
+    ip_addresses = run_shell(ip_address_command, zero_only=True)
+    time_now_command = "date +%r"
+    time_now = run_shell(time_now_command, zero_only=True)
+    # Redis stores in this order, or at least the html renders it in this order
+    stats = [{
+        "memory_total": total_memory,
+        "memory_used": used_memory,
+        "memory_free": free_memory,
+        "cpu_load": cpu_load, 
+        "cpu_temp": cpu_temp_fixed,
+        "ip_addresses": ip_addresses,
+        "time": time_now
+    }]
+    if check_for_battery():
+        battery_level_command = "acpi | grep Battery | awk {print'$3 \" \" $4'}"
+        battery_level = run_shell(battery_level_command, zero_only=True)
+        stats = [{
+            "memory_total": total_memory,
+            "memory_used": used_memory,
+            "memory_free": free_memory,
+            "cpu_load": cpu_load, 
+            "cpu_temp": cpu_temp_fixed,
+            "ip_addresses": ip_addresses,
+            "battery_level": battery_level,
+            "time": time_now
+        }]        
+    if debug_output:
+        print("=== Current Host Stats ===")
+        print(json.dumps(stats, indent=2))
+    return jsonify(stats) if as_json else stats
+
+def check_for_battery():
+    battery_check_command = "acpi | grep Battery | awk {print'$1'}"
+    battery_check = run_shell(battery_check_command, zero_only=True)
+    if battery_check == 'Battery':
+        return True
+    else:
+        return False
+
+# subroutine to run a command, return stdout as array unless zero_only then return [0]
+def run_shell(cmd, zero_only=False):
+    # Run the command and capture the output
+    result = subprocess.run(cmd, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)    
+    # Decode the byte output to a string
+    output = result.stdout.decode('utf-8')
+    # Split the output into lines and store it in an array
+    output_lines = [line for line in output.split('\n') if line]
+    # Return result
+    try:
+        return output_lines[0] if zero_only else output_lines
+    except:
+        return output_lines