diff --git a/server.py b/server.py index f7b8e13..b4a3768 100644 --- a/server.py +++ b/server.py @@ -1,7 +1,7 @@ -import argparse import subprocess +import re +import argparse -# Server IPMI details IPMIHOST = "192.168.1.223" IPMIUSER = "root" IPMIPASS = "" @@ -33,16 +33,52 @@ def disable_dynamic_fan_control(): print("Disabling dynamic fan control...") execute_ipmi_command("raw 0x30 0x30 0x01 0x00") + +def get_server_temperature(): + print("Getting server temperature...") + command = f"ipmitool -I lanplus -H {IPMIHOST} -U {IPMIUSER} -P \"{IPMIPASS}\" sdr type temperature" + 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 temperature: {temp} degrees C") + warning_threshold = 75 + critical_threshold = 90 + + # Set fan speed based on temperature ranges + if temp >= critical_threshold: + print("Temperature is critical, setting fan speed to 100%.") + set_fan_speed(100) + elif temp >= warning_threshold: + print("Temperature is high, setting fan speed to 80%.") + set_fan_speed(80) + elif temp >= 40: + print("Temperature is moderate, setting fan speed to 60%.") + set_fan_speed(60) + elif temp >= 30: + print("Temperature is low, setting fan speed to 50%.") + set_fan_speed(50) + else: + print("Temperature is very low, setting fan speed to 40%.") + set_fan_speed(40) + # 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]): +if not any([args.power, args.fan, args.dynamic, args.temp]): parser.print_help() exit() @@ -55,7 +91,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()