65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
|
import os
|
|
import subprocess
|
|
from auto_cpufreq.utils.config import config
|
|
|
|
|
|
def set_battery(value, mode, bat):
|
|
path = f"/sys/class/power_supply/BAT{bat}/charge_{mode}_threshold"
|
|
if os.path.isfile(path):
|
|
subprocess.check_output(
|
|
f"echo {value} | tee {path}", shell=True, text=True)
|
|
else:
|
|
print(f"WARNING: {path} does NOT exist")
|
|
|
|
|
|
def get_threshold_value(mode):
|
|
|
|
conf = config.get_config()
|
|
if conf.has_option("battery", f"{mode}_threshold"):
|
|
return conf["battery"][f"{mode}_threshold"]
|
|
else:
|
|
if mode == "start":
|
|
|
|
return 0
|
|
else:
|
|
return 100
|
|
|
|
|
|
def thinkpad_setup():
|
|
conf = config.get_config()
|
|
|
|
if not conf.has_option("battery", "enable_thresholds"):
|
|
return
|
|
if not conf["battery"]["enable_thresholds"] == "true":
|
|
return
|
|
|
|
if os.path.exists("/sys/class/power_supply/"):
|
|
battery_count = len([name for name in os.listdir(
|
|
"/sys/class/power_supply/") if name.startswith('BAT')])
|
|
|
|
for bat in range(battery_count):
|
|
set_battery(get_threshold_value("start"), "start", bat)
|
|
set_battery(get_threshold_value("stop"), "stop", bat)
|
|
else:
|
|
print("WARNING /sys/class/power_supply/ does NOT esixt")
|
|
|
|
|
|
def thinkpad_print_thresholds():
|
|
battery_count = len([name for name in os.listdir(
|
|
"/sys/class/power_supply/") if name.startswith('BAT')])
|
|
print("\n-------------------------------- Battery Info ---------------------------------\n")
|
|
print(f"battery count = {battery_count}")
|
|
for b in range(battery_count):
|
|
try:
|
|
with open(f'/sys/class/power_supply/BAT{b}/charge_start_threshold', 'r') as f:
|
|
print(f'battery{b} start threshold = {f.read()}', end="")
|
|
f.close()
|
|
|
|
with open(f'/sys/class/power_supply/BAT{b}/charge_stop_threshold', 'r') as f:
|
|
print(f'battery{b} stop threshold = {f.read()}', end="")
|
|
f.close()
|
|
|
|
except Exception:
|
|
print(f"ERROR: failed to read battery {b} thresholds")
|