From 5215334d9b8ae5196bd0e418979a43c056b5fb83 Mon Sep 17 00:00:00 2001 From: Tristan Smith Date: Thu, 21 Mar 2024 23:10:52 -0400 Subject: [PATCH] Fix fan speed control and add temperature-based fan speed adjustment --- server.py | 40 ++++++++++++++++++++++++++++++++-------- 1 file changed, 32 insertions(+), 8 deletions(-) diff --git a/server.py b/server.py index f7b8e13..b3638da 100644 --- a/server.py +++ b/server.py @@ -1,5 +1,6 @@ import argparse import subprocess +import re # Server IPMI details IPMIHOST = "192.168.1.223" @@ -8,7 +9,7 @@ IPMIPASS = "" def execute_ipmi_command(command): full_command = f"ipmitool -I lanplus -H {IPMIHOST} -U {IPMIUSER} -P \"{IPMIPASS}\" {command}" - result = subprocess.run(full_command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE) + result = subprocess.run(full_command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) print(result.stdout.decode().strip()) if result.stderr: print(result.stderr.decode().strip()) @@ -23,7 +24,7 @@ def power_off(): def set_fan_speed(speed): print(f"Setting fan speed to {speed}%...") - execute_ipmi_command(f"raw 0x30 0x30 0x02 0xff {hex(speed)}") + execute_ipmi_command(f"raw 0x30 0x30 0x02 0xff {speed}") def enable_dynamic_fan_control(): print("Enabling dynamic fan control...") @@ -33,19 +34,39 @@ def disable_dynamic_fan_control(): print("Disabling dynamic fan control...") execute_ipmi_command("raw 0x30 0x30 0x01 0x00") +def get_server_temperature(): + command = "ipmitool -I lanplus -H {0} -U {1} -P \"{2}\" sdr type temperature".format(IPMIHOST, IPMIUSER, IPMIPASS) + result = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True) + temperature_readings = result.stdout + temperatures = re.findall(r'\| (\d+) degrees C', temperature_readings) + temperatures = [int(temp) for temp in temperatures] + average_temp = sum(temperatures) / len(temperatures) if temperatures else 0 + return average_temp + +def adjust_fan_speed_based_on_temp(): + temp = get_server_temperature() + print(f"Current average temperature: {temp}C") + warning_threshold = 75 + critical_threshold = 90 + if temp >= critical_threshold: + print("Temperature is critical, setting fan speed to 100%.") + set_fan_speed(100) + elif temp >= warning_threshold: + print("Temperature is nearing warning level, setting fan speed to 75%.") + set_fan_speed(75) + else: + print("Temperature is normal, setting fan speed to dynamic control.") + enable_dynamic_fan_control() + # Argument Parsing parser = argparse.ArgumentParser(description='Server Management Script') parser.add_argument('-p', '--power', choices=['on', 'off'], help='Power on or off the server') parser.add_argument('-f', '--fan', type=int, choices=range(0, 101), metavar="[0-100]", help='Set fan speed percentage') +parser.add_argument('-t', '--temp', action='store_true', help='Adjust fan speed based on temperature') parser.add_argument('-d', '--dynamic', choices=['on', 'off'], help='Toggle dynamic fan control') args = parser.parse_args() -# Check for no action -if not any([args.power, args.fan, args.dynamic]): - parser.print_help() - exit() - # Execute based on arguments if args.power == 'on': power_on() @@ -55,7 +76,10 @@ elif args.power == 'off': if args.fan is not None: set_fan_speed(args.fan) +if args.temp: + adjust_fan_speed_based_on_temp() + if args.dynamic == 'on': enable_dynamic_fan_control() elif args.dynamic == 'off': - disable_dynamic_fan_control() \ No newline at end of file + disable_dynamic_fan_control()